跳到正文
OpenReview · Representation learning· Biao Zhang; Lixin Chen; Tong Liu; Bo Zheng·· 2025-05-10精选AI 评分74

SMEC:重新思考用于检索嵌入压缩的 Matryoshka Representation Learning

SMEC:Rethinking Matryoshka Representation Learning for Retrieval Embedding Compression

AI 导读

基于摘要分析。研究针对 LLM 高维检索嵌入带来的存储与计算开销,提出 Sequential Matryoshka Embedding Compression(SMEC)训练框架,目标是在降低嵌入维度的同时保留检索性能。其主要研究对象是图像、文本及多模态表示压缩,并非 EEG 或 BCI 解码。 框架包含三个部分:Sequential Matryoshka Representation Learning(SMRL)旨在缓解训练中的梯度方差;Adaptive Dimension Selection(ADS)旨在减少维度裁剪造成的信息退化;Selectable Cross-batch Memory(S-XBM)旨在增强高维与低维嵌入之间的无监督学习。摘要未给出序列训练流程、维度选择准则、记忆样本筛选机制或具体损失形式,尚不能判断各模块的独立贡献与计算代价。 作者报告,在图像、文本及多模态数据集上的实验显示,SMEC 可在显著降低维度的同时保持性能。具体而言,在 BEIR 上,压缩至 256 维的 LLM2Vec 嵌入,相较 Matryoshka-Adaptor 与 Search-Adaptor 分别提升 1.1 点和 2.7 点。摘要未明确这些分值对应的指标、BEIR 任务范围、聚合方式及训练与评估划分,因此不能将其解释为 Accuracy 或百分比提升;实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 系统已有可用于检索的高维嵌入,SMEC 的高低维表示学习与自适应维度选择可能用于探索存储压缩,但这不等同于改善 BCI 解码。迁移依赖稳定的高维参考表示及适合跨批次学习的样本关系;需调整维度选择与记忆样本筛选,以避免保留被试或会话差异而损失任务信息。低信噪比、通道配置变化和小样本均可能使压缩失效。可在固定 EEG 编码器、严格分离训练与测试被试的条件下,对比等维度的 SMEC 压缩与简单截断,检验检索质量和存储开销;已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对 SMRL、ADS 与 S-XBM 分别如何处理压缩训练中的梯度方差、维度裁剪信息损失和高低维表示对齐;摘要给出了 BEIR 对照结果,但指标及评估协议尚未验证。

来源:OpenReview · Representation learning · openreview.net