ACE: Adapter Consolidation across Experts for Parameter-Efficient Fine-Tuning of MoE LLMs
Organizations: Ulsan National Institute of Science and Technology (UNIST) Ulsan, Republic of Korea
Abstract
Parameter-efficient fine-tuning (PEFT) of mixture-of-experts (MoE) models commonly attaches a separate low-rank adapter to each expert. This expert-wise design fragments adaptation in three ways: capacity is split across narrow low-rank updates, gradient supervision becomes sparse and imbalanced under sparse routing, and execution is decomposed into many small GEMMs. We find that such expert-wise separation is often unnecessary, as subsets of LoRA adapters become functionally similar during fine-tuning, revealing redundancy among expert-specific adapters. Based on this redundancy, we propose ACE (Adapter Consolidation across Experts), which groups redundant experts and replaces their expert-specific adapters with group-shared higher-rank LoRA modules under the same PEFT budget. ACE further introduces grouped adapter execution, which consolidates fragmented expert-wise adapter computations into fewer, larger group-level GEMMs. Across evaluations covering 12 datasets and four MoE backbones, ACE achieves the highest observed mean accuracy among the parameter-matched PEFT methods on the three backbones with complete baseline coverage, while providing to wall-clock training speedup over expert-wise LoRA without increasing peak memory. Our code is available at https://github.com/UbiquitousAILab/ACE.
Figures & tables
| Model | Method | # Params (%) | HS | WG | ARC-c | ARC-e | BoolQ | OBQA | PIQA | SIQA | Avg. |
| OLMoE (1B/7B) | LoRA | 0.27 | 90.82 | 70.08 | 69.42 | 85.61 | 66.71 | 74.60 | 83.10 | 52.51 | 74.11 |
| DoRA | 0.33 | 90.98 | 70.25 | 69.46 | 85.98 | 66.96 | 74.93 | 83.31 | 52.35 | 74.28 | |
| AdaLoRA | 0.54 0.27 | 90.68 | 68.14 | 68.51 | 84.69 | 65.52 | 73.67 | 83.08 | 51.31 | 73.20 | |
| PiSSA | 0.27 | 90.98 | 70.46 | 69.96 | 86.38 | 68.71 | 75.27 | 83.44 | 52.83 | 74.75 | |
| PERFT-R | 0.27 | 87.36 | 71.46 | 66.38 | 82.76 | 64.04 | 73.87 | 80.05 | 51.62 | 72.19 | |
| EPnG | 0.27 | 90.82 | 68.95 | 68.83 | 85.75 | 66.96 | 74.47 | 83.15 | 52.27 | 73.90 |
| Model | Peak Mem. (GB) | Train Time (hr) | Speed up | ||
| LoRA | ACE | LoRA | ACE | ||
| OLMoE | 98.21 | 98.18 | 2.47 | 1.89 | 1.31 |
| Qwen | 89.90 | 89.84 | 6.18 | 4.19 | 1.48 |
| Moonlight | 48.80 | 48.80 | 23.75 | 16.77 | 1.42 |
| Kimi | 118.22 | 118.22 | 87.39 | 60.41 | 1.45 |
| OLMoE | Qwen | |||||||||||
| Method | # Params (%) | GSM8K | MATH | HEval | CHold | Avg. | # Params (%) | GSM8K | MATH | HEval | CHold | Avg. |
| LoRA | 1.08 | 65.73 | 25.95 | 30.49 | 78.53 | 50.18 | 0.86 | 66.34 | 36.31 | 77.44 | 72.11 | 63.05 |
| AdaLoRA | 2.16 1.08 | 65.20 | 24.68 | 26.22 | 75.65 | 47.94 | 1.72 0.86 | 66.64 | 38.33 | 79.27 | 69.06 | 63.33 |
| DoRA | 1.14 | 66.11 | 25.78 | 28.05 | 78.64 | 49.65 | 0.91 | 67.10 | 37.41 | 77.44 | 72.26 | 63.55 |
| PiSSA | 1.08 | 65.88 | 24.68 | 32.32 | 79.83 | 50.68 | 0.86 | 66.19 | 39.34 | 75.61 | 74.93 | 64.02 |
| ACE (ours) | 1.08 | 66.49 | 26.96 | 31.10 | 82.57 | 51.78 | 0.86 | 67.48 | 37.49 | 79.27 | 75.24 | 64.87 |
| Metric | LoRA | ACE |
| Matrix multiplication calls | 106.0K | 62.2K ( ) |
| Matrix multiplication CUDA time | 838ms | 558ms ( ) |
| CUDA kernel events | 482.9K | 389.0K ( ) |
| CUDA kernel time | 10.36s | 8.14s ( ) |
| Wall-clock time | 20.24s | 17.68s ( ) |
| Method | Sim. Group | Rank Cons. | Grp. Exec. | Avg. | Time | Mem. |
| LoRA | 74.11 | 2.47 | 98.21 | |||
| w/o Sim. Group | ||||||
| Random | ✓ | ✓ | 74.72 | 1.89 | 98.18 | |
| Full Merge | ✓ | ✓ | 73.57 | 2.03 | 97.61 | |
| w/o Rank Cons. | ✓ | ✓ | 72.63 | 1.84 | 96.54 | |
| w/o Grp. Exec. | ✓ | ✓ | 75.61 | 2.83 | 98.18 | |
| OLMoE | Qwen | |||
| Adaptation Scope | # Params | Avg. | # Params | Avg. |
| Attention only (Q, K, V) | 0.36 | 0.26 | ||
| MoE only (LoRA) | 0.27 | 0.21 | ||
| MoE only ( ACE ) | 0.27 | 0.21 | ||
| Attn + MoE (LoRA) | 0.63 | 0.47 | ||
| Attn + MoE ( ACE ) | 0.63 | 0.47 | ||
Appendix figures & tables23 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | OLMoE-1B-7B | Qwen1.5-MoE-A2.7B | Moonlight-16B-A3B | Kimi-Linear-48B-A3B |
| Total Parameters | 6.9B | 14.3B | 16B | 48B |
| Active Parameters | 1.3B | 2.7B | 3B | 3B |
| Layers ( ) | 16 | 24 | 27 | 27 |
| Hidden Dimension ( ) | 2048 | 2048 | 2048 | 2304 |
| Number of Experts ( ) | 64 | 60 + 1 (shared) | 64 + 2 (shared) | 256 + 1 (shared) |
| Top-K Routing ( ) | 8 | 4 | 6 | 8 |
| Backbone | Method | Seed 0 | Seed 1 | Seed 2 |
| OLMoE | PiSSA | 74.58 | 74.90 | 74.78 |
| ACE | 75.70 | 75.84 | 75.30 | |
| Qwen | PiSSA | 80.11 | 79.96 | 79.89 |
| ACE | 80.32 | 80.42 | 80.07 | |
| Moonlight | EPnG | 79.89 | 80.13 | 80.36 |
| ACE | 80.98 | 81.35 | 81.19 |
| Method | HS | WG | ARC-c | ARC-e | BoolQ | OBQA | PIQA | SIQA | Avg. | |
| LoRA | – | 90.82 | 70.08 | 69.42 | 85.61 | 66.71 | 74.60 | 83.10 | 52.51 | 74.11 |
| ACE | 50 | 92.14 | 72.45 | 71.33 | 86.62 | 69.36 | 77.40 | 82.37 | 53.38 | 75.63 |
| ACE | 150 | 92.27 | 73.32 | 70.82 | 86.45 | 69.14 | 76.60 | 84.06 | 52.97 | 75.70 |
| ACE | 500 | 92.52 | 70.88 | 69.62 | 86.41 | 68.78 | 76.40 | 83.73 | 53.53 | 75.23 |
| Step | Load Entropy | Aux. Loss |
| LoRA / ACE | LoRA / ACE | |
| 500 | 0.9613 / 0.9605 | 1.1168 / 1.1214 |
| 1000 | 0.9614 / 0.9616 | 1.1161 / 1.1162 |
| Final | 0.9619 / 0.9620 | 1.1142 / 1.1133 |
| Method | Group Count | Avg. | Eval Time (s) |
| LoRA | – | 74.11 | 20.82 |
| ACE ( ) | 242 | 74.59 | 19.76 |
| ACE ( ) | 215 | 75.04 | 17.67 |
| ACE ( ) | 141 | 75.21 | 17.24 |
| ACE ( , default) | 97 | 75.61 | 17.22 |
| ACE ( ) | 67 | 75.75 | 16.97 |
| Metric | Centered CKA | Uncentered CKA |
| Raw MSE Gap | -2.417e-6 | -3.911e-6 |
| Louvain Q | 0.395 | 0.374 |
| Downstream Average Accuracy | 75.70 | 75.44 |
| Probe Tokens | Jaccard | Pair-F1 | ARI |
| 512 | 0.923 | 0.953 | 0.922 |
| 1,000 | 0.946 | 0.969 | 0.948 |
| 2,000 | 0.958 | 0.976 | 0.960 |
| 5,000 | 0.989 | 0.994 | 0.991 |
| 10,000 | 1.000 | 1.000 | 1.000 |
| Method | HellaSwag | WinoGrande | OBQA | Average |
| Avg. | 92.52 | 67.17 | 76.60 | 78.76 |
| Max | 92.46 | 68.27 | 76.40 | 79.04 |
| CKA-center | 92.27 | 68.98 | 76.60 | 79.28 |
| Model | Method | # Params (%) | HellaSwag | WinoGrande | ARC-c | ARC-e | BoolQ | OBQA | PIQA | Avg. |
| OLMoE | LoRA | 0.27 | 87.58 | 64.09 | 53.92 | 67.85 | 73.55 | 64.40 | 70.13 | 68.79 |
| AdaLoRA | 0.54 0.27 | 86.06 | 62.27 | 55.63 | 73.99 | 71.83 | 62.60 | 64.64 | 68.15 | |
| DoRA | 0.33 | 87.41 | 63.06 | 52.13 | 69.99 | 75.90 | 60.80 | 74.97 | 69.18 | |
| ACE (Ours) | 0.27 | 89.08 | 71.11 | 55.46 | 75.29 | 78.90 | 64.60 | 78.51 | 73.28 | |
| Qwen | LoRA | 0.21 | 91.82 | 77.03 | 73.98 | 87.84 | 83.70 | 78.60 | 84.77 | 82.53 |
| AdaLoRA | 0.42 0.21 | 90.19 | 65.90 | 75.26 | 88.30 | 73.24 | 60.60 | 82.48 | 76.57 |
| Method | ROUGE-1 | ROUGE-2 | ROUGE-L |
| LoRA | 32.98 | 13.77 | 26.27 |
| DoRA | 33.16 | 13.90 | 26.45 |
| AdaLoRA | 30.01 | 11.82 | 24.17 |
| PiSSA | 30.48 | 12.26 | 24.72 |
| PERFT-R | 33.57 | 14.31 | 26.90 |
| ACE | 33.82 | 14.37 | 27.01 |
| Model | Method | # Params (%) | HS | WG | ARC-c | ARC-e | BoolQ | OBQA | PIQA | SIQA | Avg. |
| OLMoE (1B/7B) | LoRA | 0.27 | 90.82 | 70.08 | 69.42 | 85.61 | 66.71 | 74.60 | 83.10 | 52.51 | 74.11 |
| ACE | 0.27 | 92.24 | 72.85 | 70.88 | 86.21 | 68.04 | 77.33 | 83.93 | 53.41 | 75.61 | |
| Qwen (3B/14B) | LoRA | 0.21 | 93.23 | 77.66 | 79.44 | 90.63 | 70.84 | 82.67 | 86.80 | 56.72 | 79.75 |
| ACE | 0.21 | 93.81 | 79.19 | 79.50 | 90.66 | 71.49 | 83.87 | 86.67 | 56.99 | 80.27 | |
| Moonlight (3B/16B) | LoRA | 0.22 | 92.23 | 74.59 | 84.67 | 94.28 | 67.17 | 82.00 | 86.92 | 52.47 | 79.29 |
| ACE | 0.22 | 93.49 | 79.95 | 84.30 | 94.75 | 69.31 | 84.93 | 87.71 | 54.93 | 81.17 |
| Group. | HS | WG | ARC-c | ARC-e | BoolQ | OBQA | PIQA | SIQA | Avg. |
| Louvain | 92.24 | 72.85 | 70.88 | 86.21 | 68.04 | 77.33 | 83.93 | 53.41 | 75.61 |
| Thresh. | 92.17 | 71.77 | 71.05 | 86.66 | 67.94 | 77.07 | 84.21 | 53.15 | 75.50 |
| Component | LoRA (ms) | ACE (ms) | Reduction |
| Total step | 43.92% | ||
| Forward | 37.47% | ||
| Adapter forward | 89.16% | ||
| Backward | 49.99% | ||
| Optimizer | – | ||
| Router linear | – |
| Model | Method | Train Time (hr) | Peak Memory (GB) |
| Moonlight | LoRA | 23.75 0.60 | 48.80 0.00 |
| AdaLoRA | 43.84 2.05 | 50.07 0.62 | |
| DoRA | 41.10 0.95 | 80.88 0.02 | |
| PiSSA | 23.64 0.33 | 48.80 0.00 | |
| ACE (ours) | 16.77 0.23 | 48.80 0.00 | |
| Qwen | LoRA | 6.18 0.05 | 89.90 0.01 |
| Model | Layers | Total Experts | Warm-up Overhead (s) | CKA (s) | Louvain (s) | Consolidation (s) | Total Overhead (% of ACE) | ACE Speedup over LoRA | |
| OLMoE-1B-7B | 64 | 16 | 1,024 | 0.00 | 123.90 | 0.12 | 0.37 | 0.035 hr (1.83%) | 1.31 |
| Qwen1.5-MoE-A2.7B | 60 | 24 | 1,440 | 0.00 | 275.17 | 0.15 | 0.94 | 0.077 hr (1.83%) | 1.48 |
| Moonlight-16B-A3B | 64 | 27 | 1,728 | 0.00 | 328.89 | 0.19 | 0.56 | 0.092 hr (0.55%) | 1.42 |
| Kimi-Linear-48B-A3B | 256 | 27 | 6,912 | 0.00 | 2393.11 | 0.79 | 3.19 | 0.666 hr (1.10%) | 1.45 |