arXiv:2602.05787cs.IRcs.AI2026-02

通过袋装合并提升文本嵌入模型的鲁棒性与增量学习效率。

Bagging-Based Model Merging for Robust General Text Embeddings

  • 用采样子集训练多模型,再通过袋装方法合并成单一模型。
  • 在多个基准上优于全数据批量打乱训练,且增量学习成本大幅降低。
  • 适合需要持续更新、对泛化能力要求高的实际应用场景。

通用文本嵌入模型广泛应用于自然语言处理与信息检索,通常在大规模多任务语料上训练以实现广泛泛化。然而,现有训练策略的实际效果差异尚不明确,且面对新领域和数据类型时难以高效适配。本文从数据调度与模型合并两个角度系统研究文本嵌入的多任务训练,对比了批量打乱、顺序训练、两阶段训练及多种合并粒度,发现简单批量打乱始终表现最优,表明任务冲突有限,数据集高度互补。尽管如此,该方法在域外泛化和增量学习方面仍存在缺陷:需全量重训,成本高昂。为此,提出基于袋装的鲁棒模型合并(BOOM),在采样子集上训练多个嵌入模型,并合并为单一模型,显著提升鲁棒性并保持单模型推理效率。此外,BOOM可通过对新数据与小历史子集训练轻量更新模型,实现高效增量更新。在多个嵌入基准上的实验表明,相比全数据批量打乱,BOOM在域内与域外性能均更优,且在增量学习中训练成本大幅下降。

原文摘要 · Abstract (English)

General-purpose text embedding models underpin a wide range of NLP and information retrieval applications, and are typically trained on large-scale multi-task corpora to encourage broad generalization. However, it remains unclear how different multi-task training strategies compare in practice, and how to efficiently adapt embedding models as new domains and data types continually emerge. In this work, we present a systematic study of multi-task training for text embeddings from two perspectives: data scheduling and model merging. We compare batch-level shuffling, sequential training variants, two-stage training, and multiple merging granularities, and find that simple batch-level shuffling consistently yields the strongest overall performance, suggesting that task conflicts are limited and training datasets are largely complementary. Despite its effectiveness, batch-level shuffling exhibits two practical limitations: suboptimal out-of-domain (OOD) generalization and poor suitability for incremental learning due to expensive full retraining. To address these issues, we propose Bagging-based rObust mOdel Merging (BOOM), which trains multiple embedding models on sampled subsets and merges them into a single model, improving robustness while retaining single-model inference efficiency. Moreover, BOOM naturally supports efficient incremental updates by training lightweight update models on new data with a small historical subset and merging them into the existing model. Experiments across diverse embedding benchmarks demonstrate that BOOM consistently improves both in-domain and OOD performance over full-corpus batch-level shuffling, while substantially reducing training cost in incremental learning settings.

文本嵌入模型合并增量学习袋装

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