English

IDANI: Inference-time Domain Adaptation via Neuron-level Interventions

Computation and Language 2022-06-02 v1 Artificial Intelligence Machine Learning

Abstract

Large pre-trained models are usually fine-tuned on downstream task data, and tested on unseen data. When the train and test data come from different domains, the model is likely to struggle, as it is not adapted to the test domain. We propose a new approach for domain adaptation (DA), using neuron-level interventions: We modify the representation of each test example in specific neurons, resulting in a counterfactual example from the source domain, which the model is more familiar with. The modified example is then fed back into the model. While most other DA methods are applied during training time, ours is applied during inference only, making it more efficient and applicable. Our experiments show that our method improves performance on unseen domains.

Keywords

Cite

@article{arxiv.2206.00259,
  title  = {IDANI: Inference-time Domain Adaptation via Neuron-level Interventions},
  author = {Omer Antverg and Eyal Ben-David and Yonatan Belinkov},
  journal= {arXiv preprint arXiv:2206.00259},
  year   = {2022}
}

Comments

Our code is available at https://github.com/technion-cs-nlp/idani

R2 v1 2026-06-24T11:35:32.189Z