Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

MeteoRA: Multiple-tasks Embedded LoRA for Large Language Models

About

The pretrain+fine-tune paradigm is foundational for deploying large language models (LLMs) across various downstream applications. Within this framework, Low-Rank Adaptation (LoRA) stands out for its parameter-efficient fine-tuning (PEFT), producing numerous reusable task-specific LoRA adapters. However, this approach requires explicit task intention selection, posing challenges for autonomous task sensing and switching during inference with multiple existing LoRA adapters embedded in a single LLM. In this work, we introduce MeteoRA (Multiple-tasks embedded LoRA), a scalable and efficient framework that reuses multiple task-specific LoRA adapters into the base LLM via a full-mode Mixture-of-Experts (MoE) architecture. This framework also includes novel MoE forward acceleration strategies to address the efficiency challenges of traditional MoE implementations. Our evaluation, using the LlaMA2-13B and LlaMA3-8B base models equipped with 28 existing LoRA adapters through MeteoRA, demonstrates equivalent performance with the traditional PEFT method. Moreover, the LLM equipped with MeteoRA achieves superior performance in handling composite tasks, effectively solving ten sequential problems in a single inference pass, thereby demonstrating the framework's enhanced capability for timely adapter switching.

Jingwei Xu, Junyu Lai, Yunpeng Huang• 2024

Related benchmarks

TaskDatasetResultRank
Depth EstimationNYU Depth V2
RMSE0.86
177
Semantic segmentationNYUD v2
mIoU19.37
96
Surface Normal EstimationNYU V2
RMSE32.92
23
Saliency DetectionPASCAL-Context clean and adverse conditions v1 (unseen)
mIoU58.76
19
Human ParsingPASCAL-Context clean and adverse conditions v1 (unseen)
mIoU41.6
19
Multi-task LearningPASCAL-Context and NYUD-v2 clean and adverse conditions v1 (unseen)
Delta MADV (Adverse)-0.185
19
Semantic segmentationPASCAL-Context clean and adverse conditions v1 (unseen)
mIoU39.1
19
Surface Normal EstimationNYUD clean and adverse conditions v2 v1 (unseen)
RMSE20.7
18
Edge DetectionNYUD v2--
16
Showing 9 of 9 rows

Other info

Follow for update