arXiv:2604.04231cs.LG2026-04被引 2

提出SIFT框架,解决模型优化中目标与约束的冲突问题。

Subspace Control: Turning Constrained Model Steering into Controllable Spectral Optimization

  • 通过正交化子空间,消除任务间谱干扰
  • 在4个任务上均显著提升性能,优于基线方法
  • 适合需要安全、隐私或定制化的模型部署场景

基础模型(如大语言模型)虽强大,但部署前常需满足安全、隐私和任务特定要求,形成约束优化问题。然而此类问题研究不足,因主目标与约束目标在优化中相互干扰而极具挑战。本文提出子空间控制框架:首先从模型融合视角分析谱交叉干扰成因,提出一次性正交化合并子空间的解决方案;其次揭示该方案与谱优化器Muon中梯度正交化的关联;进而构建SIFT(无谱干扰训练)方法,利用定位机制选择性干预优化过程,实现可控更新以缓解目标-约束冲突。在机器遗忘、安全对齐、文本转语音适配和幻觉抑制四个代表性任务上评估,SIFT相比基于控制和无控制的基线均取得显著且稳健的性能提升。代码已开源。

原文摘要 · Abstract (English)

Foundation models, such as large language models (LLMs), are powerful but often require customization before deployment to satisfy practical constraints such as safety, privacy, and task-specific requirements, leading to "constrained" optimization problems for model steering and adaptation. However, solving such problems remains largely underexplored and is particularly challenging due to interference between the primary objective and constraint objectives during optimization. In this paper, we propose a subspace control framework for constrained model training. Specifically, (i) we first analyze, from a model merging perspective, how spectral cross-task interference arises and show that it can be resolved via a one-shot solution that orthogonalizes the merged subspace; (ii) we establish a connection between this solution and gradient orthogonalization in the spectral optimizer Muon; and (iii) building on these insights, we introduce SIFT (spectral interference-free training), which leverages a localization scheme to selectively intervene during optimization, enabling controllable updates that mitigate objective-constraint conflicts. We evaluate SIFT across four representative applications: (a) machine unlearning, (b) safety alignment, (c) text-to-speech adaptation, and (d) hallucination mitigation. Compared to both control-based and control-free baselines, SIFT consistently achieves substantial and robust performance improvements across all tasks. Code is available at https://github.com/OPTML-Group/SIFT.

模型优化约束学习谱方法LLM对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。