基于下游反馈的价值预训练:通过任务构建引导持续预训练 Value-Based Pre-Training with Downstream Feedback
用下游信号训练任务设计器,间接引导自监督预训练更新方向
前置知识
持续预训练 (Continued Pretraining)
在已经预训练好的基础模型上,使用新的无标注数据继续进行自监督训练,以提升模型在特定领域或能力上的表现。典型场景包括:让语言模型在数学文本上继续训练以提升推理能力,或让视觉骨干网络在自监督目标下继续训练以改善密集预测。持续预训练的训练信号通常是固定的代理任务,例如语言建模中的下一词预测或视觉中的重建/视图目标。虽然这些代理任务不需要密集人工标注且可扩展,但模型的实际性能最终由下游任务决定,而非代理损失本身。
本文的核心贡献正是改进持续预训练的范式,理解其标准流程和存在的问题是读懂本文的前提。
自监督学习 (Self-Supervised Learning, SSL)
一种不依赖人工标注的机器学习范式,通过设计前置任务(pretext task)从无标注数据本身构造监督信号。在语言模型中,典型的自监督任务是下一个词预测(next-token prediction);在视觉中,DINO系列方法通过让学生网络预测教师网络在不同视图下的表示来学习视觉特征,I-JEPA则通过掩码预测实现自监督。自监督学习的核心优势在于可以利用海量无标注数据训练强大的表征模型。
本文方法保持自监督学习的核心框架不变,但通过引入下游反馈来动态调整自监督任务的构建方式,因此理解自监督学习的基本原理是理解本文方法的基础。
元学习与双层优化 (Meta-Learning and Bilevel Optimization)
元学习是'学习如何学习'的范式,其中内层循环执行标准学习过程,外层循环优化学习过程本身的超参数或策略。双层优化是元学习的数学形式化:上层目标依赖于下层优化问题的解。在本文中,内层是自监督预训练更新模型参数 $\theta$,外层通过下游反馈训练任务设计器参数 $\phi$。传统的双层优化需要对完整训练轨迹求导,计算代价极高,本文通过一阶局部近似规避了这一问题。
本文的技术框架本质上是一个双层优化问题的高效近似求解,理解元学习和双层优化的概念有助于把握方法的理论基础和设计动机。
梯度对齐 (Gradient Alignment)
梯度对齐指的是两个梯度向量的内积 $g_{down}^T g_{pre}$ 的符号和大小。当内积为正时,意味着预训练梯度方向与下游损失梯度方向存在正向投影,即预训练更新会同时降低下游损失;当内积为负时,预训练更新可能损害下游性能。这个内积衡量了'预训练的这一步对下游任务是否有价值'。在本文中,梯度对齐被形式化为一个价值函数 $V(\phi)$,用于训练任务设计器使其构造的预训练任务能产生与下游目标对齐的梯度方向。
梯度对齐是本文价值目标的数学核心,$V = g_{down}^T g_{pre}$ 是连接下游反馈与预训练任务构建的桥梁,不理解这个概念就无法理解V-pretraining的工作原理。
DINO与视觉自监督学习
DINO(Self-Distillation with No Labels)是一种视觉自监督学习方法,采用学生-教师框架:学生网络通过视图预测任务学习,教师网络通过指数移动平均(EMA)更新。DINOv3是其最新版本,通过自监督预训练在ImageNet上学习通用视觉表征。在DINO框架中,'任务构建'指的是对输入图像进行裁剪、增强和掩码操作生成不同视图,学生网络从一个视图预测另一个视图中教师网络的输出表征。本文在DINO框架下,将固定的视图生成策略替换为由下游反馈训练的自适应视图构建。
本文在视觉模态的实例化基于DINO系列框架,理解DINO的学生-教师架构和视图构建机制是理解本文视觉实验的关键。
一阶近似与Taylor展开
一阶近似是利用Taylor展开的线性项来近似复杂函数的方法。对于函数 $f(x)$ 在点 $x_0$ 附近的Taylor展开:$f(x) \approx f(x_0) + \nabla f(x_0)^T (x - x_0) + O(\|x-x_0\|^2)$。在本文中,作者将下游损失 $L_{down}$ 在当前参数 $\theta$ 处进行一阶展开:$L_{down}(\theta - \eta g_{pre}) \approx L_{down}(\theta) - \eta g_{down}^T g_{pre}$,从而将复杂的长时域优化问题转化为每一步可计算的局部目标。这个近似在学习率 $\eta$ 足够小或预训练更新范数受控时成立。
一阶近似是本文从不可行的长时域双层优化问题推导出可扩展的局部价值目标的关键数学工具,是整个方法理论基础的核心。
研究动机
持续预训练的标准流程存在一个根本性的反馈延迟问题。当前实践中,持续预训练使用固定的自监督代理任务(如语言建模中的下一词预测、视觉中的DINO视图预测),但模型的实际价值由下游任务性能衡量。这意味着优化发生在每一步自监督更新上,而选择却发生在整个预训练完成之后。具体来说,从业者需要:(1) 运行一次完整的持续预训练,(2) 在检查点上评估下游性能,(3) 根据结果修改数据混合、目标函数、课程学习或增强策略,(4) 启动新一轮训练。这使得任务配方设计成为对完整预训练轨迹的昂贵黑盒搜索。例如,在语言模型微调中,一个标准的持续预训练可能需要数十到数百GPU小时,而每次反馈循环都要承担这样的全部成本。这种'粗粒度外循环'的问题在于:单个预训练步骤对下游能力的贡献是不可见的,模型无法在训练过程中获得'这一步是否有用'的信号。
本文的目标是本文的核心目标是回答一个具体问题:给定一个预训练好的基础模型、一个用于持续预训练的无标注数据流、以及一小组可验证的下游示例(例如1,024个GSM8K数学题),能否在持续预训练过程中利用这些反馈来提升目标能力,同时在有限计算预算下保持预训练模型的泛化能力?这个目标的关键约束是:(1) 反馈信号不能直接用于更新学习者参数,即学习者更新必须保持为自监督更新;(2) 计算开销必须在相同训练时间内消化,不能因为引入反馈机制而额外增加总训练时间;(3) 方法需要在语言和视觉两种模态上通用。
与已有工作不同的是,现有方法与本文的切入角度存在本质区别。SFT(监督微调)、偏好优化和RL方法虽然也使用下游示例,但它们将下游信号作为学习者的直接监督——标签、偏好或奖励直接定义学习者更新。这些方法回答的是'如何用下游监督更新学习者',而非'哪个无标注自监督预训练更新值得在训练结束前执行'。本文的独特视角是将下游示例视为对候选预训练更新的价值信号,而非学习者本身的监督信号。具体而言,本文将'任务构建规则'(task construction rule)作为持续预训练的控制面:标准方法在训练前固定这个规则(如one-hot下一词预测),而本文让一个轻量级的任务设计器根据下游反馈动态调整这个规则。这意味着改变的是自监督任务本身(目标分布、视图选择),而非学习者架构、优化器、数据流或评估协议。这种'间接反馈'通道是本文与所有现有方法的本质区别。
核心方法
V-pretraining的核心直觉可以用一个类比来理解:想象一位教练(任务设计器)和一位运动员(学习者)。运动员按照教练安排的训练计划进行自监督训练,但教练能观察到运动员在比赛(下游任务)中的表现。教练不会直接告诉运动员如何比赛(不用下游梯度更新运动员),而是根据比赛反馈调整训练计划的内容(构造不同的自监督任务),使得训练产生的体能提升能更好地转化为比赛成绩。技术上,V-pretraining在标准持续预训练框架中引入两个角色:学习者(参数 $\theta$)只接受自监督损失更新;任务设计器(参数 $\phi$)动态控制自监督任务的构建方式,使用下游反馈信号训练。对于每个无标注批次,设计器生成候选任务(如语言中的软目标分布或视觉中的视图配置),估计该任务诱导的学习者梯度对下游损失的影响,然后调整自身参数以最大化这个影响。之后,设计器构造的任务被分离(detach),学习者用标准自监督损失更新。整个过程中,下游标签从未直接更新学习者参数。
V-pretraining的核心创新点在于将不可行的长时域双层优化问题转化为每一步可高效计算的局部价值目标。理想的设计器目标是最小化下游损失 $L_{down}(\theta_T(\phi))$,即选择那些能让完整训练轨迹在下游表现最好的任务构建规则,但这需要对整个预训练轨迹求导,在大规模预训练中计算上不可行。本文的关键洞察是:对于足够小的学习率 $\eta > 0$,如果候选预训练梯度为 $g_{pre}(\theta; \phi)$,下游反馈梯度为 $g_{down}(\theta)$,则一阶Taylor展开给出 $L_{down}(\theta - \eta g_{pre}) \approx L_{down}(\theta) - \eta g_{down}^T g_{pre}$。因此,内积 $V(\phi; \theta, B_u, B_f) = g_{down}^T g_{pre}$ 估计了无标注更新是否会降低下游损失。这个价值函数 $V$ 既有理论保证(Proposition 2.1证明了它最大化一步下游改进的下界),又可高效计算(只需两个梯度的内积)。与已有的元学习方法不同,本文不要求对完整轨迹求导;与知识蒸馏方法不同,本文的目标不是让学习者模仿某个教师,而是让学习者的更新方向与下游目标对齐。梯度内积仅在参数子集 $S$(如最后几个Transformer块或任务相关投影层)上计算以提高效率。
方法步骤详情
V-pretraining的完整算法包含以下步骤。第一步:初始化学习者参数 $\theta$(从预训练检查点加载)和任务设计器参数 $\phi$。第二步:从无标注数据流中采样批次 $B_u \subset D_u$,从下游反馈集采样反馈批次 $B_f \subset D_{fb}$。第三步:任务设计器 $c_\phi$ 根据当前学习者状态和无标注批次构造候选自监督任务。在语言模态中,设计器生成自适应top-K软目标:候选集 $C_t = \{w_t\} \cup \text{TopK}_{K-1}(p_\theta(\cdot|w_{<t})) \setminus \{w_t\}$,输出分布 $r_{\phi,t} \in \Delta(C_t)$ 和门控常数 $\alpha_{\phi,t} \in [0, \alpha_{max}]$,最终目标 $q_{\phi,t}(v) = (1-\alpha_{\phi,t})\mathbf{1}[v=w_t] + \alpha_{\phi,t} r_{\phi,t}(v)$。在视觉模态中,设计器修改DINO的视图生成过程,通过实例级掩码或视图参数产生 $V_\phi(x)$。第四步:计算候选预训练梯度 $g_{pre} = \nabla_\theta L_{pre}(\theta; c_\phi(B_u))$ 和下游评估梯度 $g_{down} = \nabla_\theta L_{down}(\theta; B_f)$。第五步:计算价值 $V = g_{down}^T g_{pre}$,用 $L_{meta}(\phi) = -V$ 更新设计器参数 $\phi$。第六步:分离(detach)设计器构造的任务目标,让学习者用标准自监督损失应用梯度下降更新 $\theta_{t+1} = \theta_t - \eta_t g_{learn}$。关键约束是:$g_{learn}$ 中不包含任何形式的 $\nabla_\theta L_{down}(\theta_t; D_{fb})$ 项,下游标签从不直接更新学习者参数。
技术新颖性
V-pretraining在技术上有几个显著的新颖性。首先,它提出了'任务构建作为控制面'这一新概念:不是改变学习者的目标函数或架构,而是改变自监督任务本身的构造方式。这开辟了一条新的引导通道——间接反馈,既不同于直接微调(改变学习者更新),也不同于课程学习(改变数据顺序)。其次,价值目标 $V = g_{down}^T g_{pre}$ 的推导优雅地将不可行的长时域优化转化为可在线计算的一步信号,这在理论上有一阶下界保证(Proposition 2.1),在实践中只需计算两个梯度的内积。第三,同一框架在语言和视觉两种截然不同的模态上都有自然的实例化:语言中改变目标分布(软目标),视觉中改变视图构建(掩码/裁剪),但核心价值目标相同。第四,该方法与标准自监督训练完全兼容,不改变学习者架构、优化器或数据流,只在任务构建层插入一个轻量级设计器。最后,与传统元学习方法相比,V-pretraining不需要对完整训练轨迹进行反向传播,避免了大规模预训练中的内存和计算瓶颈。
实验结果
本文在严格的墙钟时间匹配条件下进行了广泛的实验,主要发现如下。在语言模态中,V-pretraining在所有测试的Qwen模型上都提升了GSM8K Pass@1:Qwen2.5-0.5B从22.20提升到29.60(+7.4分,33%相对提升),这是最大的单次运行增益;Qwen1.5-0.5B从19.15±1.16提升到22.67±1.05;Qwen1.5-4B从56.48±1.56提升到58.98±1.03;Qwen1.5-7B从65.26±1.06提升到66.17±0.63。增益随模型规模增大而减小,表明较大学习者本身就能从标准下一词预测中提取更多数学相关信号。在视觉模态中,V-pretraining在保持全局识别能力的同时改善了密集预测:DINOv3-ViT-L的ADE20K mIoU从51.33提升到52.47,NYUv2 RMSE从0.5752降低到0.5522,ImageNet线性准确率从84.07提升到84.59。值得注意的是,ImageNet线性准确率是未作为反馈的任务,但也有小幅提升,说明反馈引导没有导致表征坍塌。消融实验证实了价值信号的必要性:随机反馈(54.31)、均匀top-K平滑(54.58)和自蒸馏(57.61)都无法匹配V-pretraining(58.98)的效果。去污染实验证明增益并非来自预训练数据中的基准泄露:在移除GSM8K和MATH近似重复后,V-pretraining仍保持优势(57.5 vs 56.7)。运行时分析显示,V-pretraining引入9.4%的吞吐量下降和16.4%的峰值内存增加,这些开销已在墙钟匹配的主结果中计入。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 数学推理 (MATH Pass@1) | Pass@1 (%) | 29.60 | 22.20 | +7.4 (33%相对提升) |
| 数学推理 (GSM8K Pass@1, Qwen1.5-0.5B) | Pass@1 (%) | 22.67±1.05 | 19.15±1.16 | +3.52 |
| 数学推理 (GSM8K Pass@1, Qwen1.5-4B) | Pass@1 (%) | 58.98±1.03 | 56.48±1.56 | +2.50 |
| 数学推理 (GSM8K Pass@1, Qwen1.5-7B) | Pass@1 (%) | 66.17±0.63 | 65.26±1.06 | +0.91 |
| 语义分割 (ADE20K, DINOv3-ViT-L) | mIoU | 52.47 | 51.33 | +1.14 |
| 深度估计 (NYUv2, DINOv3-ViT-L) | RMSE ↓ | 0.5522 | 0.5752 | -0.0230 |
| 图像分类 (ImageNet-1K, DINOv3-ViT-L) | Accuracy (%) | 84.59 | 84.07 | +0.52 |
| 语义分割 (ADE20K, DINOv3-ViT-B) | mIoU | 49.68 | 48.82 | +0.86 |
| 深度估计 (NYUv2, DINOv3-ViT-B) | RMSE ↓ | 0.5697 | 0.5888 | -0.0191 |
局限与改进
本文作者坦诚地指出了几个重要局限性。首先,V-pretraining提供的是局部价值信号,而非长时域最优性保证。对齐目标可能具有噪声,依赖于反馈质量,并且可能对小型学习者过度导向狭窄的反馈任务。实验也证实了这一点:0.5B模型在MMLU上出现下降(38.08→35.01),表明小模型对窄反馈更敏感。其次,方法引入了额外的计算开销:任务生成和价值梯度估计需要额外的GPU时间和内存。运行时分析显示9.4%的吞吐量下降和16.4%的峰值内存增加(5,138 MiB),这意味着在相同计算预算下,V-pretraining处理的token数量少于基线。第三,价值信号的质量取决于反馈集的代表性和多样性。本文使用1,024个GSM8K示例作为语言反馈,这些示例可能无法覆盖所有数学推理类型。第四,该方法目前只在相对简单的下游任务上验证(GSM8K数学题、ADE20K分割、NYUv2深度),在更复杂的推理任务或多步决策场景中的效果尚不确定。第五,从Table 6的多任务反馈结果来看,V-pretraining在MMLU和MMBP上的提升幅度较小(41.90→42.17和4.00→4.00),说明对于与反馈任务关联较弱的能力,引导效果有限。最后,论文未讨论反馈集大小、反馈集质量与最终性能之间的系统性关系,这限制了方法在实际应用中的指导意义。
独立分析的弱点
V-pretraining存在几个值得关注的弱点。第一,运行时开销偏高:9.4%的吞吐量下降和16.4%的内存增加意味着在资源受限场景下可能不划算,特别是当反馈任务与目标能力关联不强时。改进方向包括:(1) 设计更轻量级的设计器架构,(2) 降低价值梯度计算频率(当前每8步一次,可进一步稀疏化),(3) 使用梯度缓存机制减少重复计算。第二,小模型上的泛化退化问题:0.5B模型在MMLU上下降了3个百分点,说明价值信号可能过度导向小模型的更新方向。改进方向包括:(1) 引入正则化项防止过度专业化,(2) 根据模型规模自适应调整价值信号的权重,(3) 混合多个反馈任务以增加多样性。第三,反馈集依赖性强:方法假设存在高质量、任务相关的反馈示例,但如何选择和构造反馈集缺乏指导。改进方向包括:(1) 研究反馈集大小与性能的关系曲线,(2) 探索自动选择反馈示例的策略,(3) 研究合成反馈信号的可能性。第四,缺乏长时域理论保证:Proposition 2.1仅提供一步下界,实际训练中梯度噪声、优化器状态、分布漂移等因素可能导致累积误差。改进方向包括:(1) 探索多步价值估计,(2) 引入自适应学习率调节机制。
未来方向
本文开辟了几个有前景的研究方向。作者明确提出的方向包括:(1) 将V-pretraining与其他任务构建规则结合,如I-JEPA中的掩码策略;(2) 在更大规模模型和更多模态上验证方法的有效性。基于本文成果可延伸的方向包括:第一,探索'任务构建空间'的更丰富设计:当前方法在语言中仅调整top-K软目标,在视觉中仅调整视图参数,但任务构建的自由度远不止于此——可以探索自适应序列长度、动态数据混合、课程学习与任务构建的联合优化等。第二,研究反馈信号的扩展形式:当前使用标注的下游示例作为反馈,但可以探索弱监督反馈(如模型自身的一致性检查)、多模态反馈(如视觉-语言对齐信号)、或在线生成的合成反馈。第三,将V-pretraining与后训练方法(SFT、RLHF、GRPO)深度整合:Figure 2已显示V-pretraining可以作为SFT+GRPO的低成本前奏,进一步研究两者的最优组合策略和训练课程将很有价值。第四,理论层面可以探索:(a) 长时域对齐的收敛保证,(b) 反馈噪声对价值信号稳定性的影响,(c) 多任务反馈的最优权重分配理论。第五,应用到更多领域:代码生成(用单元测试作为反馈)、多模态理解(用视觉问答作为反馈)、科学计算(用物理一致性作为反馈)等。
复现评估
从复现角度来看,本文提供了较为详细的实验设置,但存在一些挑战。有利因素包括:(1) 使用公开可获得的基础模型(Qwen1.5/2.5系列、DINOv3),降低了模型获取门槛;(2) 使用公开数据集(NuminaMath-CoT、ImageNet-1K、GSM8K、ADE20K、NYUv2),数据可获取;(3) 论文提供了详细的超参数设置和训练配置(Appendix F),包括学习率、batch size、序列长度等;(4) Qwen1.5结果报告了均值和标准差,便于统计比较。挑战因素包括:(1) 论文未提供开源代码实现,需要读者自行实现设计器架构、价值梯度计算和训练循环;(2) 计算资源需求较高:主实验使用GPU集群,单次Qwen1.5-4B训练需要大量GPU小时,价值梯度计算还需要额外的反向传播;(3) 价值梯度仅在参数子集 $S$ 上计算,但具体选择哪些层作为子集 $S$ 的指导不够明确;(4) 多任务反馈实验(Table 6)使用了特定的数据混合比例(70%原始数据+30%高质量指令数据),这些设计选择对结果可能有显著影响。总体而言,对于有充足计算资源和深度学习经验的研究团队,复现主结果是可行的,但精确复现所有数值可能需要一定的调参工作。
论文图表
该表格评估了在视觉模态中,密集预测任务反馈是否损害了其他任务的表征质量。使用冻结的ViT-L表示在Oxford5k和Paris6k实例检索任务上进行评估,分为Easy、Medium、Hard三个难度。结果显示大多数检索协议有所提升,仅Oxford Hard略有下降。
这个表格与Table 2互补,从视觉角度验证了V-pretraining不会将表征坍塌到反馈任务上,证明了方法在视觉模态中的泛化安全性。
该表格展示了去污染实验结果。使用MinHash LSH和n-gram Jaccard相似度从NuminaMath-CoT中移除与GSM8K和MATH相似的样本后,基线和V-pretraining的GSM8K Pass@1分别变为56.7和57.5。虽然差距缩小,但V-pretraining仍保持优势。
这个表格直接回应了'V-pretraining的增益是否来自数据泄露'的质疑,通过去污染实验排除了预训练数据中包含基准近似重复的可能性,增强了方法有效性的可信度。
该表格报告了Qwen1.5-4B设置下的消融实验。基线为标准下一词预测(56.48),随机反馈(54.31)和均匀top-K平滑(54.58)都低于基线,自蒸馏(57.61)高于基线但低于V-pretraining(58.98)。只有V-pretraining使用 $g_{down}^T g_{pre}$ 训练设计器。
这个表格是验证'价值信号有效性'的核心消融,证明了增益不是来自软标签、额外噪声或自蒸馏,而是需要任务相关的下游反馈。
该表格展示了多任务反馈实验结果。视觉多任务使用ADE20K分割和NYUv2深度估计作为联合反馈,语言多任务使用数学、代码和通用指令跟随三个领域的反馈。语言多任务结果显示V-pretraining在GSM8K(44.28→45.26)和MMLU(41.90→42.17)上均有提升,但MMBP保持不变(4.00→4.00)。
这个表格展示了V-pretraining在多任务反馈场景下的能力,验证了价值信号可以组合多个下游目标,扩展了方法的实用范围。
该表格报告了运行时性能分析。基线每步处理65,536 tokens(梯度累积32),步时2.45秒,吞吐量26.7k tokens/s,峰值内存31,302 MiB。V-pretraining每步处理40,960 tokens(梯度累积20),步时1.69秒,吞吐量24.2k tokens/s,峰值内存36,440 MiB。吞吐量下降9.4%,内存增加16.4%。
这个表格量化了V-pretraining的计算开销,帮助读者评估方法的实际可行性。它解释了为什么主实验采用墙钟时间匹配而非token数量匹配的比较方式。