arXiv:2505.06699cs.LGcs.AI2025-05ICML被引 2

用参考模型指导训练,提升模型泛化能力与数据效率。

Model Steering: Learning with a Reference Model Improves Generalization Bounds and Scaling Laws

  • 基于分布鲁棒优化构建理论框架,实现有依据的模型引导。
  • 实验验证新方法在小样本下表现更优,且扩展性优于传统CLIP。
  • 适合关注模型训练效率与泛化性能的研究者和工程师。

本文形式化了一种新兴的学习范式——模型引导(Model Steering),即利用已训练模型作为参考,通过策略性选择或加权数据来指导目标模型的训练。尽管此类方法已在大型基础模型训练中被广泛使用,但其基本原理仍缺乏深入理解,导致性能未达最优。本文提出一个理论驱动的模型引导框架——DRRho风险最小化,该框架基于分布鲁棒优化(DRO)。通过泛化分析,我们揭示了为何引入参考模型能提升泛化能力和数据效率。据我们所知,这是首次为这一学习范式提供理论解释,显著提升了对该技术的理解与实践水平。基于这些洞察,并结合对比学习与DRO的联系,我们提出了新的对比语言-图像预训练方法:DRRho-CLIP。大量实验验证了理论结论,揭示其相比无参考模型的CLIP具有更优的缩放规律,并优于现有启发式方法。

原文摘要 · Abstract (English)

This paper formalizes an emerging learning paradigm that uses a trained model as a reference to guide and enhance the training of a target model through strategic data selection or weighting, named $\textbf{model steering}$. While ad-hoc methods have been used in various contexts, including the training of large foundation models, its underlying principles remain insufficiently understood, leading to sub-optimal performance. In this work, we propose a theory-driven framework for model steering called $\textbf{DRRho risk minimization}$, which is rooted in Distributionally Robust Optimization (DRO). Through a generalization analysis, we provide theoretical insights into why this approach improves generalization and data efficiency compared to training without a reference model. To the best of our knowledge, this is the first time such theoretical insights are provided for the new learning paradigm, which significantly enhance our understanding and practice of model steering. Building on these insights and the connection between contrastive learning and DRO, we introduce a novel method for Contrastive Language-Image Pretraining (CLIP) with a reference model, termed DRRho-CLIP. Extensive experiments validate the theoretical insights, reveal a superior scaling law compared to CLIP without a reference model, and demonstrate its strength over existing heuristic approaches.

模型引导泛化能力对比学习理论分析

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