arXiv · 2311.10847
Token-Level Adaptation of LoRA Adapters for Downstream Task Generalization
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.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Joshua Belofsky. 2023-11-17. Token-Level Adaptation of LoRA Adapters for Downstream Task Generalization. https://arxiv.org/abs/2311.10847
Cite the original work for its findings. Save a collection to share your selection of sources.