LoRA on the Go: Instance-level Dynamic LoRA Selection and Merging
About
Low-Rank Adaptation (LoRA) has emerged as a parameter-efficient approach for fine-tuning large language models. However, conventional LoRA adapters are typically trained for a single task, limiting their applicability in real-world settings where inputs may span diverse and unpredictable domains. At inference time, existing approaches combine multiple LoRAs for improving performance on diverse tasks, while usually requiring labeled data or additional task-specific training, which is expensive at scale. In this work, we introduce LoRA on the Go (LoGo), a training-free framework that dynamically selects and merges adapters at the instance level without any additional requirements. LoGo leverages signals extracted from a single forward pass through LoRA adapters, to identify the most relevant adapters and determine their contributions on-the-fly. Across 5 NLP benchmarks, 27 datasets, and 3 model families, LoGo outperforms training-based baselines on some tasks upto a margin of 3.6% while remaining competitive on other tasks and maintaining inference throughput, highlighting its effectiveness and practicality.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Machine Translation | WMT en-fr 14 | BLEU30.4 | 32 | |
| Language Reasoning | BBH (BIG-Bench Hard) | Average BBH Score2.08 | 20 | |
| Big-Bench Hard | BBH | Boolean Expressions EM76.7 | 18 | |
| Struct to Text | CommonGen | ROUGE-156 | 11 | |
| Code Understanding and Generation | CodeXGLUE | Code Refinement Score46.3 | 6 | |
| Closed-book Question Answering | ARC-C | EM69.9 | 4 | |
| Natural Language Inference | ANLI R1 | ANLI R1 EM42.5 | 3 |