首页 > AI前沿 > DuplexTune: Coupled-State Replay for Memory-Efficient Mixture-of-Experts Adaptation

DuplexTune: Coupled-State Replay for Memory-Efficient Mixture-of-Experts Adaptation

arXiv机器学习 2026-10-27 00:17 3 阅读 查看原文

Sparse expert execution does not remove the stored states needed to adapt a mixture-of-experts language model.

We present DuplexTune, a framework for replaying coupled hidden states through pretrained attention and expert modules.

Trainable width bridges connect two residual streams to the backbone.

Bridge warm-up precedes backbone adaptation, with router weights kept frozen.

We analyze the implicit state-recovery equation and identify the convergence and replay conditions needed for reconstruction.

On Qwen1.5-MoE-A2.7B, the single-H800 evaluation reports 38.8 GB peak memory, versus 64.8 GB for supervised fine-tuning with checkpointing, and throughput of 25.1 versus 19.9 samples/s.

DuplexTune achieves 67.2% MMLU, 75.6% GSM8K, and 7.69 MT-Bench, with a weaker multilingual score than the checkpointed baseline.

The results characterize a configuration-dependent tradeoff among state storage, throughput, and adaptation quality.