arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2604.14769cs.LG
收藏

基于约束的预训练:从结构化约束到可扩展的模型初始化

Constraint-based Pre-training: From Structured Constraints to Scalable Model Initialization

School of Computer Science and Engineering, Southeast University, Nanjing, China(东南大学计算机科学与工程学院,南京,中国);Key Laboratory of New Generation Artificial Intelligence Technology and Its Interdisciplinary Applications (Southeast University), Ministry of Education, China(新一代人工智能技术及其跨学科应用重点实验室(东南大学),教育部,中国)

Fu Feng, Yucheng Xie, Ruixiao Shi, Jing Wang, Xin Geng

AI总结:

本文提出基于约束的预训练方法,通过结构化约束解耦无尺寸依赖的知识,利用轻量级权重标量实现不同尺寸模型的初始化,提升模型适应性与效率。

摘要:

预训练和微调范式已成为模型适应的主要方法。然而,传统预训练通常生成固定规模的模型,而实际部署常需不同规模的模型,暴露了当目标模型规模与预训练时不同时的局限性。为此,我们提出一种创新的基于约束的预训练范式,通过在预训练过程中施加结构化约束,将无尺寸依赖的知识解耦为可重用的权重模板,并将尺寸特定的适应分配给轻量级权重标量,从而将可变尺寸模型初始化重新表述为多任务适应问题。在此范式中,我们进一步引入WeiT,采用基于Kronecker的约束来正则化预训练过程。具体而言,模型参数通过连接和加权聚合表示为权重模板的组合,其中适应性连接由轻量级权重标量控制,其参数从有限数据中学习。这种设计使模型权重在不同下游规模下能够灵活高效地构建。大量实验展示了WeiT的效率和有效性,在广泛感知和具身学习任务中,如图像分类、图像生成和具身控制中,实现了初始化不同深度和宽度模型的最先进性能。此外,其有效性扩展到基于Transformer和卷积的架构,在完全训练下也能实现更快的收敛和改进的性能。

英文摘要:

The pre-training and fine-tuning paradigm has become the dominant approach for model adaptation. However, conventional pre-training typically yields models at a fixed scale, whereas practical deployment often requires models of varying sizes, exposing its limitations when target model scales differ from those used during pre-training. To address this, we propose an innovative constraint-based pre-training paradigm that imposes structured constraints during pre-training to disentangle size-agnostic knowledge into reusable weight templates, while assigning size-specific adaptation to lightweight weight scalers, thereby reformulating variable-sized model initialization as a multi-task adaptation problem. Within this paradigm, we further introduce WeiT, which employs Kronecker-based constraints to regularize the pre-training process. Specifically, model parameters are represented as compositions of weight templates via concatenation and weighted aggregation, with adaptive connections governed by lightweight weight scalers whose parameters are learned from limited data. This design enables flexible and efficient construction of model weights across diverse downstream scales. Extensive experiments demonstrate the efficiency and effectiveness of WeiT, achieving state-of-the-art performance in initializing models with varying depths and widths across a broad range of perception and embodied learning tasks, including Image Classification, Image Generation, and Embodied Control. Moreover, its effectiveness generalizes to both Transformer-based and Convolution-based architectures, consistently enabling faster convergence and improved performance even under full training.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑