跳到正文
原文
arXiv· arXiv:2312.09390· Collin Burns, Pavel Izmailov, Jan Hendrik Kirchner, Bowen Baker, Leo Gao, Leopold Aschenbrenner, Yining Chen, Adrien Ecoffet, Manas Joglekar, Jan Leike, Ilya Sutskever, Jeff Wu·本站收录· 原文发表精选关注度43

OpenAI 研究弱到强泛化:用弱监督激发强能力

Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervision

论文速读

防御

据论文 PDF 整理(AI 生成),以原文为准

OpenAI 用弱模型微调强模型研究弱到强泛化:朴素微调能超越弱监督者,NLP 上加辅助置信损失可恢复近 80% 差距。

问题
RLHF 等对齐方法依赖人类可靠监督,但未来超人模型的行为人类难以评估。作者用弱模型监督强模型作为类比,研究弱监督能否激发强模型的全部能力。
方法
用真值标签微调小模型得到弱监督者,再用其生成的弱标签微调强预训练模型,并以真值微调的强模型为上限,用 PGR 衡量恢复的性能差距;改进方法包括辅助置信损失、bootstrapping 和生成式微调。
实验与结果
朴素微调下强学生在 NLP、chess、RM 三类任务上几乎都超过弱监督者;NLP 上辅助置信损失把最小弱监督者配最大强学生的中位 PGR 从约 25% 提到近 80%,RM 上朴素 PGR 仅约 10%。
局限与可以继续做的
作者称没有方法在所有设置下都有效,RM 上仍远未恢复全部差距,方法只是概念验证;设置存在 imitation saliency 和 pretraining leakage 两个剩余差异。实验覆盖 GPT-4 系列基座模型、22 个 NLP 数据集、chess 和 RM 任务。
实验设置GPT-4 family pretrained base models、AlexNet 等 · 22 个 NLP 分类数据集(BoolQ、CosmosQA、DREAM、ETHICS、ANLI R2、CoLA、SST-2、HellaSwag、MCTACO、OpenBookQA、PAWS、QuAIL、PIQA、QuaRTz、SciQ、Social IQa、MultiRC、WiC、Twitter Sentiment)、Chess puzzles(Schwarzschild et al., 2021b,源自 lichess.org) 等 · test accuracy、PGR (performance gap recovered) 等
被测模型
GPT-4 family pretrained base modelsAlexNetDINO ResNet-50DINO ViT-B/8
基准
22 个 NLP 分类数据集(BoolQ、CosmosQA、DREAM、ETHICS、ANLI R2、CoLA、SST-2、HellaSwag、MCTACO、OpenBookQA、PAWS、QuAIL、PIQA、QuaRTz、SciQ、Social IQa、MultiRC、WiC、Twitter Sentiment)Chess puzzles(Schwarzschild et al., 2021b,源自 lichess.org)ChatGPT reward modeling 内部数据集ImageNetMATH
指标
test accuracyPGR (performance gap recovered)student-supervisor agreement
AI 导读和推荐理由OpenAI 超对齐团队研究能否用弱模型监督强得多的模型,以类比人类监督未来超人模型的核心难题。实验覆盖预训练算力跨 7 个数量级的 GPT-4 系列模型,任务包括 22 个 NLP 分类基准、国际象棋谜题和 ChatGPT 奖励建模。结果显示,用弱模型标签朴素微调强模型后,强模型表现稳定超过弱监督者,即弱到强泛化,NLP 任务上用 GPT-2 级监督者微调 GPT-4 通常恢复约一半性能差距;但奖励建模任务通常只恢复约 10%,作者据此认为朴素 RLHF 可能难以直接扩展到超人模型。辅助置信度损失、中间模型逐级自举和生成式微调等简单方法能明显改善泛化,最小弱监督者配最大强模型时 NLP 中位性能差距恢复比例从约 25% 提到近 80%。

OpenAI 超对齐团队研究能否用弱模型监督强得多的模型,以类比人类监督未来超人模型的核心难题。实验覆盖预训练算力跨 7 个数量级的 GPT-4 系列模型,任务包括 22 个 NLP 分类基准、国际象棋谜题和 ChatGPT 奖励建模。结果显示,用弱模型标签朴素微调强模型后,强模型表现稳定超过弱监督者,即弱到强泛化,NLP 任务上用 GPT-2 级监督者微调 GPT-4 通常恢复约一半性能差距;但奖励建模任务通常只恢复约 10%,作者据此认为朴素 RLHF 可能难以直接扩展到超人模型。辅助置信度损失、中间模型逐级自举和生成式微调等简单方法能明显改善泛化,最小弱监督者配最大强模型时 NLP 中位性能差距恢复比例从约 25% 提到近 80%。

推荐理由论文用弱模型监督强模型的类比实验,把对齐超人模型这一核心难题变成今天就能实证迭代的研究方向。

深度解读

6个问题,点开问题读完整回答

  1. 这篇论文试图解决什么问题?

    用弱模型监督强模型作类比,研究弱监督能否激发强模型全部能力,为超级对齐提供实证路径。

    核心问题:弱监督者(如人类)能否监督远比自己强的模型,并激发其全部能力? 作者将这一超级对齐(superalignment)难题转化为可实证的类比:用弱模型生成的标签微调强预训练模型,观察其泛化。

    • 问题场景:RLHF 依赖人类评估模型行为,但作者认为未来超人模型的复杂行为(如百万行代码是否安全、是否忠实遵循意图)人类无法可靠评估,朴素人类监督如何泛化到这些行为是未知的。
    • 现有研究的不足:作者称以往对齐研究要么直面核心挑战但限于理论框架和玩具问题,要么实证研究人类监督当今模型、未触及超人模型带来的核心挑战,缺少既能捕捉核心挑战又能迭代实证的设置。
    • 核心假设:强预训练模型对对齐相关任务已有良好表示,弱监督者不需要教它新能力,只需激发它已有的知识,因此强模型可能泛化超出弱监督水平,作者称之为 weak-to-strong generalization。
    • 论文的贡献:提出 weak-to-strong learning 实证设置,在 NLP、chess、reward modeling 三类任务上系统测量,并测试辅助置信损失、bootstrapping、生成式微调等改进方法。
    • 设置定位:作者称该设置去除了以往工作的一个主要差异,但并非最终完美类比的设置;剩余差异(imitation saliency、pretraining leakage)在 Section 6 讨论。
  2. 有哪些相关研究?

    与弱监督学习、师生训练、可扩展监督、知识激发等多条文献相关,区别在于学生远强于教师且监督质量弱。

    弱监督与噪声标签学习

    • 弱到强学习是弱监督学习的一种,相关文献包括从不可靠标签学习(Bach et al., 2017; Ratner et al., 2017)和噪声标签学习(Song et al., 2022),常见方法有 bootstrapping、噪声鲁棒损失、噪声建模。
    • 论文指出区别:本文弱监督的错误是 instance-dependent 的,比均匀标签噪声更难处理;也与半监督学习相关,附录 C 研究了 easy-to-hard generalization 变体。

    师生训练与知识蒸馏

    • 先训教师再用伪标签训学生的框架广泛用于半监督学习、领域自适应和知识蒸馏(Hinton et al., 2015; Xie et al., 2020 等)。
    • 论文指出区别:本文关注学生远强于教师的设置;Furlanello et al. (2018) 和 Xie et al. (2020) 虽考虑学生不弱于教师,但学生随机初始化且能接触真值标签。本文的置信辅助损失与 Grandvalet & Bengio (2004) 的条件熵最小化类似。

    预训练鲁棒性、去偏与对齐方法

    • 预训练带来更鲁棒的表示,微调常损害分布外泛化;缓解方法包括权重集成、只微调部分层等。作者在初步探索中未得到强结果(附录 B)。
    • 去偏文献多处理已知偏差,本文可视为偏差未知的去偏问题。模仿学习与偏好学习(RLHF、Constitutional AI)都假设有高质量人类监督,作者认为对超人模型是否有效未知。

    可扩展监督与知识激发

    • Scalable oversight 通过辩论、批判、任务分解提升人类监督能力(Irving et al., 2018; Leike et al., 2018);本文设置类似用强弱人类的 sandwiching 提案(Cotra, 2021; Bowman, 2022)。
    • ELK(Christiano et al., 2022)要在最坏假设下激发潜在知识;作者认为本文设置可作为实证研究 ELK 和 honesty 的通用方法。

    定位:作者称本文与多数师生训练工作相反,聚焦学生远强于教师、监督质量弱的设置,目标是泛化超出人类监督而非改进人类监督本身。

  3. 论文如何解决这个问题?

    三步流程:训弱监督者、用弱标签微调强学生、以真值微调为上限,用 PGR 衡量恢复差距,再加三类改进方法。

    整体思路:把"人类监督超人模型"替换为"弱模型监督强模型",用 GPT-4 系列预训练基座模型(跨越 7 个数量级的预训练算力)在 NLP、chess、reward modeling 三个设置中实证测量。

    问题设定与核心指标

    • 三步流程(Section 3):1) 用真值标签微调小预训练模型得到弱监督者,其在留出集上的预测作为弱标签;2) 用弱标签微调强模型得到 strong student,其表现为 weak-to-strong performance;3) 用真值标签微调同一强模型作为 strong ceiling performance。
    • 核心指标:PGR = (weak-to-strong − weak)/(strong ceiling − weak),衡量弱监督恢复的性能差距比例;完美泛化时 PGR 为 1,不超过弱监督者时为 0。

    三个任务设置

    • NLP:22 个流行分类数据集,统一转为类别平衡的二分类,弱标签为软标签(类别概率);微调 2 个 epoch、batch size 32,按弱标签验证准确率早停。
    • Chess puzzles:预测局面最优首着,生成式任务;弱标签由弱模型温度 0 采样得到;训练 5 个 epoch、不早停;训练用 50k+50k 谜题、测试 5k。
    • ChatGPT reward modeling:用训练 ChatGPT RM 的内部比较数据,预测成对比较结果;训练 1 个 epoch、batch size 220、不早停;只研究 RM 准确率,不研究 RL 步骤。

    改进方法

    • Bootstrapping(Section 4.3.1):构造递增模型序列 M1→M2→…→Mn,每步用上一步模型生成的新弱标签微调下一步;每轮 bootstrapping 跑三次弱到强学习(两个中间尺寸),在 chess 上评测。
    • 辅助置信损失(Section 4.3.2):Lconf(f) = (1−α)· CE(f(x), fw(x)) + α· CE(f(x), f̂t(x)),即在弱标签交叉熵上加一项强化强模型自身(经阈值 t 硬化的)预测的损失;t 自适应取使批内恰好一半样本超过阈值,αmax 对最大学生取 0.75、其余 0.5,前 20% 训练线性预热。设计动机是让学生学监督者的意图而不模仿其错误。
    • 生成式微调(Section 5.2.2):在 RM 数据的所有 prefix-completion 对上做语言建模目标的无监督微调(忽略人类偏好),提高任务在强模型中的显著性;作者指出补全可能来自人类或更强模型,可能存在类似 pretraining leakage 的泄漏。

    分析与评测方式

    • 模仿分析(Section 5.1):用训练过程中的真值测试准确率衡量对弱标签的过拟合;用 student-supervisor agreement(学生与弱监督者预测一致的测试样本比例)直接衡量模仿。
    • 显著性分析(Section 5.2):比较 zero-shot、5-shot 提示与微调;用线性探针(logistic regression)测任务在最终层激活中的线性表示程度。
    • 超参数调优:只在小模型和 8 个数据集的子集上调参;GPT-4 和约一半 NLP 任务在整个项目中保持留出。
  4. 论文做了哪些实验?

    三类任务上朴素微调普遍超越弱监督者;NLP 上置信损失将中位 PGR 从约 25% 提至近 80%,RM 上仅约 10%。

    实验设置

    • 被测模型:GPT-4 系列的预训练基座模型,跨越 7 个数量级(OOM)的预训练算力;论文用脚注说明该系列与 GPT-4 同架构和预训练数据,但不包括 GPT-2、GPT-3、GPT-3.5,正文以"GPT-2-level""GPT-3.5-level"描述算力档位。
    • 数据集:22 个 NLP 二分类数据集(Table 1);chess puzzles(Schwarzschild et al., 2021b,训练 50k+50k、测试 5k);ChatGPT reward modeling 内部数据集(规模未说明)。
    • 基线与上限:弱监督者表现(weak)、朴素弱标签微调(naive baseline)、真值微调上限(strong ceiling);改进方法为 bootstrapping、辅助置信损失、生成式微调。
    • 指标:test accuracy 与 PGR;NLP 结果取 22 个任务的中位数;NLP 随机猜测为 50%。
    • 重复与误差:多数曲线未报告重复次数;Figure 13(b) 报告 s.e.m. 误差棒;Figure 27 报告均值±标准差。

    主结果

    主结果以曲线图(Figure 3)报告,论文未给出逐点数值表;下表汇总正文给出的代表性数字。

    表格较宽,可左右滑动

    设置朴素微调 PGR最佳方法最佳方法 PGR出处
    NLP(GPT-2 级监督 GPT-4)约 25%(最小弱监督者配最大学生)辅助置信损失近 80%Section 4.3.2
    NLP(最大学生总体)常高于 50%——Section 4.2
    Chess(小监督-学生差距)可高于 40%bootstrapping提升,尤其大学生Section 4.2、4.3.1
    ChatGPT RM约 10%,几乎从不超 20%生成式微调提升约 10–20%Section 4.2、5.2.2
    • 作者称三个设置中强学生几乎普遍超过弱监督者,NLP 上 PGR 随弱监督者尺寸和学生尺寸增大而上升。
    • 作者称 chess 上最小弱模型时 PGR 接近零,且给定弱监督者时 PGR 随学生变大而下降,曲线呈凹形、可能 inverse scaling。
    • 作者称 RM 上朴素弱到强泛化差,认为这提供了朴素 RLHF 可能难以扩展到超人模型的实证证据。

    过拟合与模仿分析(Section 5.1)

    • 测了什么:训练过程中强学生的真值测试准确率,以及学生-监督者一致率。
    • 结果:RM 任务上弱强差距大时,弱到强准确率在不足一个 epoch 内先升后降;真值早停("作弊"方法)在 RM 上带来约 5 个百分点 PGR 提升,在 NLP 上比训练结束模型高 15 个百分点、比按弱标签早停高 10 个百分点。朴素微调下学生-监督者一致率常高于弱监督者准确率;更大的学生对监督者错误的一致率反而更低(inverse scaling)。
    • 作者解读:学生在模仿监督者的部分错误;置信损失降低了对监督者错误的一致率(Figure 8c),在部分设置中一致率甚至低于学生真值准确率。

    提示与显著性实验(Section 5.2)

    • Few-shot 对比:7 个代表性 NLP 任务上,zero-shot 与 5-shot 对小模型差、对大模型大幅改善,最大模型上 5-shot 与微调相当;带弱标签的 5-shot 弱于带真值标签的 5-shot,最大学生上 5-shot 弱到强表现可与朴素微调相当或更好,但仍不如加置信损失的微调(Figure 9)。
    • 生成式微调(RM):在 RM 数据上做无监督语言建模微调后,即使上限也经同样微调,PGR 仍提高约 10–20%;与真值早停叠加可达约 30–40% PGR(Figure 10、17)。
    • 线性探针(NLP 子集):真值线性探针平均 72%,真值微调 82%;先弱标签微调再真值线性探针达 78%,弥合了线性探针与微调之间 60% 的差距,且优于朴素弱到强微调(Figure 11)。

    其他消融与分析

    • Bootstrapping 的适用范围:chess 上提升明显;NLP 上初步实验只有小幅提升,RM 上无提升(Section 4.3.1)。
    • 置信阈值过滤:只取弱监督者最自信的 5%–20% 样本训练,仅在弱标签很差(约 60% 准确率)时有用,弱标签达 70%–80% 后不再有用(附录 B)。
    • 无效方法:LP-FT、权重衰减/dropout、LoRA、数据增强一致性损失、广义交叉熵、标签加噪、权重平均均未带来改进(附录 B)。
    • Easy-to-hard(附录 C):chess 上只用 Elo≤800 的易题训练,平均测试表现仍好但难题(Elo≥2000)上与 zero-shot 相当或略差;NLP 上提高难度截断一般提升准确率。
    • ImageNet(附录 D.1):AlexNet(56.6% top-1)监督 DINO ResNet-50/ViT-B/8 线性探针,学生达 60.7%/64.2%,PGR 57.8%/41.5%(Table 3)。
    • 弱标签错误结构(附录 E):相同弱标签准确率下,可完美模拟的错误结构几乎无泛化,随机噪声标签泛化最好;把弱标签直接放进提示词使模仿变得平凡时,平均 PGR 降至 0.002±0.003(基线)和 0.046±0.108(置信损失)(Figure 27)。
  5. 有什么可以进一步探索的点?

    作者称方法只是概念验证且设置有两类剩余差异;实验覆盖 GPT-4 系列模型与三类任务。

    作者指出的局限与后续方向

    • 方法不一致有效(Introduction、Section 4.3):作者称没有任何方法在所有设置下都有效,RM 设置上仍远未恢复全部性能差距,方法更多是概念验证而非可部署的实用方案。
    • Imitation saliency 差异(Section 3、6.1):未来超人模型可能很容易模仿人类错误,而本文强模型未被显式预训练去模仿弱模型,"模仿弱监督者"在本文设置中可能比未来更容易避免,可能抬高今天的泛化表现。
    • Pretraining leakage 差异(Section 3、6.1):预训练数据隐含人类监督,许多任务可能在预训练中被间接观察到,激发能力可能因此被人为变容易;超人知识可能更 latent、更难激发,作者推测这对提示法的抬高多于微调。
    • 后续方向——更类比的设置(Section 6.2.1):修复剩余差异、用如三年级人类监督最强模型来验证差异不严重、推广到复杂生成任务、测试高 PGR 的弱到强分类器在优化压力(如 RL)下的鲁棒性。
    • 后续方向——可扩展方法与科学理解(Section 6.2.2、6.2.3):寻找无需真值即可指定期望泛化的无监督性质;回答 NLP 与 RM 结果差异的原因、什么使概念易激发、能否无标签估计泛化误差、能否用 scaling law 外推等问题。

    实验覆盖范围

    • 被测模型为 GPT-4 系列预训练基座模型(跨 7 个数量级算力),附录另测 AlexNet 与两个 DINO 视觉模型;论文未在实验中运行 GPT-2/3/3.5 本体。
    • 任务覆盖 22 个 NLP 二分类数据集、chess puzzles、ChatGPT RM 内部数据集,附录覆盖 ImageNet 与 MATH 的难度分析;RM 实验只测 reward model 准确率,未做 RL 优化步骤。
    • 改进方法评测包括 bootstrapping(chess 主测)、辅助置信损失(NLP 主测)、生成式微调(RM 主测);附录 B 报告了置信阈值、LP-FT、LoRA、权重平均等未奏效的方法。
    • 论文未报告多数实验的随机种子数与重复次数;除 Figure 13(b) 的 s.e.m. 和 Figure 27 的标准差外,未给出误差棒或统计检验。
    • 超参数只在小模型和 8 个 NLP 数据集子集上调优,GPT-4 与约一半 NLP 任务全程留出(Section 3 脚注、附录 A.1)。
  6. 总结一下论文的主要内容

    弱模型监督强模型的实证类比:朴素微调普遍超越弱监督者,简单方法可大幅改进,但距恢复全部能力仍远。

    一篇对齐方向的实证研究:把"人类监督超人模型"转化为"弱模型监督强模型",用 GPT-4 系列预训练基座模型在 NLP、chess、ChatGPT reward modeling 三个设置中测量弱到强泛化。

    • 要解决的问题:RLHF 依赖人类可靠评估,但作者认为超人模型的复杂行为人类无法可靠监督;需要能今天迭代实证的类比设置。
    • 方法要点:真值微调小模型得到弱监督者→弱标签微调强学生→真值微调强模型为上限,用 PGR 衡量恢复的性能差距;改进手段包括辅助置信损失、bootstrapping、生成式微调。
    • 朴素微调普遍有效但不够:三个设置中学生几乎总是超过弱监督者;NLP 上最大学生 PGR 常高于 50%,但 RM 上通常只有约 10%、几乎不超 20%,作者据此认为朴素 RLHF 可能难以扩展到超人模型。
    • 简单方法可大幅改进:NLP 上辅助置信损失把最小弱监督者配最大强学生的中位 PGR 从约 25% 提到近 80%;chess 上 bootstrapping 提升 PGR;RM 上生成式微调提升约 10–20%,与真值早停叠加达约 30–40%。
    • 机理发现:强学生会过拟合弱标签错误(大差距时不足一个 epoch 就出现),但更大的学生模仿监督者错误反而更少;先弱标签微调可使真值概念更线性(探针准确率达 78%)。
    • 作者的结论与启示:作者称弱到强泛化是普遍现象、改进它是可行的,今天就可以对超级对齐的根本挑战做实证研究;同时强调方法只是概念验证,设置仍存在 imitation saliency 与 pretraining leakage 两类差异,需持续迭代。
阅读原文arxiv.org