Investigating Task Arithmetic for Zero-Shot Information Retrieval
Large Language Models (LLMs) have shown impressive zero-shot performance across a variety of Natural Language Processing tasks, including document re-ranking. However, their effectiveness degrades on unseen tasks and domains, largely due to shifts in vocabulary and word distributions. In this paper, we investigate Task Arithmetic, a technique that combines the weights of LLMs pre-trained on different tasks or domains via simple mathematical operations, such as addition or subtraction, to adapt retrieval models without requiring additional fine-tuning. Our method is able to synthesize diverse tasks and domain knowledge into a single model, enabling effective zero-shot adaptation in different retrieval contexts. Extensive experiments on publicly available scientific, biomedical, and multilingual datasets show that our method improves state-of-the-art re-ranking performance by up to 18% in NDCG@10 and 15% in P@10. In addition to these empirical gains, our analysis provides insights into the strengths and limitations of Task Arithmetic as a practical strategy for zero-shot learning and model adaptation. We make our code publicly available at https://github.com/DetectiveMB/Task-Arithmetic-for-ZS-IR.
Authors
- Marco Braga (ORCID: https://orcid.org/0009-0004-7619-8399)
- Alessandro Raganato (ORCID: https://orcid.org/0000-0002-7018-7515)
- Pranav Kasela (ORCID: https://orcid.org/0000-0003-0972-2424)
- Gabriella Pasi (ORCID: https://orcid.org/0000-0002-6080-8170)
Institutions
- Politecnico di Torino (IT)
- University of Milano-Bicocca (IT)
- Politecnico di Milano (IT)
Publication Details
- Published
- 2025-07-13
- DOI
- https://doi.org/10.1145/3726302.3730216
- Citations
- 1
- Primary Topic
- Information Retrieval and Search Behavior
- Type
- article
- Field-Weighted Citation Impact
- 3.25
Funders
- Dipartimenti di Eccellenza
- European Commission
- Università degli Studi di Milano-Bicocca