Learning Retrieval Models with Sparse Autoencoders

TL;DR

提出SPLARE,基于稀疏自编码器的多语言稀疏检索模型,性能优于词汇空间方法。

cs.LG 🔴 高级 2026-02-27 46 次浏览
Thibault Formal Maxime Louis Hervé Dejean Stéphane Clinchant
信息检索 稀疏自编码器 多语言 深度学习 模型压缩

核心发现

方法论

本文将预训练的稀疏自编码器(SAEs)引入到大规模语言模型(LLMs)中,通过在Transformer的中间层提取稀疏潜在特征,替代传统的词汇空间投影。采用基于对比学习的蒸馏训练方式,结合稀疏正则化(如FLOPS损失)实现高效稀疏表示。模型在多语言、多领域数据上进行微调,仅调节线性投影头,保持SAE参数冻结,从而实现跨语言泛化。训练过程中采用Top-K池化调节稀疏度,确保检索效率与效果平衡。模型在MMTEB、多语言检索任务中表现优异,超越词汇空间的SPLADE模型,尤其在多语种和域外任务中表现出色。

关键结果

  • SPLARE-7B模型在MMTEB多语和英语任务中达到了最高性能,平均nDCG@10提升2.4点,显著优于SPLADE-Llama,且参数量仅为2B-参数版本的三分之一,推理速度快,索引存储空间大幅减少。
  • 在多语言和跨域任务中,SPLARE模型表现出优异的泛化能力,尤其在低资源语言和专业领域(如法律、医疗)中,效果优于词汇空间模型,提升幅度达1.8点。
  • 通过在不同层和宽度的SAE上进行消融,发现中间层(约模型深度的2/3)提取的稀疏特征最优,宽度越大,检索性能越好,模型可扩展性强,支持超过百万维的潜在特征空间。

研究意义

该研究突破了词汇空间检索的局限,利用稀疏潜在特征实现跨语言、多领域的高效检索,解决多语种、多模态信息检索中的瓶颈。模型在保持高效的同时,增强了特征的语义结构和可解释性,为大规模多语言信息检索提供了新思路,推动了检索模型的泛化能力和实用性。特别是在多语种环境下,减少了词表依赖,提升了模型的鲁棒性和扩展性,具有广泛的应用前景。

技术贡献

本文首次将预训练SAEs引入到大模型的潜在空间中,提出SPLARE框架,结合中间层特征提取与稀疏正则化,有效提升多语言、多领域检索性能。通过只微调线性投影头,简化训练流程,降低模型复杂度,实现跨语种泛化。模型支持百万级潜在特征空间,突破了词汇空间的限制,为稀疏检索提供了理论基础和工程实现路径。实验验证了在多任务、多语言环境中的优越性,推动了稀疏表示在信息检索中的应用。

新颖性

本研究首次将大规模预训练SAEs应用于多语言稀疏检索,提出潜在词汇空间的稀疏表示替代传统词汇表投影,显著提升跨语言和跨领域的泛化能力。不同于以往词汇空间模型依赖固定词表,SPLARE利用可扩展的潜在特征空间实现更丰富的语义表达,解决多语种、多模态检索的瓶颈。这一创新突破了稀疏检索的技术边界,为未来多模态、多任务检索系统奠定基础。

局限性

  • 模型在极端专业领域(如特定代码或高度领域特化文本)表现仍有限,可能需要针对特定任务训练专用SAE。
  • 稀疏正则化参数调节复杂,需在效率和效果之间权衡,且在某些场景下稀疏度难以达到理想水平。
  • 训练大规模SAE模型成本较高,依赖丰富的计算资源,限制了其普及和应用范围。

未来方向

未来将探索多模态潜在空间的联合学习,结合视觉、声音等信息源,提升多模态检索能力。同时,优化稀疏正则化策略,增强模型在极端场景下的表现。还计划结合知识图谱和外部知识库,丰富潜在特征的语义表达,推动模型在实际应用中的鲁棒性和可解释性。

AI 总览摘要

随着大规模语言模型(LLMs)在信息检索中的广泛应用,如何提升检索效率与跨语种泛化能力成为研究热点。传统的词汇空间检索方法依赖固定词表,难以应对多语种和多模态场景的复杂需求。本文提出SPLARE,一种基于稀疏自编码器(SAEs)的潜在空间稀疏检索框架,创新性地将预训练的SAEs引入到Transformer的中间层,提取高维稀疏潜在特征,替代词汇空间投影。通过在多语言、多领域数据上微调,仅调节线性投影头,模型实现了跨语种的高效检索,显著优于词汇空间模型,尤其在多语种和专业领域表现出色。

实验结果显示,SPLARE-7B模型在MMTEB多语和英语任务中达到了最高性能,平均nDCG@10提升2.4点,参数量仅为2B,推理速度快,存储空间大幅减少。模型在多语种和跨域任务中展现出优异的泛化能力,特别是在低资源语言和专业场景中,效果优于传统模型。通过调节潜在特征空间宽度,模型具有良好的扩展性,支持百万级别的潜在维度。

该研究突破了词汇空间的限制,利用潜在特征实现跨语言、多模态的高效检索,为多语种信息检索提供了新思路。未来将结合多模态信息和知识图谱,进一步提升模型的语义丰富性和鲁棒性,推动检索技术的持续发展。

深度分析

研究背景

近年来,深度学习推动信息检索技术快速发展,密集表示(如BERT、CLIP)已成为主流。SPLADE等稀疏检索模型通过词汇空间实现高效索引,兼具可解释性,但受限于固定词表,难以应对多语种和多模态场景。多语言检索的挑战在于词表规模和多样性,导致模型泛化能力不足。近年来,预训练SAEs在多语种和多模态任务中展现出良好的潜力,提供了更丰富的语义特征,但尚未广泛应用于检索中。本文结合两者优势,提出潜在空间稀疏检索,旨在突破词汇空间的局限。

核心问题

传统词汇空间检索模型受限于固定词表,难以扩展到多语种和多模态场景,且在低资源语言和专业领域表现不足。密集表示虽性能优越,但存储和计算成本高,难以实现大规模高效索引。如何利用潜在空间的稀疏特征,兼顾效率和泛化能力,成为亟待解决的问题。该问题的核心在于如何设计高效、可扩展的潜在特征表示,并确保其在多语种、多领域中的表现一致性。

核心创新

本研究的创新点包括:1)引入预训练SAEs到Transformer中间层,提取稀疏潜在特征,替代词汇空间;2)设计稀疏正则化(如FLOPS)与Top-K池化,控制稀疏度,兼顾效率与效果;3)仅微调线性投影头,简化训练流程,增强模型泛化能力;4)支持百万级潜在特征维度,突破词表限制,提升多语种、多模态检索能力。这些创新解决了词汇空间的局限,提供了更丰富的语义表达。

方法详解

  • �� 选择预训练的SAEs(如Llama Scope)在Transformer中间层提取稀疏特征。• 利用稀疏自编码器(Encoder+Decoder)将密集表示转化为高维稀疏向量。• 采用对比蒸馏训练,保持模型性能,同时引入FLOPS正则化确保稀疏性。• 只微调线性投影头,冻结SAE参数,简化训练流程。• 在训练中引入Top-K池化调节稀疏度,确保索引效率。• 利用倒排索引结构支持稀疏向量的高效检索。• 通过多语种、多领域数据进行微调,增强模型泛化能力。

实验设计

在MMTEB、多语种和专业领域数据集上进行评估,比较SPLARE与词汇空间模型(如SPLADE)的性能差异。采用nDCG@10、Recall等指标,调节潜在空间宽度和稀疏正则化参数。进行多层次消融,分析不同层和宽度对性能的影响。模型训练在大规模多语种数据上进行,采用LoRA微调,确保泛化能力。实验还验证了稀疏正则化和Top-K池化对效率和效果的影响。

结果分析

SPLARE-7B在MMTEB多语和英语任务中表现优异,平均nDCG@10提升2.4点,参数量仅为2B,推理速度快,存储空间减少70%。在低资源语言和专业领域表现优于词汇模型,提升幅度达1.8点。中间层(约深度的2/3)提取的特征最优,宽度越大,性能越好,支持百万维潜在空间。模型的稀疏性调节灵活,能在保持效果的同时大幅提升检索效率。

应用场景

该模型适用于多语种搜索引擎、跨模态信息检索、专业领域知识库检索等场景。只需少量参数调节,便能实现高效索引和快速检索,尤其适合大规模多语种、多模态环境。未来可结合知识图谱,丰富潜在特征的语义信息,提升系统的智能化水平。

局限与展望

模型在极端专业场景(如特定代码或高度领域化文本)表现有限,需训练专用SAE。稀疏正则化参数调节复杂,可能影响模型稳定性。大规模SAE训练成本高,限制其普及。未来需优化稀疏控制策略,提升模型的适应性和效率。

通俗解读 非专业人士也能看懂

想象你在一个工厂里,工厂里有很多工人(代表模型的不同部分),他们负责把原材料(信息)变成成品(检索结果)。以前,工厂只用一个固定的仓库(词汇表)存放所有原材料,但仓库太小,不能装下所有新材料,也不能灵活应对不同的订单。现在,工厂引入了一个智能的存储系统(潜在空间),可以根据订单自动生成和存储新材料的特征(稀疏潜在特征),这样工人们就能更快、更准确地找到所需的材料。这个新系统不仅节省空间,还能应对各种不同的订单(多语种、多领域),让工厂变得更高效、更智能。通过这个比喻,你可以理解,SPLARE就像这个智能存储系统,帮助搜索引擎更快、更好地找到信息。

简单解释 像给14岁少年讲一样

想象你在学校图书馆找书,以前你只能用书架上的标签(词汇)来找书,但标签有限,不能描述所有内容。现在,图书馆引入了一种新方法,把每本书的内容用一组关键词(潜在特征)描述,这些关键词可以更灵活、更详细地反映书的内容。你只需要记住这些关键词,就能更快找到想要的书。这个新方法就像SPLARE,把信息用很多隐藏的关键词表示,不依赖固定的标签,能在不同语言和不同类型的书中都找到你想要的内容。这样,搜索变得更智能、更快,也更适合多种场景。

术语表

稀疏自编码器 (Sparse Autoencoder)

一种神经网络模型,用于将密集表示转化为稀疏潜在特征,便于高效索引和解释。

在本文中,SAE用于提取Transformer中间层的稀疏潜在特征。

潜在空间 (Latent Space)

由模型学习得到的高维特征空间,用于表示输入数据的抽象语义信息。

SPLARE在潜在空间中进行稀疏表示,替代传统词汇空间。

Top-K池化 (Top-K Pooling)

一种稀疏正则化技术,保留最大的K个特征,抑制其他特征。

用于调节模型输出的稀疏度,提升检索效率。

对比蒸馏 (Contrastive Distillation)

利用教师模型的输出作为目标,训练学生模型以模仿其表现。

本文采用此方法训练稀疏潜在特征。

MMTEB

多语种多任务信息检索基准,评估多语种模型性能。

用以测试模型在多语种、多领域的检索效果。

开放问题 这项研究留下的未解疑问

  • 1 如何进一步提升潜在空间的表达丰富性,尤其在极端专业场景下的表现仍有待探索。
  • 2 稀疏正则化参数的自动调节机制尚未完善,影响模型的稳定性和普适性。

应用场景

近期应用

多语种搜索引擎

利用SPLARE实现跨语种信息检索,提升多语环境下的搜索效率和准确性。

远期愿景

多模态知识库

结合视觉、声音等多模态信息,构建更智能的多模态检索系统,推动智能助理和机器人技术发展。

原文摘要

Sparse autoencoders (SAEs) provide a powerful mechanism for decomposing the dense representations produced by Large Language Models (LLMs) into interpretable latent features. We posit that SAEs constitute a natural foundation for Learned Sparse Retrieval (LSR), whose objective is to encode queries and documents into high-dimensional sparse representations optimized for efficient retrieval. In contrast to existing LSR approaches that project input sequences into the vocabulary space, SAE-based representations offer the potential to produce more semantically structured, expressive, and language-agnostic features. Building on this insight, we introduce SPLARE, a method to train SAE-based LSR models. Our experiments, relying on recently released open-source SAEs, demonstrate that this technique consistently outperforms vocabulary-based LSR in multilingual and out-of-domain settings. SPLARE-7B, a multilingual retrieval model capable of producing generalizable sparse latent embeddings for a wide range of languages and domains, achieves top results on MMTEB's multilingual and English retrieval tasks. We also developed a 2B-parameter variant with a significantly lighter footprint.

cs.LG cs.AI cs.IR