Investigating Task Arithmetic for Zero-Shot Information Retrieval
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 54 of its 79 references that are in the map. First posted on arXiv 1 May 2025.
Builds on
54 of 79 references.
References · 54
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,818
- 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,605
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
38 more
- 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,252
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 766
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
- 2021Merging Models with Fisher-Weighted Averagingcited by 621
- 2019On the Cross-lingual Transferability of Monolingual Representationscited by 578
- 2020Fact or Fiction: Verifying Scientific Claimscited by 554
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
- 2021Recent Advances in Natural Language Processing via Large Pre-Trained Language Models: A Surveycited by 463
- 2020SPECTER: Document-level Representation Learning using Citation-informed Transformerscited by 430
- 2020Document Ranking with a Pretrained Sequence-to-Sequence Modelcited by 376
- 2023Large Language Models for Information Retrieval: A Surveycited by 331
- 2023Large Language Models are Effective Text Rankers with Pairwise Ranking Promptingcited by 307
- 2023Fine-Tuning LLaMA for Multi-Stage Text Retrievalcited by 272
- 2020Pretrained Transformers for Text Ranking: BERT and Beyondcited by 267
- 2022Large Language Models are Few-Shot Clinical Information Extractorscited by 207
- 2022Branch-Train-Merge: Embarrassingly Parallel Training of Expert Language Modelscited by 191
- 2021UniPELT: A Unified Framework for Parameter-Efficient Language Model Tuningcited by 162
- 2021mMARCO: A Multilingual Version of the MS MARCO Passage Ranking Datasetcited by 116
- 2020The Early Phase of Neural Network Trainingcited by 115
- 2024Language Models are Homer Simpson! Safety Re-Alignment of Fine-tuned Language Models through Task Arithmeticcited by 112
- 2022Fusing finetuned models for better pretrainingcited by 110
- 2024Branch-Train-MiX: Mixing Expert LLMs into a Mixture-of-Experts LLMcited by 109
- 2021SciFive: a text-to-text transformer model for biomedical literaturecited by 99
- 2023AdapterSoup: Weight Averaging to Improve Generalization of Pretrained Language Modelscited by 84
- 2023A Setwise Approach for Effective and Highly Efficient Zero-shot Ranking with Large Language Modelscited by 55
- 2023Chat Vector: A Simple Approach to Equip LLMs with Instruction Following and Model Alignment in New Languagescited by 54
- 2024Understanding the Performance and Estimating the Cost of LLM Fine-Tuningcited by 48
- 2021GermanQuAD and GermanDPR: Improving Non-English Question Answering and Passage Retrievalcited by 39
- 2023Elastic Weight Removal for Faithful and Abstractive Dialogue Generationcited by 29
- 2023Language and Task Arithmetic with Parameter-Efficient Layers for Zero-Shot Summarizationcited by 28
- 2023Position: Key Claims in LLM Research Have a Long Tail of Footnotescited by 20
- 2023Exploring the Effectiveness of Instruction Tuning in Biomedical Language Processingcited by 16
- 2023Can Query Expansion Improve Generalization of Strong Cross-Encoder Rankers?cited by 12
- 2022On the Effectiveness of Compact Biomedical Transformerscited by 11
- 2024Synthetic Data Generation with Large Language Models for Personalized Community Question Answeringcited by 8
- 2024Leveraging Large Language Models for Medical Information Extraction and Query Generationcited by 8
50 most cited of 54 All 54 (JSON)
Built on by
Citing papers per year: 2025: 2; 2026: 2.
4 papers.
- 2026Bus-Conditioned Zero-Shot Trajectory Generation via Task Arithmeticcited by 2
- 2025Look the Other Way: Designing 'Positive' Molecules with Negative Data via Task Arithmeticcited by 0
- 2025Investigating Large Language Models' Linguistic Abilities for Text Preprocessingcited by 0
- 2026Improving Ad-hoc Search Effectiveness for Conversational Information Retrieval via Model Mergingcited by 0
- 2026Improving Ad-hoc Search Effectiveness for Conversational Information Retrieval via Model Merging9 Jul 2026
- 2026Bus-Conditioned Zero-Shot Trajectory Generation via Task Arithmetic13 Feb 2026
- 2025Investigating Large Language Models' Linguistic Abilities for Text Preprocessing13 Oct 2025
- 2025Look the Other Way: Designing 'Positive' Molecules with Negative Data via Task Arithmetic23 Jul 2025