在密集检索模型后加专家混合模块,小模型效果更优
Investigating Mixture of Experts in Dense Retrieval
- 在最后Transformer层后加入单个专家混合块(SB-MoE)
- TinyBERT加MoE后在4个数据集上均优于微调模型
- 小模型用少量数据即可提升效果,大模型需更多训练样本
尽管密集检索模型(DRMs)在信息检索领域取得进展,但其泛化能力和鲁棒性仍有限。为应对这一问题,本文探索在密集检索模型中引入专家混合(MoE)架构。不同于以往将MoE嵌入Transformer层的做法,本文在最终Transformer层输出后添加一个单一的MoE块(SB-MoE)。实验评估了该结构在检索有效性上的表现,对比了标准微调方法。我们对三种模型(TinyBERT、BERT、Contriever)在四个基准数据集上进行了有无MoE块的对比实验,并进一步研究了专家数量等参数对性能的影响。结果表明,对于参数较少的小模型(如TinyBERT),SB-MoE显著优于原始微调模型,在所有四个数据集上均表现更好;而对于参数较多的模型(如BERT和Contriever),则需要更多训练样本才能获得更好的检索效果。
原文摘要 · Abstract (English)
While Dense Retrieval Models (DRMs) have advanced Information Retrieval (IR), one limitation of these neural models is their narrow generalizability and robustness. To cope with this issue, one can leverage the Mixture-of-Experts (MoE) architecture. While previous IR studies have incorporated MoE architectures within the Transformer layers of DRMs, our work investigates an architecture that integrates a single MoE block (SB-MoE) after the output of the final Transformer layer. Our empirical evaluation investigates how SB-MoE compares, in terms of retrieval effectiveness, to standard fine-tuning. In detail, we fine-tune three DRMs (TinyBERT, BERT, and Contriever) across four benchmark collections with and without adding the MoE block. Moreover, since MoE showcases performance variations with respect to its parameters (i.e., the number of experts), we conduct additional experiments to investigate this aspect further. The findings show the effectiveness of SB-MoE especially for DRMs with a low number of parameters (i.e., TinyBERT), as it consistently outperforms the fine-tuned underlying model on all four benchmarks. For DRMs with a higher number of parameters (i.e., BERT and Contriever), SB-MoE requires larger numbers of training samples to yield better retrieval performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。