mlsanityData to 7 Oct 2026

Mixture-of-Control: State-Aware Fine-Tuning for Transformer-based Models

arXiv 2606.31397 ↗First posted 30 Jun 2026cs.LG cs.AIcited by 0

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2024The Llama 3 Herd of Modelscited by 14,192
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  5. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
  6. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  7. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
  8. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  9. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  10. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  11. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
  12. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
29 more
  1. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  2. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  3. 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
  4. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
  5. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
  6. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
  7. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
  8. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  9. 2022AdaptFormer: Adapting Vision Transformers for Scalable Visual Recognitioncited by 714
  10. 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 658
  11. 2021Compacter: Efficient Low-Rank Hypercomplex Adapter Layerscited by 458
  12. 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 391
  13. 2022Surgical Fine-Tuning Improves Adaptation to Distribution Shiftscited by 205
  14. 2017Beyond Finite Layer Neural Networks: Bridging Deep Architectures and Numerical Differential Equationscited by 203
  15. 2023Progressive Prompts: Continual Learning for Language Modelscited by 195
  16. 2024Mixture of LoRA Expertscited by 179
  17. 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 166
  18. 2019Understanding and Improving Transformer From a Multi-Particle Dynamic System Point of Viewcited by 108
  19. 2023The Expressive Power of Low-Rank Adaptationcited by 100
  20. 2024What Matters in Transformers? Not All Attention is Neededcited by 83
  21. 2017Maximum Principle Based Algorithms for Deep Learningcited by 77
  22. 2024A Closer Look into Mixture-of-Experts in Large Language Modelscited by 47
  23. 2019Deep learning as optimal control problems: models and numerical methodscited by 33
  24. 2023Brainformers: Trading Simplicity for Efficiencycited by 31
  25. 2020Universal Approximation Power of Deep Residual Neural Networks via Nonlinear Control Theorycited by 17
  26. 2023Less is More: Selective Layer Finetuning with SubTuningcited by 13
  27. 2024MomentumSMoE: Integrating Momentum into Sparse Mixture of Expertscited by 9
  28. 2025MoLEx: Mixture of Layer Experts for Finetuning with Sparse Upcyclingcited by 5
  29. 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

  1. 2026Circuit Fine-Tuning for Compute-Efficient Transformer Adaptation
  2. 2023WeGeFT: Weight-Generative Fine-Tuning for Multi-Faceted Efficient Adaptation of Large Models
  3. 2025Mixture-of-Transformers Learn Faster: A Theoretical Study on Classification Problems
  4. 2024MambaPEFT: Exploring Parameter-Efficient Fine-Tuning for Mamba
  5. 2024Dynamic Mixture of Experts: An Auto-Tuning Approach for Efficient Transformer Models
  6. 2025Equivalence of Context and Parameter Updates in Modern Transformer Blocks
  7. 2026Local Linearity of LLMs Enables Activation Steering via Model-Based Linear Optimal Control
  8. 2024PERFT: Parameter-Efficient Routed Fine-Tuning for Mixture-of-Expert Model