您所在的位置: 成果库 基于平衡树索引结构的向量检索模型的训练方法

基于平衡树索引结构的向量检索模型的训练方法

发布时间: 2023-06-25

基本信息

合作方式： 合作开发
成果类型： 发明专利,实用新型专利
行业领域：
电子信息技术
成果介绍
本发明提供了一种基于平衡树索引结构的向量检索模型的训练方法。
成果亮点
该方法包括：根据预设比例，从原始数据集中随机选择向量构成训练数据集，并计算出训练数据集中每个向量的邻居向量作为每个向量的标签值；根据预设的平衡树结构信息构建多个完全平衡叉树，并将原始数据集中的向量随机平均分配到多个完全平衡叉树的结点；根据优化目标函数，利用带有标签值的训练数据集对向量检索模型进行参数优化；利用参数优化后的向量检索模型通过预设的平衡树算法对多个完全平衡叉树中的空间划分进行动态更新；迭代进行模型参数优化操作以及空间更新操作，直到满足预设收敛条件，得到训练完成的向量检索模型。
团队介绍
连德富，陈恩红，黎武超，冯超，谢堉鑫，刘海峰
成果资料