STE 正则化让 ColBERT 实现 5 倍压缩仍保留 99.4% 检索效果

antoine_chaffin · x · 2026-11-11

Antoine Chaffin 介绍 Hugging Face 上的新文章:在之前分层池化(hierarchical pooling)可将 ColBERT 存储减半且不损检索的基础上,引入与修复 MUVERA/SMVE 时相同的 STE 正则化技巧,把压缩率推到 5 倍、保留 99.4% 检索效果,且不伤全 token 性能。核心洞察:想要可压缩性,最好的方式是直接为它训练。文章还探讨了按预算匹配目标、自适应预算(不同文档信息密度不同)等内容,并发布 LateOn-hpool-regularized 等模型。作者同时提到 MTRL 对多向量模型有价值,认为当前多向量维度仍太大。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →