跳转到正文
GETAI.APP
开发动态Hugging FaceGETAI.APP查看原文

Hugging Face 演示多向量检索压缩训练

在减少文档向量数量时一同训练检索器,降低压缩带来的信息损失。

Mean NanoBEIR nDCG@10 versus stored document vectors for pool factors 2, 3, 4, and 6 and unpooled retrieval, comparing ordinary and pooling-aware training.

Hugging Face 于 11 月 11 日发布 Token Pooling-Aware Training 教程,介绍如何让多向量检索模型适应压缩后的文档表示。

方法把相似的 Token 向量合并,同时在训练中加入压缩表示的检索目标。在作者的 NanoBEIR 实验里,文档向量减少至约四分之一时,普通训练保留原始检索分数的 94.6%,加入感知训练后提高至 97.8%。

文章提供 Sentence Transformers 复现示例,并说明压缩预算可以按文档长度或距离阈值调整。上述比例是相对未压缩分数的保留率,不是检索结果的绝对正确率。

来源:Hugging Face 原文。

图片来源:Hugging Face。

登录

每一次探索,都更进一步。

从这里,开启新的灵感。

还没有账号?