Papers › MixLoRA: Enhancing Large Language Models Fine-Tuning with LoRA-based Mixture of Experts

MixLoRA: Enhancing Large Language Models Fine-Tuning with LoRA-based Mixture of Experts

22 Apr 2024arXiv:2404.15159archive 2025-07-28

Dengchun Li, Yingzi Ma, Naizheng Wang, Zhengmao Ye, Zhiyuan Cheng, Yinghao Tang, Yan Zhang, Lei Duan, Jie Zuo, Cal Yang, Mingjie Tang

Fine-tuning Large Language Models (LLMs) is a common practice to adapt pre-trained models for specific applications. While methods like LoRA have effectively addressed GPU memory constraints during fine-tuning, their performance often falls short, especially in multi-task scenarios. In contrast, Mixture-of-Expert (MoE) models, such as Mixtral 8x7B, demonstrate remarkable performance in multi-task learning scenarios while maintaining a reduced parameter count. However, the resource requirements of these MoEs remain challenging, particularly for consumer-grade GPUs with less than 24GB memory. To tackle these challenges, we propose MixLoRA, an approach to construct a resource-efficient sparse MoE model based on LoRA. MixLoRA inserts multiple LoRA-based experts within the feed-forward network block of a frozen pre-trained dense model and employs a commonly used top-k router. Unlike other LoRA-based MoE methods, MixLoRA enhances model performance by utilizing independent attention-layer LoRA adapters. Additionally, an auxiliary load balance loss is employed to address the imbalance problem of the router. Our evaluations show that MixLoRA improves about 9% accuracy compared to state-of-the-art PEFT methods in multi-task learning scenarios. We also propose a new high-throughput framework to alleviate the computation and memory bottlenecks during the training and inference of MOE models. This framework reduces GPU memory consumption by 40% and token computation latency by 30% during both training and inference.

PaperPDFCodeCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2404.15159")

Code

Syntology Ran 6 of 11 code samples harvested from 2 repositories linked to this paper; 5 have no recorded run. Of those that ran: 6 ran with no contract checked.

By repository: official repository: 11 samples from 2 repositories, 6 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

TUDB-Labs/MixLoRA officialmentioned in papermentioned on GitHubpytorchApache-2.0 report
mikecovlee/mLoRA officialmentioned on GitHubpytorchApache-2.0 report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

11 samples harvested; 6 ran; 0 honoured the contract we drafted; 5 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

6ran
5unverified

Licence: 0 of the 11 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from 2 repositories linked to this paper, official or community; each sample names its own and says which. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

apply_rotary_pos_emb mikecovlee/mLoRA/mlora/models/modeling_chatglm.py official repository ran Apache-2.0 (permissive) · 0a82ce127c214e4b · report
dequantize_module_weight TUDB-Labs/MixLoRA/mixlora/lora_linear.py official repository ran Apache-2.0 (permissive) · 5b93021aaf3e8b4c · report
is_package_available TUDB-Labs/MixLoRA/mixlora/utils.py official repository ran Apache-2.0 (permissive) · 45a78242fc74dcd9 · report
logits_process mikecovlee/mLoRA/mlora/generator.py official repository ran Apache-2.0 (permissive) · 22f4910fcb84030d · report
split_tensor_along_last_dim mikecovlee/mLoRA/mlora/models/modeling_chatglm.py official repository ran Apache-2.0 (permissive) · 9281708eb4cf9161 · report
swiglu mikecovlee/mLoRA/mlora/models/modeling_chatglm.py official repository ran fingerprinted Apache-2.0 (permissive) · 7f59a97bd91fdef5 · report
apply_partial_rotary_emb mikecovlee/mLoRA/mlora/models/modeling_phi.py official repository unverified Apache-2.0 (permissive) · 602bfa722a261a61 · report
dequantize_bnb_weight TUDB-Labs/MixLoRA/mixlora/lora_linear.py official repository unverified Apache-2.0 (permissive) · 3ce1a041a1abc119 · report
load_adapter_weights TUDB-Labs/MixLoRA/mixlora/model.py official repository unverified Apache-2.0 (permissive) · ae2f069de9355ede · report
load_dataset mikecovlee/mLoRA/mlora/dispatcher.py official repository unverified Apache-2.0 (permissive) · 949439173c840737 · report
query_yes_no mikecovlee/mLoRA/mlora.py official repository unverified Apache-2.0 (permissive) · 4cd248daacbead2f · report

Tasks

Common Sense ReasoningMixture-of-ExpertsMulti-Task LearningQuantizationQuestion AnsweringSentence CompletionText Classification

1 archive task tag without a task page not shown.

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Common Sense Reasoning ARC (Challenge) LLaMA-3 8B + MixLoRA Accuracy 79.9 #17 of 54 Archive leaderboard report
Common Sense Reasoning ARC (Challenge) LLaMA-2 13B + MixLoRA Accuracy 69.9 #18 of 54 Archive leaderboard report
Common Sense Reasoning ARC (Challenge) LLaMA-2 7B + MixLoRA Accuracy 58.1 #26 of 54 Archive leaderboard report
Common Sense Reasoning ARC (Easy) LLaMA-3 8B + MixLoRA Accuracy 86.5 #5 of 47 Archive leaderboard report
Common Sense Reasoning ARC (Easy) LLaMA-2 13B + MixLoRA Accuracy 83.5 #10 of 47 Archive leaderboard report
Common Sense Reasoning ARC (Easy) LLaMA-2 7B + MixLoRA Accuracy 77.7 #20 of 47 Archive leaderboard report
Common Sense Reasoning WinoGrande LLaMA-2 13B + MixLoRA Accuracy 86.3 #9 of 77 Archive leaderboard report
Common Sense Reasoning WinoGrande LLaMA-3 8B + MixLoRA Accuracy 82.1 #12 of 77 Archive leaderboard report
Common Sense Reasoning WinoGrande LLaMA-2 7B + MixLoRA Accuracy 76.8 #24 of 77 Archive leaderboard report
Question Answering BoolQ LLaMA-2 13B + MixLoRA Accuracy 77.1 #31 of 65 Archive leaderboard report
Question Answering BoolQ LLaMA-3 8B + MixLoRA Accuracy 75 #36 of 65 Archive leaderboard report
Question Answering BoolQ LLaMA-2 7B + MixLoRA Accuracy 72.7 #40 of 65 Archive leaderboard report
Question Answering OpenBookQA LLaMA-3 8B + MixLoRA Accuracy 84.8 #16 of 45 Archive leaderboard report
Question Answering OpenBookQA LLaMA-2 13B + MixLoRA Accuracy 83 #19 of 45 Archive leaderboard report
Question Answering OpenBookQA LLaMA-2 7B + MixLoRA Accuracy 81.6 #24 of 45 Archive leaderboard report
Question Answering PIQA LLaMA-3 8B + MixLoRA Accuracy 87.6 #4 of 67 Archive leaderboard report
Question Answering PIQA LLaMA-2 13B + MixLoRA Accuracy 86.8 #7 of 67 Archive leaderboard report
Question Answering PIQA LLaMA-2 7B + MixLoRA Accuracy 83.2 #15 of 67 Archive leaderboard report
Question Answering SIQA LLaMA-2 13B + MixLoRA Accuracy 82.5 #2 of 24 Archive leaderboard report
Question Answering SIQA LLaMA-3 8B + MixLoRA Accuracy 78.8 #10 of 24 Archive leaderboard report
Question Answering SIQA LLaMA-2 7B + MixLoRA Accuracy 78 #11 of 24 Archive leaderboard report
Sentence Completion HellaSwag LLaMA-2 13B + MixLoRA Accuracy 94.7 #7 of 89 Archive leaderboard report
Sentence Completion HellaSwag LLaMA-3 8B + MixLoRA Accuracy 93.3 #9 of 89 Archive leaderboard report
Sentence Completion HellaSwag LLaMA-2 7B + MixLoRA Accuracy 93.1 #10 of 89 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Introduced by this paper: MixLoRA

Absolute Position EncodingsAdamAttentionBPEDense ConnectionsDropoutLabel SmoothingLayer NormalizationLinear LayerMixLoRAMoEMulti-Head AttentionPosition-Wise Feed-Forward LayerResidual ConnectionSoftmaxTransformer

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections