神经灌木丛:多样化的任务专家密布于预训练权重周围 Neural Thickets: Diverse Task Experts Are Dense Around Pretrained Weights
大模型预训练权重附近密布着多样化的任务专家
前置知识
预训练权重空间
预训练后的模型参数可以看作高维空间中的一个点 $\theta \in \mathbb{R}^d$,其中 $d$ 是参数量(如 7B 参数对应 $d \approx 7 \times 10^9$)。该点周围的邻域包含了对不同下游任务有不同表现的参数变体。传统的微调方法通过梯度下降在这个空间中搜索更好的参数点,而本文的核心发现是这个邻域中好解的密度随模型规模增大而急剧增加。
理解参数空间的几何结构是理解本文核心论点——随机猜测在大模型上能奏效——的前提
高斯扰动与随机猜测
给定预训练权重 $\theta$,通过添加高斯噪声 $\epsilon \sim \mathcal{N}(0, \sigma^2 I)$ 来生成扰动后的权重 $\theta' = \theta + \sigma \cdot \epsilon$,其中 $\sigma$ 控制扰动幅度。传统观点认为这种随机搜索在高维空间中是低效的(Schmidhuber et al., 2001),但本文发现预训练后这一假设不再成立。
这是 RandOpt 算法的核心机制,理解它才能理解为什么随机方法能在大模型上与梯度方法竞争
解密度 (Solution Density)
论文正式定义为 $\delta(m) = P_{\epsilon \sim \mathcal{N}(0, \sigma^2 I)}[s(\theta + \epsilon) \geq s(\theta) + m]$,即随机扰动能使模型在任务 $s$ 上性能提升至少 $m$ 的概率。直觉上,它衡量了随机猜测的命中率。实验中使用 $\sigma = 0.005$ 作为扰动尺度,发现解密度随模型规模单调递增。
这是量化灌木丛效应的核心指标,直接决定了随机搜索能否成功
谱不一致性 (Spectral Discordance)
定义为 $D = 1 - \frac{1}{M(M-1)} \sum_{j \neq k} C_{jk}$,其中 $C_{jk}$ 是任务 $j$ 和 $k$ 之间百分位排名的皮尔逊相关系数。$D \to 1$ 意味着扰动在不同任务上表现正交(专家化),$D \to 0$ 意味着扰动在所有任务上表现一致(通用化)。理论界为 $[0, \frac{M-1}{M}]$。
这个指标证明了扰动是专家而非通才,解释了为什么集成学习有效
测试时多数投票 (Test-Time Majority Vote)
在推理阶段,对同一输入使用 $K$ 个不同的模型副本生成 $K$ 个回答,然后通过多数投票选择最终答案。公式为 $\hat{y} = \text{mode}\{\arg\max_y f_{\theta_i}(y|x) \mid i \in \mathcal{I}_{top}\}$。这是一种简单但有效的集成策略。
RandOpt 的推理阶段依赖此方法来聚合多个专家的能力,是性能提升的关键
GRPO / PPO / ES
三种标准的后训练方法。PPO (Proximal Policy Optimization) 是基于策略梯度的强化学习方法;GRPO (Group Relative Policy Optimization) 是 PPO 的变体,通过组内相对排名来估计优势函数;ES (Evolutionary Strategies) 使用基于排名的随机梯度估计来优化参数。这些方法都需要数百步的顺序优化,而 RandOpt 完全并行。
这些是本文的主要基线方法,RandOpt 的核心卖点之一就是与它们竞争性能的同时实现 O(1) 的墙钟时间
研究动机
传统观点认为,在高维参数空间中随机猜测是不可行的。Schmidhuber、Hochreiter 和 Bengio 在 2001 年的论文中明确指出随机猜测不能被视为合理的学习算法。这一观点基于一个直觉:从零开始随机猜测一个十亿维的参数向量并使其具有 ChatGPT 的能力,其概率是天文数字级的。然而,本文发现预训练彻底改变了这一局面。在小模型(如 0.5B 参数)中,好的下游任务解确实占据了参数空间的极小比例,处于大海捞针的困境中,需要梯度下降等结构化搜索算法。但当模型规模增大到 1.5B 到 32B 参数时,任务改进解的密度急剧增加,进入了一个完全不同的灌木丛(thicket)体制。这意味着当前广泛使用的 PPO、GRPO、ES 等需要数百步迭代优化的后训练方法,可能在大模型上过于复杂了——好解就在预训练权重附近,随机采样就足以找到它们。
本文的目标是本文的具体目标是多维度的。首先,作者希望量化预训练权重周围解空间的几何结构,特别是解密度和解多样性随模型规模的变化规律。其次,基于这些发现,作者提出了一个极其简单的后训练算法 RandOpt——随机采样 $N$ 个参数扰动,在训练数据上评估,选择 top-$K$ 个,通过多数投票集成预测。这个算法完全并行,不需要梯度计算,墙钟时间为 $O(1)$。最终目标不是证明 RandOpt 优于其他方法,而是用它作为一种探针:如果随机猜测就能成功,说明预训练已经将模型带入了密集解的区域,后训练变得容易了。
与已有工作不同的是,本文的独特切入角度在于从参数空间的几何视角来理解预训练和后训练的关系。此前的工作主要关注两个方向:一是优化后训练算法本身(如 PPO、GRPO),二是从输出空间分析预训练模型的行为分布。本文则首次系统地量化了预训练权重周围的解密度和解多样性,揭示了一个关键但被忽视的现象:预训练不仅找到了一个好的初始化点,而是将模型带入了一个充满多样化任务专家的区域。这个视角统一解释了为什么简单的后训练方法也能有效(因为好解密集),为什么集成学习有帮助(因为解是专家化的而非通用的),以及为什么预训练规模如此重要(因为解密度随规模单调递增)。这种参数空间的几何视角此前从未被系统地研究过。
核心方法
RandOpt 的整体思路可以用一个直觉来概括:大模型的预训练权重周围是一个灌木丛,里面密布着各种任务专家,随机探索就能找到它们,然后把它们的能力聚合起来。具体来说,给定预训练权重 $\theta$,算法首先生成 $N$ 个随机高斯扰动 $\epsilon_i \sim \mathcal{N}(0, \text{Id})$,每个扰动乘以一个随机选择的噪声尺度 $\sigma_i$,得到扰动后的权重 $\theta'_i = \theta + \sigma_i \cdot \epsilon_i$。然后用一个小型训练集 $D_{train}$ 评估每个扰动模型的表现,选择得分最高的 $K$ 个作为专家。在推理阶段,对每个测试输入,用这 $K$ 个专家分别生成预测,最后通过多数投票得到最终答案。这个过程完全并行——所有 $N$ 个模型可以同时评估,不需要任何梯度计算或顺序更新。
RandOpt 的核心创新在于它利用了预训练的一个此前未被认识的特性:大模型参数空间中任务改进解的高密度和高多样性并存。与已有方法的本质区别在于三个方面。第一,与 PPO/GRPO 等基于梯度的方法不同,RandOpt 完全不使用梯度,而是通过随机采样和评估来选择好的权重,这使得它天然适合并行化。第二,与进化策略 ES 不同,RandOpt 不需要维护种群的统计信息或进行迭代更新,它只需要一轮采样和选择。第三,与 Best-of-N 等输出空间搜索方法不同,RandOpt 是在权重空间进行搜索——它不是对一个模型生成多个输出然后选最好的,而是直接搜索多个不同的模型。作者强调,RandOpt 的成功不是因为它优于其他方法,而是因为它作为一种探针,揭示了预训练已经将模型带入了一个好解密集的区域——在这个区域里,不管你用什么方法(梯度、进化、暴力搜索),都能找到好的后训练解。
方法步骤详情
RandOpt 算法的完整步骤如下。第一步,初始化:从预训练权重 $\theta \in \mathbb{R}^d$ 出发,定义一组噪声尺度 $\Sigma = \{\sigma_1, \ldots, \sigma_M\}$(实验中使用 4 个尺度)。第二步,随机生成:采样 $N$ 个随机种子 $\{s_1, \ldots, s_N\}$,每个种子对应一个标准高斯噪声向量 $\epsilon(s_i) \sim \mathcal{N}(0, \text{Id})$,同时为每个种子随机分配一个噪声尺度 $\sigma_i \in \Sigma$,生成扰动权重 $\theta'_i = \theta + \sigma_i \cdot \epsilon(s_i)$。第三步,评估选择:用训练集 $D_{train}$ 评估每个扰动模型 $f_{\theta'_i}$ 的性能得分 $v_i$,选择 top-$K$ 个模型的索引 $\mathcal{I}_{top} = \text{arg topK}(v_i)$。第四步,集成推理:对于测试输入 $x$,用 $\mathcal{I}_{top}$ 中的每个模型生成预测,通过多数投票得到最终输出 $\hat{y} = \text{mode}\{\arg\max_y f_{\theta_i}(y|x) \mid i \in \mathcal{I}_{top}\}$。关键超参数包括 $N$(种群大小,实验中使用 500 到 100000)、$K$(集成大小,实验中使用 1 到 50)和 $\Sigma$(噪声尺度集合)。
技术新颖性
RandOpt 的技术新颖性体现在多个层面。首先,它将 Best-of-N 范式从输出空间迁移到了权重空间,这是一个概念性的转变:不是在同一个模型上搜索最好的输出,而是在不同的模型上搜索最好的权重。其次,论文首次系统地量化了预训练权重周围的解密度——用数学定义为 $\delta(m) = P_{\epsilon}[s(\theta + \epsilon) \geq s(\theta) + m]$——并揭示了它随模型规模的单调递增规律(Scaling Law)。第三,论文提出了谱不一致性指标 $D$ 来量化解的多样性,证明了扰动是任务专家而非通用改进者。第四,从损失景观分析的角度,论文揭示了一个反直觉的现象:预训练后的任务特定损失景观不是平坦的,而是崎岖的——预训练权重可能处于任务准确率的谷底,周围环绕着许多更高的山峰。这与传统的平坦极小值观点形成了有趣的对比。第五,论文通过 1D 信号实验展示了灌木丛现象在简单设置中也能出现,表明这不是大语言模型独有的特性,而是一个更普遍的预训练现象。
实验结果
论文的核心发现可以归纳为五个相互关联的实验结论。第一,解密度随模型规模单调递增:在 Qwen2.5 模型系列(0.5B 到 32B)上,对 GSM8K 任务,使准确率提升 5% 的扰动密度从 0.5B 模型的约 10% 增加到 32B 模型的约 60%。第二,解多样性也随规模递增:谱不一致性 $D$ 从 0.5B 的约 0.3 增加到 32B 的约 0.9,说明大模型周围的扰动是高度专家化的。第三,RandOpt 在多个任务和模型上与标准方法竞争力相当:在 Qwen2.5-3B-Instruct 的 GSM8K 任务上,RandOpt(K=50)达到 86.7% 准确率,超过 GRPO 的 82.0% 和基线的 56.1%。第四,RandOpt 具有极高的时间效率:在 200 个 GH200 GPU 的集群上,用 N=2000、K=50 对 OLMo-3-7B-Instruct 进行 Countdown 任务的后训练仅需 3.2 分钟即达到 70% 准确率。第五,性能增益来自两个方面:格式修复和推理能力提升。在 Qwen2.5-3B-Instruct 的 GSM8K 实验中,RandOpt(K=50)的 86.7% 准确率中,19.0% 来自格式修复(模型原本答对但输出格式错误),12.3% 来自推理提升(模型原本答错但现在答对),仅 0.7% 为性能退化。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| GSM8K (数学推理) | 准确率 (%) | RandOpt K=50: 86.7% | 基线: 56.1%, GRPO: 82.0% | 相对基线 +30.6%, 相对 GRPO +4.7% |
| Countdown (数学推理) | 准确率 (%) | RandOpt K=50 约 70% (OLMo3-7B-Instruct) | 基线约 40% | 约 +30% |
| GQA (视觉推理) | 准确率 (%) | RandOpt K=50: 69.0% | 基线: 56.6% | +12.4% |
| GSM8K 蒸馏 (3B) | 准确率 (%) | Qwen2.5-3B-Inst 蒸馏: 84.3% | RandOpt K=50 集成: 87.1%, 基线: 79.8% | 蒸馏损失仅 2.8%, 远优于基线 |
| GSM8K 蒸馏 (1.5B) | 准确率 (%) | Qwen2.5-1.5B-Inst 蒸馏: 74.9% | RandOpt K=50 集成: 76.4%, 基线: 58.8% | 蒸馏损失仅 1.5%, 远优于基线 |
局限与改进
论文承认的局限性包括以下几个方面。第一,RandOpt 不适合从零训练:在未预训练的模型上(如随机初始化),RandOpt 的性能接近零(Figure 8 红色虚线),在小预训练模型(如 Qwen 0.5B)上增益也很小,说明预训练是必要条件。第二,推理成本问题:RandOpt 在推理时需要 $K$ 次前向传播,虽然蒸馏可以缓解(训练成本仅为原始的 2%),但蒸馏引入了额外的训练步骤、破坏了完全并行性,且蒸馏方法目前仅适用于有明确最终答案的推理任务。第三,性能增长趋于饱和:Figure 7 和 Figure 10 显示,随着种群大小 $N$ 的增加,准确率增长曲线在对数资源下仍然可见饱和现象,暗示对于需要更大幅度能力提升的任务,可能需要离开局部灌木丛进行更广泛的搜索。第四,集成方法限于离散预测:多数投票适用于有明确答案的任务(数学、编程),但对于结构化预测(写故事、生成图像、设计分子),如何集成和蒸馏仍不清楚。作者在附录中展示了扩散模型上的均值集成作为概念验证,但承认这不是最优选择。第五,性能增益的来源有浅层成分:在 GSM8K 上,约 19% 的性能提升来自格式修复而非真正的推理能力提升,这说明部分灌木丛是表面的。
独立分析的弱点
从独立分析的角度,本文存在几个值得关注的弱点。首先,实验任务主要集中在有明确评判标准的任务上(数学题有标准答案、编程题有测试用例、化学反应有产物验证),对于更开放的任务(如创意写作 ROCStories),论文只展示了定性结果,缺乏系统的定量评估框架。改进建议:引入自动化的开放任务评估指标(如 RUBRIC 评分、人类偏好评估),并验证 RandOpt 在这类任务上的有效性。其次,噪声尺度 $\Sigma$ 的选择对性能有显著影响(Figure 7 中 K/N 的变化表明不同选择比例下最优性能差异明显),但论文缺乏系统的超参数敏感性分析。改进建议:提供自适应的噪声尺度选择策略,例如基于预训练损失曲率来确定合理的搜索范围。第三,论文使用高斯扰动 $\epsilon \sim \mathcal{N}(0, \text{Id})$ 作为默认的搜索分布,但没有探索更结构化的搜索分布(如低秩扰动、基于梯度信息的定向扰动),这可能错失了进一步提升性能的机会。改进建议:结合 LoRA 等低秩方法,探索在低秩子空间中的随机搜索,可能同时降低搜索维度和提升搜索效率。第四,蒸馏实验仅在两个模型规模(1.5B 和 3B)和一个任务(GSM8K)上进行,缺乏对蒸馏方法泛化性的系统评估。改进建议:在更多模型和任务上验证蒸馏的有效性,特别是探索是否可以通过课程学习(从简单任务到复杂任务)来提升蒸馏效果。
未来方向
作者提出了几个重要的未来研究方向。第一,探索预训练为何会进入灌木丛体制:论文的 1D 信号实验表明,在多种信号类型上预训练是关键因素,但这在大规模语言模型中的具体机制仍不清楚。未来需要研究预训练目标、学习动态和数据多样性如何共同塑造参数空间的几何结构。第二,扩展到结构化预测任务:目前的多数投票集成仅适用于离散预测,对于文本生成、图像生成等结构化输出,需要开发新的集成策略。作者在附录中展示了扩散模型上的均值集成作为概念验证,但最优的集成或蒸馏方法仍有待探索。第三,从本文的成果可以延伸的方向包括:探索灌木丛是否可以解释其他令人惊讶的后训练现象(如随机奖励训练的有效性),研究灌木丛在不同架构(Transformer vs. SSM vs. 混合架构)中的普遍性,以及开发基于灌木丛理论的新一代后训练算法——例如在参数空间中进行有指导的探索(而非纯随机)来更高效地找到最优的专家组合。第四,联邦学习场景:由于 RandOpt 完全并行且不需要梯度通信,它天然适合数据隐私受限的联邦设置,每个参与方可以独立地在本地搜索好的扰动,然后只共享性能分数。
复现评估
论文在可复现性方面提供了良好的支持。代码已开源在 GitHub(https://github.com/sunrainyg/RandOpt),项目页面也已公开(https://thickets.mit.edu)。实验使用了多个公开可用的模型(Qwen2.5 系列、Llama3.1-8B、OLMo3-7B),这些模型的权重和训练细节都是公开的。数据集也是标准基准(GSM8K、MATH-500、MBPP 等)。在算力方面,论文展示了在 200 个 GH200 GPU 上仅需 3.2 分钟即可完成一次完整的 RandOpt 后训练(N=2000, K=50),这表明在中等规模的 GPU 集群上复现是可行的。对于小规模实验(N=500 到 3000),甚至可以在单机多卡上完成。复现的主要挑战在于:需要获取预训练模型权重(特别是 Qwen2.5 系列的 Instruct 变体),需要正确设置评估管道(特别是对于 Countdown、USPTO 等非标准任务),超参数选择($\Sigma$、$N$、$K$)可能需要根据具体任务和模型进行调整。总体而言,由于算法本身非常简单(随机采样加评估加选择加集成),复现的技术难度较低,主要成本在于计算资源。
论文图表
包含三个子图。(a) 概念示意图:左侧展示小模型的大海捞针体制——好的下游任务解占据预训练权重周围极小的区域,需要梯度下降等智能搜索算法;右侧展示大模型的灌木丛体制——任务特定解密布周围,随机采样即可找到,然后集成。(b) 解密度随模型规模的缩放定律——更大模型在高斯邻域中有更高密度的任务改进权重。(c) RandOpt 在 Countdown 任务上与 PPO、GRPO、ES 的性能对比——RandOpt 使用 5000 个随机权重猜测并集成 top K,与基线方法竞争力相当,且是 O(1) 训练步数和 FLOP 高效的。
这张图是论文的核心概念图,用直观的可视化传达了两个关键信息:(1) 大模型预训练权重周围的解空间结构与小模型有本质不同,(2) RandOpt 方法虽然极其简单但能与复杂方法竞争。理解这张图是理解整篇论文的基础。