English

Token-Level Adaptation of LoRA Adapters for Downstream Task Generalization

Computation and Language 2023-12-04 v2 Machine Learning

Abstract

This paper introduces a method for adapting LoRA adapters in smaller-sized language models to arbitrary downstream tasks. Unlike standard mixture-of-expert architectures, our method employs a gradient-free routing function to choose a weighted combination of experts without increasing the compute requirements for training or inference. The results show that token-level adaptation of LoRA adapters outperforms the base Llama-2-7b model across mathematical (GSM8K), scientific (ARC-Challenge), reading comprehension (SQuAD), and coding (CodeAlpaca-20k) tasks. Further evaluations also show that the average performance of token-level adaptation outperforms individual models fine-tuned for each of the tasks with the best performance observed in adaptation of every-other token during inference. The code for this study is made available through a public repository.

Keywords

Cite

@article{arxiv.2311.10847,
  title  = {Token-Level Adaptation of LoRA Adapters for Downstream Task Generalization},
  author = {Joshua Belofsky},
  journal= {arXiv preprint arXiv:2311.10847},
  year   = {2023}
}
R2 v1 2026-06-28T13:24:43.629Z