用小波变换统一大模型转小模型和小模型转大模型的知识迁移。
A Unified Framework for Knowledge Transfer in Bidirectional Model Scaling
- 将模型权重视为连续信号,用小波变换实现大小模型间的双向知识迁移。
- 在DeiT/BERT/GPT上实现最高67.1%的预训练计算量节省,性能达顶尖水平。
- 无需参数调整,适用于不同架构间高效、通用的知识转移,适合模型压缩与扩展场景。
将预训练知识从源模型迁移至不同规模的目标模型是实现灵活高效模型缩放的关键挑战。然而,现有参数空间方法将小模型到大模型(S2L)与大模型到小模型(L2S)视为相互独立的问题,分别侧重于参数合成与选择,导致工具碎片化,难以形成统一的双向框架。本文提出BoT(双向知识迁移),首个无尺寸依赖的统一框架,核心思想是将模型权重视为连续信号,不同规模模型即为知识的不同离散化表示。这一多分辨率视角将S2L与L2S分别建模为上采样与下采样操作,自然引入离散小波变换(DWT)及其逆变换(IDWT)。BoT利用小波的递归特性,以分解层级作为动态缩放因子,实现参数无须调整且计算高效的跨规模知识迁移。在DeiT、BERT和GPT上的大量实验表明,该方法在S2L中最多节省67.1%预训练浮点运算量,在L2S中节省52.8%,并在GLUE和SQuAD等基准测试中达到业界领先性能。
原文摘要 · Abstract (English)
Transferring pre-trained knowledge from a source model to a target model of a different architectural size is a key challenge for flexible and efficient model scaling. However, current parameter-space methods treat Small-to-Large (S2L) and Large-to-Small (L2S) scaling as separate, incompatible problems, focusing on parameter synthesis and selection, respectively. This fragmented perspective has resulted in specialized tools, hindering a unified, bidirectional framework. In this paper, we propose BoT (Bidirectional knowledge Transfer), the first size-agnostic framework to unify S2L and L2S scaling. Our core insight is to treat model weights as continuous signals, where models of different sizes represent distinct discretizations of the transferable knowledge. This multi-resolution perspective directly casts S2L and L2S scaling as the signal processing operations of upsampling and downsampling, naturally leading to the adoption of the Discrete Wavelet Transform (DWT) and its Inverse (IDWT). BoT leverages the recursive nature of wavelets, using the decomposition level as a dynamic scaling factor to bridge disparate model sizes in a parameter-free and computationally efficient manner. Extensive experiments on DeiT, BERT, and GPT demonstrate significant pre-training FLOPs savings (up to 67.1% for S2L, 52.8% for L2S) and state-of-the-art performance on benchmarks like GLUE and SQuAD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。