Sentence Transformers 开源多向量池化训练
先了解这件事
Sentence Transformers 维护者 tomaarsen 在 v6.3 中开源面向 ColBERT 式 late interaction 多向量嵌入模型的 token pooling 感知训练方法,并公开训练代码、checkpoints 和评测结果。该方法通过合并相似 token 向量,减少文档索引所需的向量数量。 官方博客在相同 12 万条 MS MARCO 训练数据、初始化与优化设置下称,pool_factor=4 时文档向量减少 74.8%,nDCG@10 从 49.03 升至 51.23;另一篇汇总报道则称可节省 84.7% 存储、检索性能下降约 2.7%,两者口径不同,不能视为同一实验结论。后续实验中,antoine_chaffin 称 token pooling 与非对称嵌入量化组合效果较好,也可兼容 Matryoshka 表示学习;但池化程度和兼容性取决于骨干模型规模。
AI 根据报道生成 · 2 小时前更新
事件进展
2 个进展
- 11月12日 16:50 · 1 篇报道分享 token pooling 组合实验结果AGI Hunt:sentence-transformers 维护者实验:token pooling 与嵌入量化、Matryoshka 表示学习组合效果出色
- 11月11日 08:00 · 3 篇报道发布 Embedding 池化训练方法博客Hugging Face Blog:如何在 Sentence Transformers 中训练 Token Pooling 多向量嵌入模型
报道时间线
沿着报道,了解事件的不同侧面。
- AGI Huntsentence-transformers 新增 token pooling、量化与 MRL 训练支持
sentence-transformers 近期版本新增 token pooling、嵌入量化、多向量 Late Interaction 和 MRL 训练支持。token pooling 与非对称嵌入量化组合使用效果出色,也兼容 Matryoshka 表示;借助 Matryoshka 嵌入,token 向量可降至 16 维并保持可用,为嵌入存储与检索提供降本方案。
- AGI Huntsentence-transformers 维护者实验:token pooling 与嵌入量化、Matryoshka 表示学习组合效果出色
sentence-transformers 维护者分享实验结果:token pooling 与(非对称)嵌入量化组合效果很好,并可兼容 Matryoshka 表示学习。较大的骨干模型可采用 64 的池化且损失不大,但小骨干模型不适用;Matryoshka 损失更大,但对原生维度远大于 128 的模型效果不错。
- AGI Huntsentence-transformers 维护者开源 embedding 池化训练方法与代码
sentence-transformers 维护者 tomaarsen 开源了 embedding 池化训练方法及配套代码,并发布 checkpoints 与评测结果。该方法建立在 Learn to Pool、pooling 正则化研究之上,采用了 H Company 的 Late Interaction Kernels。作者还计划推出更多功能,并征集社区对 ST 使用瓶颈的建议。
- AGI HuntSentence Transformers v6.3 发布 token pooling 感知训练
Sentence Transformers v6.3 发布 token pooling 感知训练,面向 ColBERT 式 late interaction 多向量嵌入模型。训练后压缩文档向量可节省 84.7%,检索性能仅下降约 2.7%。
- Hugging Face Blog如何在 Sentence Transformers 中训练 Token Pooling 多向量嵌入模型
Hugging Face 介绍如何在 Sentence Transformers 中训练 Token Pooling 多向量嵌入模型,通过合并相似 token 向量减少文档索引占用。
本事件热度走势
- 可比范围当前
- 13
- 可比范围峰值
- 1611月12日 00:00
- 近 24 小时变化
- –
趋势仅比较持续完整观测到的相同主体,范围可能小于当前热度统计。移动指针或点击图表查看每小时热度;键盘可用左右方向键切换。