arXiv:2601.19694cs.CVcs.LG2026-01被引 1

用可扩展权重模板,让视觉模型轻松适配不同大小。

Self-Supervised Weight Templates for Scalable Vision Model Initialization

  • 学习共享权重模板+轻量缩放器,实现灵活架构初始化。
  • 在多任务上优于现有方法,支持深度宽度任意调整。
  • 适合需要快速部署多种规模模型的工程场景。

现代模型参数规模日益增大,预训练模型的重要性凸显。但实际部署常需不同尺寸的架构,传统预训练与微调面临局限。为此,我们提出SWEET,一种基于约束的自监督框架,通过Tucker分解学习共享权重模板和尺寸相关缩放因子,实现视觉任务中可扩展的初始化。不固定模型大小,而是通过轻量级缩放器组合模板进行初始化,其参数可从极少数据中高效学习。为增强宽度扩展灵活性,引入沿宽度维度的随机缩放机制,正则化模板并促进鲁棒的、与宽度无关的表示,提升跨宽度泛化能力。在分类、检测、分割和生成任务上的大量实验表明,SWEET在初始化可变规模视觉模型方面达到当前最优性能。

原文摘要 · Abstract (English)

The increasing scale and complexity of modern model parameters underscore the importance of pre-trained models. However, deployment often demands architectures of varying sizes, exposing limitations of conventional pre-training and fine-tuning. To address this, we propose SWEET, a self-supervised framework that performs constraint-based pre-training to enable scalable initialization in vision tasks. Instead of pre-training a fixed-size model, we learn a shared weight template and size-specific weight scalers under Tucker-based factorization, which promotes modularity and supports flexible adaptation to architectures with varying depths and widths. Target models are subsequently initialized by composing and reweighting the template through lightweight weight scalers, whose parameters can be efficiently learned from minimal training data. To further enhance flexibility in width expansion, we introduce width-wise stochastic scaling, which regularizes the template along width-related dimensions and encourages robust, width-invariant representations for improved cross-width generalization. Extensive experiments on \textsc{classification}, \textsc{detection}, \textsc{segmentation} and \textsc{generation} tasks demonstrate the state-of-the-art performance of SWEET for initializing variable-sized vision models.

视觉模型可扩展初始化自监督权重模板

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