arXiv:2510.15683cs.IRcs.AI2025-10中稿 · publication in Pro…

轻量模型上加专家混合模块,提升检索泛化能力。

Mixture of Experts Approaches in Dense Retrieval Tasks

  • 在最后层加单个专家混合模块,减少参数开销。
  • 轻量模型上效果显著,零样本跨数据集表现更优。
  • 适合资源有限但需跨领域泛化的检索任务。

稠密检索模型(DRMs)是信息检索的重要进展,但常受限于训练域的泛化能力。现有方法在Transformer层中引入多专家混合(MoE)框架,虽有效却大幅增加参数量。本文提出一种更高效设计:在最终Transformer层后添加单一专家混合块(SB-MoE)。我们在三个信息检索任务上进行实证评估,采用两种设置:第一,在7个基准上对四种底层模型微调并测试;第二,在MSMARCO上微调后,在13个BEIR数据集上进行零样本评估。结果表明,SB-MoE在轻量模型(如TinyBERT、BERT-Small)上表现优异,显著优于标准微调;而对大模型(如BERT-Base、Contriever),需更多训练样本才能提升性能。代码已开源。

原文摘要 · Abstract (English)

Dense Retrieval Models (DRMs) are a prominent development in Information Retrieval (IR). A key challenge with these neural Transformer-based models is that they often struggle to generalize beyond the specific tasks and domains they were trained on. To address this challenge, prior research in IR incorporated the Mixture-of-Experts (MoE) framework within each Transformer layer of a DRM, which, though effective, substantially increased the number of additional parameters. In this paper, we propose a more efficient design, which introduces a single MoE block (SB-MoE) after the final Transformer layer. To assess the retrieval effectiveness of SB-MoE, we perform an empirical evaluation across three IR tasks. Our experiments involve two evaluation setups, aiming to assess both in-domain effectiveness and the model's zero-shot generalizability. In the first setup, we fine-tune SB-MoE with four different underlying DRMs on seven IR benchmarks and evaluate them on their respective test sets. In the second setup, we fine-tune SB-MoE on MSMARCO and perform zero-shot evaluation on thirteen BEIR datasets. Additionally, we perform further experiments to analyze the model's dependency on its hyperparameters (i.e., the number of employed and activated experts) and investigate how this variation affects SB-MoE's performance. The obtained results show that SB-MoE is particularly effective for DRMs with lightweight base models, such as TinyBERT and BERT-Small, consistently exceeding standard model fine-tuning across benchmarks. For DRMs with more parameters, such as BERT-Base and Contriever, our model requires a larger number of training samples to achieve improved retrieval performance. Our code is available online at: https://github.com/FaySokli/SB-MoE.

稠密检索专家混合轻量模型零样本

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