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

arXiv:2606.31397 · cs.LG, cs.AI · Submitted 2026-06-30 · Read on arXiv

cs.LG, cs.AI

Submitted: 2026-06-30

Updated: 2026-10-21

Comments: Advances in Neural Information Processing Systems (NeurIPS 2026 Main Conference), 38 pages, 13 figures, 13 tables

Journal ref: NeurIPS 2026

License: http://creativecommons.org/licenses/by/4.0/

The gist: 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

Terminology

Abstract

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), an efficient cross-layer communication that adaptively integrates local and global control signals to enhance representation learning during adaptation. MoC treats block-wise control states as experts in a sparse mixture-of-experts process, enabling efficient communication across transformer blocks. Experiments on 33 datasets and 10 pretrained models across diverse transformer-based benchmarks demonstrate that MoC consistently outperforms existing state-based fine-tuning methods while maintaining comparable efficiency in memory and compute.

Related papers