让多向量表示自带可聚类性,大幅减少存储开销且不损失检索效果
CRISP: Clustering Multi-Vector Representations for Denoising and Pruning
- 训练时直接学习可聚类的多向量表示,而非事后聚类
- 在BEIR上实现3倍向量压缩,性能仍超原始模型
- 11倍压缩仅损失3.6%性能,适合资源受限场景
多向量模型(如ColBERT)通过多个上下文嵌入表示查询和文档,在神经信息检索中表现卓越。但其庞大的表示规模带来了显著存储与计算开销,限制了实际应用。现有方法常对冻结向量进行聚类以缓解问题,但效果受限于嵌入本身的可聚类性。本文提出CRISP(Clustered Representations with Intrinsic Structure Pruning),一种端到端训练中直接学习固有可聚类表示的新方法。通过将聚类融入训练过程而非事后处理,CRISP在所有表示规模下均显著优于事后聚类及其它词元剪枝方法。在BEIR检索基准上,CRISP实现约3倍向量数缩减,同时性能超过原始未剪枝模型;更激进的聚类下,向量数减少11倍,仅造成3.6%的质量损失,表明学习到的聚类有效去噪,生成更鲁棒的多向量表示。
原文摘要 · Abstract (English)
Multi-vector models, such as ColBERT, are a significant advancement in neural information retrieval (IR), delivering state-of-the-art performance by representing queries and documents by multiple contextualized token-level embeddings. However, this increased representation size introduces considerable storage and computational overheads which have hindered widespread adoption in practice. A common approach to mitigate this overhead is to cluster the model's frozen vectors, but this strategy's effectiveness is fundamentally limited by the intrinsic clusterability of these embeddings. In this work, we introduce CRISP (Clustered Representations with Intrinsic Structure Pruning), a novel multi-vector training method which learns inherently clusterable representations directly within the end-to-end training process. By integrating clustering into the training phase rather than imposing it post-hoc, CRISP significantly outperforms post-hoc clustering at all representation sizes, as well as other token pruning methods. On the BEIR retrieval benchmarks, CRISP achieves a significant rate of ~3x reduction in the number of vectors while outperforming the original unpruned model. This indicates that learned clustering effectively denoises the model by filtering irrelevant information, thereby generating more robust multi-vector representations. With more aggressive clustering, CRISP achieves an 11x reduction in the number of vectors with only a $3.6\%$ quality loss.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。