arXiv:2606.07289cs.LGcs.CV2026-06

提出闭式谱正则化方法,让多任务模型合并更快更省显存。

Closed-Form Spectral Regularization for Multi-Task Model Merging

论文配图:Closed-Form Spectral Regularization for Multi-Task Model Merging
图 1 · 摘自论文原文
  • 将模型合并建模为带噪声的线性逆问题,用谱滤波隐式正则化。
  • 新方法在4个通用基准和跨模态任务上媲美或超越现有方法。
  • 无需训练数据,显存降低50%,速度提升28-72倍,适合部署优化。

模型合并将多个独立微调的专家模型融合为单个多任务模型,无需训练数据,可降低大模型的存储、推理和服务成本。现有方法将合并建模为逐层二次干扰最小化问题,虽有闭式伪逆解,但实际性能远逊于数百轮梯度下降。我们发现,迭代求解器并非主要作为优化器,而是对病态正规方程施加隐式谱正则化,其中小特征值方向会放大代理噪声。基于此,我们将多任务模型合并形式化为含噪线性逆问题,并提出谱滤波估计器,参数为每方向滤波函数。我们实例化该估计器为SWUDI,结合软指数滤波(匹配梯度流轨迹)与硬Top-K截断(抑制噪声放大方向)。此外,提出SWUDI-A,以逐层秩规则替代全局秩超参,提升跨架构鲁棒性。两者均只需每层一次对称特征分解,无需训练数据或优化器状态。在四个通用基准及涵盖VQA、几何、图表、OCR、定位和模态融合的多模态基准上,新方法匹配或超越当前最优。关键优势在于壁钟时间减少28-72倍,峰值显存降低最高50%。

原文摘要 · Abstract (English)

Model merging combines several independently fine-tuned experts into a single multi-task model without any training data, reducing the storage, serving, and decentralized-development costs of large foundation models. State-of-the-art merging methods formulate merging as a layer-wise quadratic interference minimization problem. Although this problem admits an exact closed-form pseudoinverse solution, that solution underperforms hundreds of iterations of gradient descent in practice. The iterative loop dominates the cost of the pipeline, yet its effectiveness has remained unexplained. We revisit this regime and show that the iterative solver does not primarily act as an optimizer; rather, it serves as an implicit spectral regularizer for an ill-posed normal equation, where small-eigenvalue directions of the per-layer interference operator amplify proxy noise. Building on this finding, we formalize multi-task model merging as a noisy linear inverse problem and propose a spectral filtering estimator parameterized by a per-direction filter. We instantiate this estimator with SWUDI, a closed-form method that combines a soft exponential filter, which matches the gradient-flow trajectory of iterative descent, with a hard top-K truncation that suppresses noise-amplifying small-eigenvalue directions. Furthermore, we propose SWUDI-A, an adaptive variant that replaces the global rank hyperparameter with per-layer rank rules, further improving robustness across architectures. Both variants share a single symmetric eigendecomposition per linear layer and require no training data or optimizer state. Across four general benchmarks and a multimodal merging benchmark spanning VQA, Geometry, Chart, OCR, Grounding, and modality merging, our proposed spectral solvers match or outperform state-of-the-art merging methods. Crucially, they reduce wall-clock time by 28-72x and peak GPU memory by up to 50%.

模型合并谱正则化高效推理多任务学习

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