跳到正文
arXiv cs.AI· Duc Anh Nguyen, Tien Ngoc Luu, Tung Pham, Toan Tran·· 12 小时前AI 评分29

Mixture-of-Control:面向 Transformer 的状态感知微调方法

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

AI 导读

研究者提出 Mixture-of-Control(MoC),将逐块控制状态视为稀疏 MoE 中的专家,实现 Transformer 各层间的跨层通信,自适应融合局部与全局控制信号。在 33 个数据集和 10 个预训练模型上的实验显示,MoC 持续优于现有基于状态的微调方法,且内存与计算效率相当。该工作已被 NeurIPS 2026 主会接收。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:基于状态的微调已成为基于权重的 Transformer 适配的一种有吸引力的替代方案,它将轻量控制更新为状态而非模型权重,在保持参数效率的同时大幅节省内存。然而,现有大多数基于状态的方法通常只进行逐块控制更新,这限制了块间信息交换并制约了表示适配。同时,此前实现跨块通信的机制往往引入相当大的计算开销,降低了其用于高效微调的实用性。我们提出控制混合(Mixture-of-Control,MoC),一种高效的跨层通信方法,可自适应地整合局部与全局控制信号,以增强适配过程中的表示学习。MoC 将逐块控制状态视为稀疏专家混合过程中的专家,从而实现 Transformer 各块之间的高效通信。在 33 个数据集和 10 个预训练模型上、跨多种基于 Transformer 的基准进行的实验表明,MoC 始终优于现有的基于状态的微调方法,同时保持相当的内存与计算效率。
评论: Advances in Neural Information Processing Systems (NeurIPS 2026 Main Conference),38 页,13 幅图,13 张表
主题: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2606.31397 [cs.LG]
  (或 arXiv:2606.31397v2 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2606.31397

由 DataCite 颁发的 arXiv DOI

期刊 参考: NeurIPS 2026

提交历史

来自:Duc-Anh Nguyen [查看电子邮件]
[v1] 2026 年 6 月 30 日星期二 09:25:37 UTC(745 KB)
[v2] 2026 年 10 月 21 日星期三 03:11:53 UTC(722 KB)

来源:arXiv cs.AI · arxiv.org