Mixture-of-Control: State-Aware Fine-Tuning for Transformer-based Models
Not on the map. Why?
Abstract (arXiv)
State-based fine-tuning has emerged as a compelling alternative to weight-based adaptation for transformers, updating lightweight controls into states rather than model weights, offering substantial memory savings while retaining parameter efficiency. However, most existing state-based methods typically apply only per-block control updates, which limits inter-block information exchange and restricts representational adaptation. Meanwhile, prior mechanisms that enable cross-block communication often introduce considerable computational overhead, reducing their practicality for efficient fine-tuning. We introduce Mixture-of-Control (MoC), a lightweight fine-tuning framework that adaptively integrates local and global control signals to enhance representation learning. MoC treats block-wise control states as experts in a sparse mixture-of-experts process, enabling efficient communication across transformer blocks. Empirical results across diverse transformer-based benchmarks demonstrate that MoC outperforms state-based methods while maintaining a comparable memory and computational efficiency.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 41 of its 53 references that are in the map. First posted on arXiv 30 Jun 2026.
Builds on
41 of 53 references.
References · 41
- 2017Attention Is All You Needcited by 59,932
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
29 more
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2022AdaptFormer: Adapting Vision Transformers for Scalable Visual Recognitioncited by 714
- 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 658
- 2021Compacter: Efficient Low-Rank Hypercomplex Adapter Layerscited by 458
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 391
- 2022Surgical Fine-Tuning Improves Adaptation to Distribution Shiftscited by 205
- 2017Beyond Finite Layer Neural Networks: Bridging Deep Architectures and Numerical Differential Equationscited by 203
- 2023Progressive Prompts: Continual Learning for Language Modelscited by 195
- 2024Mixture of LoRA Expertscited by 179
- 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 166
- 2019Understanding and Improving Transformer From a Multi-Particle Dynamic System Point of Viewcited by 108
- 2023The Expressive Power of Low-Rank Adaptationcited by 100
- 2024What Matters in Transformers? Not All Attention is Neededcited by 83
- 2017Maximum Principle Based Algorithms for Deep Learningcited by 77
- 2024A Closer Look into Mixture-of-Experts in Large Language Modelscited by 47
- 2019Deep learning as optimal control problems: models and numerical methodscited by 33
- 2023Brainformers: Trading Simplicity for Efficiencycited by 31
- 2020Universal Approximation Power of Deep Residual Neural Networks via Nonlinear Control Theorycited by 17
- 2023Less is More: Selective Layer Finetuning with SubTuningcited by 13
- 2024MomentumSMoE: Integrating Momentum into Sparse Mixture of Expertscited by 9
- 2025MoLEx: Mixture of Layer Experts for Finetuning with Sparse Upcyclingcited by 5
- 2025Selective Sinkhorn Routing for Improved Sparse Mixture of Expertscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Circuit Fine-Tuning for Compute-Efficient Transformer Adaptation
- 2023WeGeFT: Weight-Generative Fine-Tuning for Multi-Faceted Efficient Adaptation of Large Models
- 2025Mixture-of-Transformers Learn Faster: A Theoretical Study on Classification Problems
- 2024MambaPEFT: Exploring Parameter-Efficient Fine-Tuning for Mamba
- 2024Dynamic Mixture of Experts: An Auto-Tuning Approach for Efficient Transformer Models
- 2025Equivalence of Context and Parameter Updates in Modern Transformer Blocks
- 2026Local Linearity of LLMs Enables Activation Steering via Model-Based Linear Optimal Control
- 2024PERFT: Parameter-Efficient Routed Fine-Tuning for Mixture-of-Expert Model