Beyond instruction-conditioning, MoTE: Mixture of Task Experts for Multi-task Embedding Models

Miguel Romero Calvo, Shuoyang Ding, Corey D Barrett, Georgiana Dinu, George Karypis


Abstract
Dense embeddings are fundamental to modern machine learning systems, powering Retrieval-Augmented Generation (RAG), information retrieval, and representation learning. While instruction-conditioning has become the dominant approach for embedding specialization, its direct application to low-capacity models imposes fundamental representational constraints that limit the performance gains derived from specialization. In this paper, we analyze these limitations and introduce the Mixture of Task Experts (MoTE) transformer block, which leverages task-specialized parameters trained with Task-Aware Contrastive Learning () to enhance the model’s ability to generate specialized embeddings. Empirical results show that MoTE achieves 64% higher performance gains in retrieval datasets (+3.27→ +5.21) and 43% higher performance gains across all datasets (+1.81→ 2.60). Critically, these gains are achieved without altering instructions, training data, inference time, or number of active parameters.
Anthology ID:
2025.findings-acl.1168
Volume:
Findings of the Association for Computational Linguistics: ACL 2025
Month:
July
Year:
2025
Address:
Vienna, Austria
Editors:
Wanxiang Che, Joyce Nabende, Ekaterina Shutova, Mohammad Taher Pilehvar
Venues:
Findings | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
22731–22746
Language:
URL:
https://preview.aclanthology.org/acl25-workshop-ingestion/2025.findings-acl.1168/
DOI:
Bibkey:
Cite (ACL):
Miguel Romero Calvo, Shuoyang Ding, Corey D Barrett, Georgiana Dinu, and George Karypis. 2025. Beyond instruction-conditioning, MoTE: Mixture of Task Experts for Multi-task Embedding Models. In Findings of the Association for Computational Linguistics: ACL 2025, pages 22731–22746, Vienna, Austria. Association for Computational Linguistics.
Cite (Informal):
Beyond instruction-conditioning, MoTE: Mixture of Task Experts for Multi-task Embedding Models (Calvo et al., Findings 2025)
Copy Citation:
PDF:
https://preview.aclanthology.org/acl25-workshop-ingestion/2025.findings-acl.1168.pdf