跳到正文

目录

Model Discovery Agent 深度解读:Kevin Murphy 让 LLM 与 Bayesian 各管一层,破解 mechanistic discovery 的 M-open 死结

这篇文章在回答什么

arXiv:2608.09696 是 Kevin Murphy(Probabilistic Machine Learning 教材作者,Uber ML 时代的大佬级人物)2026 年 8 月挂出来的论文:Model Discovery Agent (MDA): LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models

Murphy 这篇的核心主张一句话能说完:传统 Bayesian 实验设计只解决「给定假设类,挑最 informative 的实验」——但科学发现并不总停留在给定的假设类里。真值可能落在假设类之外,统计学把这叫 M-open。LLM 的长项是「根据上下文,提议你没想到的假设」——但它单独跑,既没有 posterior 校准,也没有 likelihood 评估,更没有 data efficiency 保障。

MDA 把这两件事拼成一个 4 步循环:

LLM proposer (提议结构) → SMC 更新 posterior over (structure, parameters)
       Predictive check (held-out 干预预测误差)
       误差超阈值 → 扩张假设空间 (LLM 提新机制)
       VoI design (挑下一个最 informative 的实验)

物理 / 化学 / 生物三个 benchmark 都验证同一套循环:ForceBench(粒子间未知力律)、ChemBench(酶动力学率律)、NeuronBench(新贡献:6 个 mystery neuron + Hodgkin-Huxley ODE + 部分可观察 + 随机扩展 NeuronBenchStoch)。

这篇文章要回答三件事:

  1. 为什么 mechanistic model 必须靠 intervention —— 被动观测永远 underdetermine 机制
  2. 为什么 LLM 当 proposer 不会让 Bayesian 那套垮掉 —— 严格区分「结构」(LLM 出)和「参数/似然/VoI」(标准 Bayesian 出)
  3. M-open 的 predictive check 怎么不变成「LLM 瞎补」 —— 残差阈值 + 现有证据排序挑前 N_m

一个直觉:interventional question 不可能靠 curve fit 答

论文 §1 开篇那句很重:

Predicting the answer to interventional “what if” questions — the outcome of an action never taken — requires a mechanistic, causal model, not a curve fit; and learning such a model requires experiments, because passive data leaves its mechanisms unidentified.

Murphy 把「曲线拟合 vs 因果模型」当成科学发现的核心张力:一个药物从未给过的病人、一个轨道从未飞行过的探测器、一个从未实施过的政策——回答「如果做了会怎样」需要 mechanistic 模型,因为两个机制可以在所有观测数据上完全一致,却给出截然相反的干预预测。Richens & Everitt 2024 的定理说,能 robust 预测全范围干预的 agent 隐式学到了 causal world model。MDA 反过来:显式表示 causal model,让 LLM 注入 prior knowledge,用 Bayesian 推理不确定性,给用户可解释的模型。

但 mechanistic model 有第二个问题:passive data 永远 underdetermine 它。两个机制可以生成同一组观测;只有 intervene(扰动 + 看响应)才能打破这个 degeneracy。问题是「实验贵」——实验室化验、临床试验、卫星发射——所以操作性问题变成「data efficiency」:用尽可能少的实验,识别出足以回答 query 的机制。这正是 Bayesian experimental design 的古典命题(Lindley 1956):挑预期产出最 informative 的干预。Murphy 强调:这条路径几乎从未和「科学发现需要的开放式假设创造」结合过。MDA 就是这两个领域的接缝。

论文给了三个 anchor benchmark:

领域benchmark来源MDA 的 SOTA
物理ForceBench (DiscoverPhysics 包装)Wiemann et al. 2026在 6 个 two-particle world 上数据效率显著优于 Opus 4.7 / DeepSeek-v4 Pro baseline
化学ChemBench (ActiveSciBench-Chem 包装)Kabra et al. 20268 个实验达到 ≈56% SA;LLM-AutoSciLab baseline 在 B=60 步只到 ≈42%
生物NeuronBench (新贡献)本文作者6 个 mystery neuron (HH 模型加 novel 通道);Stochastic 扩展 NeuronBenchStoch

三个 benchmark 的共性是:真值已知(合成数据),所以可以同时测「预测误差」和「机制恢复对错」。LLM-AutoSciLab 在 ChemBench 上能拿到 RMSLE=0.001 但机制是错的(PySR 返回 10^a log(·) 这种「数值对、符号错」的形式),MDA 返回的则是 substrate inhibition 的真实方程——两种方法对「发现」的定义由此分道。

系统地图:MDA 的四层 inference

论文 §3 Methods 的核心是 4 层嵌套 inference,每一层用一个标准 Bayesian 工具。其中 SMC(Sequential Monte Carlo,序贯蒙特卡洛)是一族粒子采样方法:用一组带权重的粒子近似目标分布,逐轮吸收新数据后重加权、重采样,适合这里「边做实验边更新」的场景。

关键设计:LLM 只在结构层出现。一旦结构 m 固定,参数 θ、似然 p(D|m,θ)、实验设计 VoI 全部走标准 Bayesian 路径——SMC、particle filter、closed-form VoI。这是 Murphy 整篇论文的核心工程取舍:LLM 出结构,Bayesian 出数值。

下面按层拆解。

结构层:LLM proposer 的 prompt 设计

SMC-S 方法(Piriyakulkij et al. 2024)的 LLM proposal kernel 形式是:

p_b(m_b | {m_{b-1}^i}, D_{0:b-1}, r_{b-1}, C)

即第 b 步的结构提议 = 之前所有结构 + 它们的残差 + 当前数据集 + 自然语言 context C(领域描述)。MDA 在结构扩张时(残差超阈值)让 LLM 提议「novel unnamed mechanism」,赋予 broad prior——这是 M-open 的「补」操作。Breaker–Builder 方法(Buehler 2026)也走这条路。附录 G 给出了完整 prompt 模板。

LLM 不是每一步都重新提议——MDA 设了一个 R_m 阈值和最大扩张轮数 R_max > 0,超出后才启用「扩 N_new 个新候选 → 算证据 → 留前 N_m」的剪枝。这是「LLM 帮出主意,Bayesian 帮筛掉」的工程实现。

证据层:adaptive-tempered SMC 的 Occam 罚

每个 LLM 提议的结构 m 要算 evidence(marginal likelihood,边缘似然):

Z_m = p(D_{0:b} | m) = ∫ p(D_{0:b} | m, θ) p(θ | m) dθ

这是经典的模型选择量。Murphy 强调:「the integration over model parameters provides an automatic Occam penalty factor for complex models with many parameters (MacKay 1991)」。含义:复杂模型必须解释更多数据才能在 evidence 上赢过简单模型——SMC 在做结构选择时自动做了复杂度-拟合度 trade-off,不需要额外 BIC / AIC 那一类调整。

Algorithm 3 用 per-structure adaptive-tempered SMC 计算这个 evidence;Algorithm 2 用 SMC 在结构空间采样——两层 SMC 嵌套,内层算 Z_m,外层更新 p(m|D)。§1 Fig. 1(b) 的 Pareto curve 把每个候选模型画在 (accuracy, complexity) 平面上——Bayesian description length -log₂ p(m|D) 当 x 轴,准确度当 y 轴。

参数层:似然的两条路

参数 inference 的关键是 likelihood p(D|m,θ)。论文给了两条路:

路 1:确定性潜在动力学(ForceBench / ChemBench / 确定性 NeuronBench):

p(y_{1:T} | z_0, m, θ) = ∏_t p(y_t | z_t, m, θ)
z_t = m_θ^t(z_0)  # 推 m_θ 共 t 次

这是 closed-form——ODE 跑完直接出轨迹。Gaussian observation noise 时 VoI 还有 analytic expression(论文 Eq. 5)。

路 2:随机潜在动力学(NeuronBenchStoch,新贡献):

当动力学是 SDE 时,likelihood intractable。MDA 用 bootstrap particle filter 估计:

p̂(y_{1:T} | m, θ) = ∏_t (1/N_z ∑_i w_t^(i))

这个无偏估计 plugged into Algorithm 3 的 per-class tempered SMC 当 loglik,返回 model evidence Z_m = p(D|m)。这是 simulation-based inference (SBI) 的标准做法(Cranmer et al. 2020)。当确定性成立时 N_z=1,回退到路 1。

另一条路:单轨迹太噪时用 summary statistics。把轨迹转成 summary statistics s_j(y_{1:T}),用 trajectory-level likelihood:

p(y_{1:T} | m, θ) = ∏_j p(s_j(y_{1:T}) | m, θ)

附录 A.4 给出了「学 summary statistics 本身」作为 1d CNN 特征的初步结果。

设计层:VoI 在 M-open 下还管用吗

实验设计的目标是 Lindley 1956 的 mutual information:

ξ_⋆ = argmax_{ξ ∈ Ξ} I(M; Y_ξ | D)

确定性动力学 + Gaussian noise 时有 closed-form VoI(论文 Eq. 5):选 posterior-predictive variance 最大的设计。因为 per-structure 参数 posterior 通常比较集中,这个方差主要来自跨模型分歧——VoI 选的设计就是「最能把不同机制区分开的实验」。

这是 model discrimination 目标(区分机制)。论文在 ChemBench 上对比了 exploit 替代:

ξ_⋆^{mean} = argmax_{ξ ∈ Ξ} E_{m,θ|D}[r(ξ; m, θ)]

这是 Bayesian optimization 风格的 exploit——挑预期 rate 最高的输入。论文里叫「MDA (Mean)」baseline。结果显示 VoI 在 M-open benchmark 上明显占优:MDA 默认比 exploit 快 5-7 倍达到 ceiling。

M-open 的 predictive check:怎么不变成「LLM 瞎补」

论文 §3 「Expanding and shrinking the hypothesis space」一节给出的机制:

For each held-out test experiment ξ_i:
    1. 用当前 MAP 模型 m_* 预测 E[Y|ξ_i, m_*]
    2. 与 ground truth 对比,得 residual r_i
    3. 累积 residual R_m;若 R_m > threshold τ_r,启用扩张

扩张触发后:

  1. LLM 提议 N_new 个 novel unnamed mechanisms,赋予 broad priors
  2. 加入当前结构池
  3. 每个新候选算 evidence Z_m
  4. 留 evidence 最高的前 N_m 个

关键 invariant:「扩」是 lazy + threshold-gated——只在残差超阈值时启动,不是每轮都让 LLM 自由发挥。论文报告说这一条对 challenging domains 很关键(Fig. 16 ablation)。

论文给了一个 Yukawa world 的具体例子——Yukawa 力律 F = q_i q_j K₁(r/λ) / λ,其中 K₁ 是修正 Bessel 函数。短距离(r ≤ λ)时 K₁(r/λ)/λ 和普通 power law 几乎完全重合——只用短距离数据无法区分。VoI 设计的「长距离 probe」触发了 aha moment:在前沿上,本来 K₁ 和 power law 同居前沿中段,没有一个明显胜出;长距离 probe 加进来后,前沿整体下移,真模型掉到 convex corner。

Pareto curve 这一刻就是模型「真正 grok」概念的瞬间。x 轴不是语法复杂度(不是 Halstead 那种),是 Bayesian description length -log₂ p(m|D)——每个候选机制的 posterior code-length。

三 benchmark 的关键设计取舍

ForceBench(物理):8 步内见真章

任务:2D 空间里两个粒子,agent 控制其中一个的初位置 + 初速度 + 几个环境参数,找出未知力律 F

关键:力律用 Green’s function 表示,Newton 第二定律积分两步得轨迹。LLM 提出候选 F,MDA 算 posterior over F 和 VoI。设计空间离散化成 13 个组合(Table 5)。

结果:在 6 个 two-particle world 上聚合,MDA 比 Opus 4.7 / DeepSeek-v4 Pro 的 LLM baseline 数据效率显著高(Fig. 2)。Opus 4.7 在「unthrottled」模式下每步跑多个实验,16 轮才能到 MDA 8 轮的水平。

Yukawa world 的 aha moment 见上面 §M-open 段——长距离 probe 把真模型推到 Pareto 前沿的 convex corner。

ChemBench(化学):56% vs 42% 的 SOTA

任务:从 7 个可控输入(底物、抑制剂、第二底物、产物、酶载、温度、pH)映射到标量反应速率 r 的代数方程。

关键:噪声是 multiplicative Gaussian,likelihood 用相对噪声形式 p(y|ξ,f,θ) = N(y | f(ξ,θ), σ_rel · f(ξ,θ)),对齐 benchmark 的 RMSLE 度量。

结果

方法B=8 实验B=60 实验
MDA (VoI)≈56% SA
LLM-AutoSciLab≈42% SA
LLM-AutoSciLab (paper 原报)35.1% SA(用 gpt-4o-mini)

论文用 Opus 4.7 替换 gpt-4o-mini 跑 baseline,所以 42% > 35.1%——这是诚实的「替换 baseline 模型」操作。

Table 1 的对比最能说明问题:MDA 在 substrate inhibition 上返回完全正确的方程;LLM-AutoSciLab 的 PySR 在 hard noncompetitive 域返回 RMSLE=0.001(数值完美),但机制是「nested 10^a log(·) 和 stretched exponential」——数值对、符号错。这正是 Kabra et al. 2026 讨论的「high-exact / low-symbolic pathology」。

NeuronBench(生物):本文新贡献

任务:6 个 mystery neuron,每个基于 generalized Hodgkin-Huxley 模型(描述神经元 spike 的非线性 ODE),电流由 I_Na + I_K + I_L + I_Z 组成——前 3 个通道(钠、钾、漏电)是标准的 HH 模型,第 4 个 I_Z 是作者故意设计的 novel 膜机制,避免 LLM 直接靠「记忆」调出标准 HH 模型。

关键

  • LLM 只给表型(observable signature,不是机制),提议 2-5 个候选通道
  • 候选映射到一个共享 channel library
  • 真值有时根本不被提议——例如 z-rebound world 里 LLM 完全漏掉 low-threshold inward current,这是 genuine M-open miss,作者刻意保留,让 residual 能 reopen the pool
  • MDA 跑 Poisson-evidence selection 在候选池上 + VoI 实验设计

结果(Fig. 4):6 个 world 的 test error vs 实验数——Bayes-forecaster(蓝色)在每个 world 上都明显优于 in-context forecaster(紫色)。

Stochastic 扩展 NeuronBenchStoch(附录 F):在 latent 动力学加 finite-channel gating noise,把模型变成 SDE——likelihood intractable 是其他 benchmark 都没有的 regime。关键发现:

a deterministic likelihood gives poor results (the method confidently selects the wrong model), whereas a particle filter approximation to the marginal likelihood (Algorithm 4) gives the correct results.

确定性 likelihood 在 SDE 上「自信地选错」:用错误的 likelihood 算 evidence,越自信越危险。

工程取舍:哪些决策是钉死的

LLM 只在结构层。一旦 LLM 提议了 m,参数 θ、likelihood p(D|m,θ)、VoI 全部走标准 Bayesian。这条分界线把「LLM 会让 Bayesian 失效」的质疑挡在门外:LLM 出结构,Bayesian 算 evidence 和 VoI,两边互不越界。

Adaptive-tempered SMC 而不是 vanilla SMC。证据层的标准做法是 per-structure adaptive-tempered SMC(Naesseth et al. 2019; Chopin & Papaspiliopoulos 2020),这样在高维参数空间里 particle 不会全部 collapse 到一个 mode。代价是要调温度调度参数。

M-open 用 threshold-gated lazy expansion。不是每轮都让 LLM 自由发挥,而是 residual R_m > τ_r 才触发 N_new 个新候选 → 算 evidence → 留前 N_m。论文 Fig. 16 ablation 显示这一条对 challenging domains 很关键——没有 threshold gate 的版本在 NeuronBench 上 evidence 不会收敛。

两套 likelihood:deterministic ODE 与 particle filter SDE。ForceBench/ChemBench/确定性 NeuronBench 走 closed-form ODE likelihood;NeuronBenchStoch 走 bootstrap particle filter 估计(Algorithm 4)。这条决策的核心证据是反直觉的:deterministic likelihood on SDE gives confidently wrong answers——用错误的 likelihood 算 evidence 越自信越危险,所以 likelihood 实现必须按物理真实性选。

One-step myopic design。VoI 设计是单步贪心,不看未来多步。这是计算成本和性能的现实权衡——论文没在正文展开,附录 Figure 16 的 ablation 展示了 multi-step rollout 的成本收益曲线。

NeuronBench 故意设计 I_Z 通道避免 LLM 记忆。这是 benchmark 设计层面的关键决策——如果 LLM 直接调出标准 HH 模型,「发现」就退化成「回忆」,没有验证意义。

这件事为什么重要

Murphy 这篇论文在 2026 年这个时间点回答的是一个根本性命题:「LLM + Bayesian」怎么拼才不互相拖后腿。过去一年多 LLM-for-science 的论文大致分两派:

  • LLM 中心派:让 LLM 提议机制、用 LLM 评分、用 LLM 设计实验(Piriyakulkij et al. 2024, Abhyankar et al. 2026, Prystawski et al. 2026 等)——简单、通用,但 posterior 校准和 data efficiency 没有严格保障。
  • Bayesian 中心派:严格 Bayesian experimental design(Lindley 1956 路径)——有 posterior、有 VoI,但假设类是固定的,不能「想到新机制」。

MDA 是第三种拼法:LLM 做结构层(开放式假设创造),Bayesian 做参数层 + 设计层(严格 inference + 严格 VoI)。落到具体行为上是三条:

  • LLM 的「开放式创造」不再是 free-form——它每一步提议都被 SMC posterior 排序
  • Bayesian 的「固定假设类」不再是束缚——LLM 在 residual 超阈值时自动扩张假设空间
  • 两边的短板由对方补齐:LLM 提议被 Bayesian 校验,Bayesian 假设空间被 LLM 打开

这个范式可以拓展的方向:multi-step VoI(不是 one-step myopic)、learned summary statistics 的端到端优化、把 SMC 的 particle filter 替换成 diffusion-based posterior estimator(NeuronBenchStoch 是天然测试场)。

代码层面,Typert 严格清单 + adaptive-tempered SMC + bootstrap particle filter + threshold-gated LLM expansion 是这个范式能工程化落地的四个钉子,加上 likelihood 按物理真实性选,共五个。缺任何一项,要么 LLM 提议失去校验,要么 Bayesian 假设空间打不开。

2608.09696 版本的论文 + NeuronBench 开源 benchmark(https://github.com/murphyk/neuronbench)是这件事的当前最优解。下一版本会是什么——也许是 diffusion-based posterior 替换 particle filter(解决 SDE regime 的计算成本)、也许是 learned simulator 把 ODE 跑得更便宜——但「LLM 在结构层、Bayesian 在其它层」这条混合架构原则,大概率会留下。

采用建议:谁该从这里入手

想复现这条路线,从 murphyk/neuronbench 和论文附录 G 的 prompt 模板出发最快。具体路径:

  • 做 scientific discovery 工具的人:先在确定性 benchmark(ForceBench / ChemBench)上跑通 MDA 循环,再进 NeuronBenchStoch 处理 SDE regime——后者是 likelihood 实现最容易踩坑的地方。
  • 做 LLM-for-science baseline 的人:注意 Kabra et al. 2026 的 high-exact / low-symbolic pathology——RMSLE 低不代表机制对,评测时要把「数值误差」和「符号形式」分开报告。
  • 不急着采用的人:如果工作流是单假设验证(给定机制查参数),MDA 的结构扩张是多余开销,经典 Bayesian 实验设计就够;如果实验便宜、数据量大,data efficiency 的收益也不明显,不值得为 VoI 付复杂度。

维护指引:从 MDA 这篇论文读后续工作的几件事

引用网络。Bayesian experimental design 路线从 Lindley 1956 → Box & Hill 1967 → Chaloner & Verdinelli 1995 → Rainforth et al. 2024,论文里点名的「modern work scales it with amortised and gradient estimators」指 Foster et al. 2021 的 neural VoI。SMC 的两条技术源头:ModelSMC(Wahl et al. 2026)和 SMC-S(Piriyakulkij et al. 2024)——后者是 LLM-as-proposer 的 SMC 范式源头。

Benchmarks。三个 benchmark 的关系:DiscoverPhysics(Wiemann et al. 2026)是物理基线、ActiveSciBench-Chem(Kabra et al. 2026)是化学基线、NeuronBench(本文)是生物贡献。未来工作的候选 benchmark 包括 NewtonBench(Zheng et al. 2026)。

Stochastic 扩展。NeuronBenchStoch 是论文附录 F 的核心扩展,处理 latent SDE 的 intractable likelihood。这条路径的关键 reference 是 Cranmer et al. 2020(SBI)和 Fearnhead & Prangle 2012(learned summary stats)。Appendix E.8 给出了 1d CNN 学 summary stats 的实现。

替代 Bayesian 工具。论文用 SMC 是为了并行化和 particle-based 的天然优势;如果换 NUTS / HMC 也能算 evidence,但要权衡计算成本和粒子退化。MuZero / PPO 类 Bayesian RL 也能做实验设计,但只覆盖 reward optimization,不覆盖 model discovery。

为什么 NeuronBenchStoch 的 deterministic likelihood 错。直觉上「确定性 ODE likelihood 是 SDE 的近似,应该不会太错」。但当 SDE 噪声在参数空间产生多峰时,deterministic likelihood 会忽略这些多峰,导致 evidence 计算偏向「最拟合均值」的参数,反而错失正确的随机动力学模型。这是 SBI 领域的老教训,但 NeuronBenchStoch 给出了一个干净的具体例子。

参与讨论

使用 GitHub 登录。欢迎补充事实、异议与实践。