arXiv:2506.04567cs.LGcs.CV2025-06被引 1

用权重统计信息指导模型融合,无需标签也能高效合并多模型。

StatsMerging: Statistics-Guided Model Merging via Task-Specific Teacher Distillation

  • 通过SVD奇异值捕捉任务相关权重分布,作为任务重要性依据。
  • 在8个任务上优于现有方法,对未见任务泛化能力强,抗图像质量波动。
  • 适合资源受限场景下的多模型融合,尤其无标签数据时应用价值高。

模型融合已成为在有限内存预算下部署多个大模型的有前景方案。本文提出StatsMerging,一种基于权重分布统计的轻量级学习型模型融合方法,无需真实标签或测试样本。该方法具有三大优势:(1) 创新性地利用奇异值分解(SVD)的奇异值捕捉任务特定的权重分布,作为任务重要性的代理信号以指导任务系数预测;(2) 引入轻量级学习器StatsMergeLearner建模各任务预训练模型的权重分布,提升泛化能力并增强对未见样本的适应性;(3) 提出任务特异性教师蒸馏(Task-Specific Teacher Distillation),用于异构架构视觉模型的融合,通过任务特异性教师蒸馏避免昂贵的真实标签需求。我们设计了两种知识蒸馏方式:(a) 从任务特异性模型蒸馏知识至StatsMergeLearner;(b) 在融合前从异构模型蒸馏知识。在八个任务上的大量实验表明,StatsMerging在整体准确率、对未见任务的泛化能力以及对图像质量变化的鲁棒性方面均超越当前最优技术。

原文摘要 · Abstract (English)

Model merging has emerged as a promising solution to accommodate multiple large models within constrained memory budgets. We present StatsMerging, a novel lightweight learning-based model merging method guided by weight distribution statistics without requiring ground truth labels or test samples. StatsMerging offers three key advantages: (1) It uniquely leverages singular values from singular value decomposition (SVD) to capture task-specific weight distributions, serving as a proxy for task importance to guide task coefficient prediction; (2) It employs a lightweight learner StatsMergeLearner to model the weight distributions of task-specific pre-trained models, improving generalization and enhancing adaptation to unseen samples; (3) It introduces Task-Specific Teacher Distillation for merging vision models with heterogeneous architectures, a merging learning paradigm that avoids costly ground-truth labels by task-specific teacher distillation. Notably, we present two types of knowledge distillation, (a) distilling knowledge from task-specific models to StatsMergeLearner; and (b) distilling knowledge from models with heterogeneous architectures prior to merging. Extensive experiments across eight tasks demonstrate the effectiveness of StatsMerging. Our results show that StatsMerging outperforms state-of-the-art techniques in terms of overall accuracy, generalization to unseen tasks, and robustness to image quality variations.

模型融合知识蒸馏轻量化

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