arXiv:2512.20612cs.IRcs.CL2025-12ACL被引 3

通过压缩MLP层让大模型更高效地做文本检索。

Making Large Language Models Efficient Dense Retrievers

  • 发现检索任务中MLP层可大幅裁剪,注意力层仍关键。
  • 提出EffiR框架,实现模型尺寸与推理成本显著降低。
  • 适配多种大模型和数据集,保持原有检索效果。

近期研究显示,直接微调大语言模型(LLMs)用于密集检索可取得优异性能,但其庞大的参数量导致计算效率低下。尽管已有研究揭示了生成任务中存在显著的层冗余,但在需将整段序列编码为固定表示的检索任务中,这种冗余是否同样存在尚不明确。为此,我们对基于LLM的密集检索器进行了全面的层冗余分析。结果表明,与生成任务不同,在检索任务中MLP层具有更高的可压缩性,而注意力层对语义聚合仍至关重要。基于此洞察,我们提出EffiR框架,通过粗粒度深度削减与细粒度宽度缩减相结合的大规模MLP压缩策略,并辅以检索专用微调。在多个BEIR数据集及不同LLM主干上,EffiR实现了模型尺寸与推理开销的大幅降低,同时保持与全尺寸模型相当的性能。

原文摘要 · Abstract (English)

Recent work has shown that directly fine-tuning large language models (LLMs) for dense retrieval yields strong performance, but their substantial parameter counts make them computationally inefficient. While prior studies have revealed significant layer redundancy in LLMs for generative tasks, it remains unclear whether similar redundancy exists when these models are adapted for retrieval tasks, which require encoding entire sequences into fixed representations rather than generating tokens iteratively. To this end, we conduct a comprehensive analysis of layer redundancy in LLM-based dense retrievers. We find that, in contrast to generative settings, MLP layers are substantially more prunable, while attention layers remain critical for semantic aggregation. Building on this insight, we propose EffiR, a framework for developing efficient retrievers that performs large-scale MLP compression through a coarse-to-fine strategy (coarse-grained depth reduction followed by fine-grained width reduction), combined with retrieval-specific fine-tuning. Across diverse BEIR datasets and LLM backbones, EffiR achieves substantial reductions in model size and inference cost while preserving the performance of full-size models.

大模型压缩密集检索MLP压缩EffiR

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。