mlsanityData to 7 Oct 2026

Bagging-Based Model Merging for Robust General Text Embeddings

arXiv 2602.05787 ↗First posted 5 Feb 2026cs.IR cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 31 of its 64 references that are in the map. First posted on arXiv 5 Feb 2026.

Builds on

31 of 64 references.

References · 31

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
  3. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
  4. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
  5. 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,591
  6. 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,362
  7. 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,225
  8. 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,216
  9. 2020Gradient Surgery for Multi-Task Learningcited by 1,210
  10. 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
  11. 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,175
  12. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
19 more
  1. 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 705
  2. 2019ELI5: Long Form Question Answeringcited by 567
  3. 2023C-Pack: Packed Resources For General Chinese Embeddingscited by 556
  4. 2020SPECTER: Document-level Representation Learning using Citation-informed Transformerscited by 426
  5. 2020Efficient Intent Detection with Dual Sentence Encoderscited by 419
  6. 2024LLM2Vec: Large Language Models Are Secretly Powerful Text Encoderscited by 396
  7. 2022Text and Code Embeddings by Contrastive Pre-Trainingcited by 358
  8. 2023Fine-Tuning LLaMA for Multi-Stage Text Retrievalcited by 271
  9. 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
  10. 2022SGPT: GPT Sentence Embeddings for Semantic Searchcited by 204
  11. 2021Mr. TyDi: A Multi-lingual Benchmark for Dense Retrievalcited by 118
  12. 2025Gemini Embedding: Generalizable Embeddings from Geminicited by 93
  13. 2024Model Stock: All we need is just a few fine-tuned modelscited by 87
  14. 2024Making Text Embedders Few-Shot Learnerscited by 78
  15. 2023T2Ranking: A large-scale Chinese Benchmark for Passage Rankingcited by 59
  16. 2024FuseChat: Knowledge Fusion of Chat Modelscited by 53
  17. 2024Improving General Text Embedding Model: Tackling Task Conflict and Data Imbalance through Model Mergingcited by 14
  18. 2025Unleashing the Power of LLMs in Dense Retrieval with Query Likelihood Modelingcited by 5
  19. 2025Conan-Embedding-v2: Training an LLM from Scratch for Text Embeddingscited by 4

Built on by

Not cited yet.