arXiv:2604.12391cs.CVcs.AI2026-04中稿 · CVPR

通过模型链顺序迁移知识,让视觉大模型训练提速超7倍且不丢性能。

Chain-of-Models Pre-Training: Rethinking Training Acceleration of Vision Foundation Models

论文配图:Chain-of-Models Pre-Training: Rethinking Training Acceleration of Vision Foundation Models
图 1 · 摘自论文原文
  • 构建从小到大的模型链,后继模型从前置小模型迁移知识。
  • 在45个数据集上验证,大模型训练成本降低72%,加速比达7.09倍。
  • 适合大规模视觉模型预训练,尤其适合资源有限的团队使用。

本文提出链式模型预训练(CoM-PT),一种无需牺牲性能的视觉基础模型训练加速方法。该方法不单独优化每个模型,而是以模型家族为单位构建按大小递增排列的模型链:仅最小模型进行标准预训练,其余模型通过从其前序较小模型中联合复用参数空间和特征空间的知识实现高效训练。实验表明,所有模型性能均优于传统独立训练,同时大幅降低训练开销,在45个数据集上的零样本与微调任务中得到充分验证。值得注意的是,模型家族越大,效率越高:在CC3M数据集上,当最大模型为ViT-L时,逐步加入更小模型使计算复杂度降低72%;固定规模下,模型数量从3增至7,加速比从4.13X提升至7.09X。CoM-PT对预训练范式无特定依赖,已开源代码,可推广至大语言模型等更复杂的训练场景。

原文摘要 · Abstract (English)

In this paper, we present Chain-of-Models Pre-Training (CoM-PT), a novel performance-lossless training acceleration method for vision foundation models (VFMs). This approach fundamentally differs from existing acceleration methods in its core motivation: rather than optimizing each model individually, CoM-PT is designed to accelerate the training pipeline at the model family level, scaling efficiently as the model family expands. Specifically, CoM-PT establishes a pre-training sequence for the model family, arranged in ascending order of model size, called model chain. In this chain, only the smallest model undergoes standard individual pre-training, while the other models are efficiently trained through sequential inverse knowledge transfer from their smaller predecessors by jointly reusing the knowledge in the parameter space and the feature space. As a result, CoM-PT enables all models to achieve performance that is mostly superior to standard individual training while significantly reducing training cost, and this is extensively validated across 45 datasets spanning zero-shot and fine-tuning tasks. Notably, its efficient scaling property yields a remarkable phenomenon: training more models even results in higher efficiency. For instance, when pre-training on CC3M: i) given ViT-L as the largest model, progressively prepending smaller models to the model chain reduces computational complexity by up to 72%; ii) within a fixed model size range, as the VFM family scales across 3, 4, and 7 models, the acceleration ratio of CoM-PT exhibits a striking leap: from 4.13X to 5.68X and 7.09X. Since CoM-PT is naturally agnostic to specific pre-training paradigms, we open-source the code to spur further extensions in more computationally intensive scenarios, such as large language model pre-training.

视觉模型训练加速知识迁移模型链

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