ICML 2025 · AI4Science 与跨界

MolJO:梯度引导贝叶斯流网络的结构化分子优化

连续坐标、离散类型与 backward correction 如何共同优化分子

ICML 2025 / PMLR 267 2026-09-20 结构基础药物设计 · 分子优化

MolJO 把联合梯度变成了可用的分子优化器,但 docking 代理、尺寸偏置与搜索预算意味着“分数更好”仍不是“药物更好”。

日期: 2026-09-20
领域: AI4Science 与跨界——结构基础药物设计、分子生成、连续—离散联合优化
核心论文: Keyue Qiu, Yuxuan Song, Jie Yu, Hongbo Ma, Ziyao Cao, Zhilong Zhang, Yushuai Wu, Mingyue Zheng, Hao Zhou, Wei-Ying Ma. Empower Structure-Based Molecule Optimization with Gradient Guided Bayesian Flow Networks. ICML 2025, PMLR 267:50645–50671.
一句话核心定位: MolJO 把原子坐标与原子类型投到可微的贝叶斯流参数空间,用联合梯度与滑窗式 backward correction 同时追求结合亲和力、药物相似性和可合成性;它把 CrossDocked2020 成功率从最强同协议基线的 39.4% 推到 51.3%,但证据仍是 docking 代理、单一基准和未归一化搜索预算,不能等价为真实药效突破。


0. 链接验证表与证据口径

下列链接均在 2026-09-20 实际访问。核心论文的精确数字只取自 arXiv v4 HTML/PDF 的表格、算法与正文;本报告写明“复算”的百分比仅由论文原报数字做四则运算。Docking 分数越负通常表示预测结合越强,QED、SA、Success Rate、Connected 与 Diversity 越高越好;Energy、Clash 与 RMSD 越低越好。对不同论文、不同采样数或不同 success 定义的数字不做无条件横比。

对象 会议/正式页 arXiv(版本、日期) 代码 数据/权重 项目页/视频 验证结论
核心:MolJO ICML 2025 / PMLR 267;OpenReview arXiv:2411.13280v4,初版 2024-11-20,v4 2025-06-05;HTML;PDF GitHub:MolCRAFT series;PMLR 旧链接会重定向到官方仓库 Hugging Face:GenSI/MolJO;CrossDocked 数据链接、property predictors、backbone checkpoint 与作者生成样本由仓库/HF 卡给出 独立项目页/视频未找到;代码仓库兼作项目总览 正式页注明 PMLR 267:50645–50671;arXiv 版本、代码、数据说明与模型卡均可访问
技术源头:Bayesian Flow Networks 正式会议页未找到;以 arXiv 为准 arXiv:2308.07037,2023-08-14 初版 未找到与论文绑定的官方仓库 未公开/不适用 未找到独立项目页/视频 提供用贝叶斯更新在参数空间建模连续与离散数据的基础形式
生成骨干:MolCRAFT ICML 2024 / PMLR 235 arXiv:2404.12141v4,v4 2024-05-28 同一官方仓库 HF 卡给出 backbone 与数据入口 独立视频未找到 在连续参数空间统一分子坐标与类型,是 MolJO 直接骨干
梯度前驱:TAGMol ICML 2024 ML4LMS OpenReview arXiv:2406.01650v1,2024-06-03 GitHub:MoleculeAI/TAGMol 仓库提供配置;独立权重页未找到 未找到独立项目页/视频 对扩散坐标显式引导、类型隐式响应,是“梯度只容易作用于连续模态”的直接问题来源
搜索前驱:DecompOpt ICLR 2024 Proceedings arXiv:2403.13829v1,2024-03-07 GitHub:bytedance/DecompOpt 仓库提供训练/优化代码;独立权重页未找到 未找到独立项目页/视频 以迭代生成—筛选—再训练做黑盒优化,强但昂贵,构成 MolJO “一次生成过程内优化”的对照

检索覆盖说明。 候选身份与正式发表状态以 PMLR、OpenReview 和 arXiv 为主;Hugging Face Daily Papers、Papers with Code 与 Semantic Scholar 用于交叉发现和查找代码/引用,但本轮未找到 MolJO 稳定的独立 Papers with Code 条目,Semantic Scholar 入口由 arXiv 的 References & Citations 区可达,故没有把聚合页当作数字证据。ACL Anthology 在本日 AI4Science 主题下用于排查生物医学 NLP 候选,但最终随机池以 ICML 2025 主会的可核验论文为主。

1. 候选池、排重与随机选择

先核对 ml-report-index.md,排除最近 14 天已深度覆盖的 Foundation Molecular Grammar(2026-09-06,分子神经符号语法)和 PemNN(2026-09-13,蛋白力谱物理归纳偏置),也避免重复前一日的鲁棒理论主题。随后冻结 24 篇候选,均为 2025 年 ICML 主会论文,距运行日不超过 24 个月:

  1. From Mechanistic Interpretability to Mechanistic Biology: Training, Evaluating, and Interpreting Sparse Autoencoders on Protein Language Models
  2. Training Flexible Models of Genetic Variant Effects from Functional Annotations using Accelerated Linear Algebra
  3. Raptor: Scalable Train-Free Embeddings for 3D Medical Volumes Leveraging Pretrained 2D Foundation Models
  4. Visual and Domain Knowledge for Professional-level Graph-of-Thought Medical Reasoning
  5. Breaking the Barrier of Hard Samples: A Data-Centric Approach to Synthetic Data for Medical Tasks
  6. MedRAX: Medical Reasoning Agent for Chest X-ray
  7. HealthGPT: A Medical Large Vision-Language Model for Unifying Comprehension and Generation via Heterogeneous Knowledge Adaptation
  8. TinyMIG: Transferring Generalization from Vision Foundation Models to Single-Domain Medical Imaging
  9. Distribution-aware Fairness Learning in Medical Image Segmentation From A Control-Theoretic Perspective
  10. Modalities Contribute Unequally: Enhancing Medical Multi-modal Learning through Adaptive Modality Token Re-balancing
  11. Multimodal Medical Code Tokenizer
  12. BoxLM: Unifying Structures and Semantics of Medical Concepts for Diagnosis Prediction in Healthcare
  13. LangDAug: Langevin Data Augmentation for Multi-Source Domain Generalization in Medical Image Segmentation
  14. GraphCL: Graph-based Clustering for Semi-Supervised Medical Image Segmentation
  15. MMedPO: Aligning Medical Vision-Language Models with Clinical-Aware Multimodal Preference Optimization
  16. MedXpertQA: Benchmarking Expert-Level Medical Reasoning and Understanding
  17. Hierarchical Graph Tokenization for Molecule-Language Alignment
  18. Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment
  19. Empower Structure-Based Molecule Optimization with Gradient Guided Bayesian Flow Networks
  20. Compositional Flows for 3D Molecule and Synthesis Pathway Co-design
  21. Protriever: End-to-End Differentiable Protein Homology Search for Fitness Prediction
  22. Protein Structure Tokenization: Benchmarking and New Recipe
  23. ReQFlow: Rectified Quaternion Flow for Efficient and High-Quality Protein Backbone Generation
  24. All-atom inverse protein folding through discrete flow matching

实际执行 shuf -i 1-24 -n 1,输出 19。MolJO 有主会正式页、v4 全文、代码、数据说明、权重与完整补充实验,资料足以满足深读,因此重抽次数为 0。这个过程不按热度挑选;所谓“值得看”是在抽签后基于证据解释,而不是在抽签前主观筛热门。

2. Motivation:为什么“让分子更好”比“生成一个分子”更难

结构基础分子生成不是纯图像生成。一个候选配体同时包含连续的三维原子坐标与离散的元素/原子类型;优化目标又常常彼此拉扯:Vina 亲和力更强并不保证药物相似性 QED、更不保证可合成性 SA,也不保证几何连通。若只对坐标求梯度,模型可能把原子推到有利于代理评分的位置,却没有相应地改变类型;若对离散类型做硬 argmax,梯度又断掉;若用强化学习或反复生成—筛选—再训练,搜索成本会迅速增长。

论文把痛点量化在同一个 CrossDocked2020 协议上。TAGMol 已是梯度引导方法,但 Success Rate 只有 11.1%;迭代式 DecompOpt 是更强的搜索基线,Success Rate 为 39.4%;MolJO 达 51.3%(主文表 1,HTML 行 279–297)。因此相对 TAGMol,MolJO 绝对提升 40.2 pp、复算相对提升 362.16%;相对最强同协议基线 DecompOpt,则提升 11.9 pp、复算 30.20%。它不是只把 Vina 做得更负:相对 DecompOpt,Vina Dock 从 -7.63 到 -9.05,绝对幅度改善 1.42 kcal/mol(按绝对值复算 +18.61%),SA 从 0.73 到 0.78(+0.05,复算 +6.85%),QED 同为 0.56。

但这组成绩也说明问题的本质是多目标而非单榜单。MolJO 的 Diversity 为 0.66,低于 TAGMol 0.69、IPDiff 0.74、DecompOpt 0.63则略高;若从每个 pocket 生成 10 个样本再由规则挑最优,MolJO† Success Rate 从 51.3% 升至 70.3%,但 Diversity 从 0.66 降至 0.61(表 1)。这相当于成功率 +19.0 pp(复算 +37.04%),同时多样性 -0.05(复算 -7.58%)。性能的很大一部分来自搜索预算,不能把 top-of-10 与单样本质量混成一个数字。

3. 论文故事线、核心假设与权衡

论文故事线可压缩成四步。第一,继承 MolCRAFT 的 Bayesian Flow Network(BFN),把离散原子类型表示为概率参数,把坐标表示为高斯参数;这样“离散样本”背后有一个连续可微的信念状态。第二,在该参数空间训练亲和力、QED 与 SA 能量预测器,对坐标参数和类型概率共同求梯度,避免 TAGMol 的“坐标显式、类型隐式”。第三,指出一次局部梯度仍容易早熟,于是在最近 (k) 个历史状态内回退并重采样,形成 backward correction(B.C.),用滑窗长度调节 exploitation 与 exploration。第四,证明连续坐标引导保持 SE(3) 等变,并把同一机制扩展到 R-group redesign、fragment growing、scaffold hopping、联合目标和大 pocket OOD。

背后有三条关键假设。其一,能量预测器 (E_phi( heta,t,p)) 在 BFN 参数轨迹附近给出的梯度方向与真实属性改善一致;论文验证的是代理分数而非湿实验。其二,连续参数化足以让类型概率梯度表达离散化后的选择变化;实际采样仍要经过 categorical 抽样,梯度与最终原子类型不是一一对应。其三,滑窗回退会探索到更好的局部盆地而不会破坏化学有效性;附录的 (k) 扫描支持一个经验稳定区间,却没有全局最优保证。

核心权衡是三组:属性—有效性,更强亲和力引导会牺牲连通性;开发—探索,B.C. 窗口过短积累局部误差,过长增加回退成本;质量—预算,top-of-10 明显提高成功率,但其成本与单样本方法不再等价。论文的价值正在于没有把这些权衡藏起来,附录给出了可读的相变;不足则是主摘要的“state of the art”没有把这些边界写在同一句话里。

4. 方程级形式化与算法

4.1 BFN 的可微参数空间

对数据 (x) 的潜在参数 ( heta),BFN 在时间 (t) 上维护发送分布、贝叶斯更新分布与输出分布。论文式 (1)–(3)(HTML 行 145–155)将接收噪声 (y) 后的参数更新写成

[ p_F(x;t)=E_{yp_S(yx;(t))},!(-h(_0,y,(t))), ]

并以神经网络 ((,t,p)) 预测洁净数据或下一步分布;(p) 是 pocket 条件。对坐标,参数可视作高斯均值/精度;对类型,参数是类别概率向量。关键不是把离散原子“假装连续”,而是在它被采样前对概率单纯形求导。

4.2 联合梯度引导

给定属性能量 (E(,t,p)) 与引导尺度 (s),论文式 (4)–(5)(行 170–177)以指数倾斜近似条件转移:

[ p(‘,p,E)p(’,p). ]

对连续坐标,式 (6)–(7)(行 180–189)把均值沿 (-_{}E) 方向偏移;对离散类型,式 (8)–(9)(行 191–203)在类别概率上作对应修正,再归一化。因为能量对坐标只依赖 SE(3) 不变量/等变表示,命题 4.4(行 204–210)说明更新保持旋转平移等变。联合目标取若干属性梯度的平均;这简单、可解释,但不是 Pareto 自适应权重。

4.3 Backward correction

单步引导会把当前误差一路传下去。B.C. 在第 (i) 步保存历史状态,回看最近 (k) 步,从历史后验重新采样并用当前属性梯度校正。论文式 (11)–(14)(行 243–258)可概括为

[ i^{} =!({i-k}, y_{i-k+1:i}, -s_E(_i,t_i,p)), ]

其中离散分支在实现上引入近似导数 ((1-)) 以避免数值不稳定。该替代是工程启发式,不是由精确 categorical Jacobian 唯一推出;它是方法有效性与理论完整性之间最值得复现实验的一处缝隙。

4.4 伪代码

输入:protein pocket p;预训练 MolCRAFT;属性能量 Eφ;步数 n=200;尺度 s=50;窗口 k=130
初始化:坐标信念 θx,0、类型概率 θv,0;历史缓存 H←∅
for i = 1..n:
    t ← i/n
    x̂, v̂ ← Φ(θx,i-1, θv,i-1, t, p)
    gx ← ∂Eφ/∂θx ; gv ← ∂Eφ/∂θv
    θx,i ← ContinuousBFNUpdate(θx,i-1, x̂, -s·gx)
    θv,i ← DiscreteBFNUpdate(θv,i-1, v̂, -s·gv)
    H.push(θx,i, θv,i)
    if correction step:
        j ← max(0, i-k)
        (θx,i, θv,i) ← ReconstructAndResample(H[j:i], gx, gv)
采样最终原子类型,解码坐标,检查连通性与化学有效性
输出:分子候选;可选地每 pocket 采 10 个再按成功条件挑最优

5. 关键机制与工程细节

层 论文/仓库原报 为什么重要 可复用注意点
生成骨干 直接加载预训练 MolCRAFT,不为 MolJO 重新微调;坐标噪声 (_1=0.03),类型精度 (_1=1.5)(附录实现,行 517–538) 把收益集中到 test-time guidance,而非更大生成器 迁移到新分子域前先校准骨干密度覆盖;域外梯度可能沿低密度区逃逸
网络 kNN 图 (k=32),9 层,隐藏维 128,16 heads;原子/口袋特征共同编码(行 517–538) 同一表示同时服务坐标和类型能量 pocket 截取半径、邻居数与大蛋白内存应单独消融
属性预测器 MSE 训练 Vina、QED、SA 能量;Adam,lr (5^{-4}),batch 8;单 RTX 3090 小于 8 小时、至多 5 epochs(行 517–538) 梯度质量由预测器决定 必须报告 predictor 的 OOD 校准和梯度平滑度,不能只报最终采样分数
采样 200 steps;位置与类型 guidance scale 均为 50;多目标梯度简单平均(仓库 sample 命令;论文附录) 推理期不需要 RL/再训练 不同目标量纲会让平均梯度隐式偏权;应先归一化梯度范数
B.C. 最终 (k=130);附录称 (k>50) 改善,(k(50,200]) 较稳(行 744–750) 回退是主要性能来源 k 与采样步数耦合;换成 50/100/1000 steps 时应按时间比例重扫
数据 CrossDocked2020:去掉 RMSD>1Å,蛋白序列相似度 <30% 聚类;100,000 poses 训练、100 test proteins;主结果每 pocket 100 分子(行 267–275) 这是严格但单一的 benchmark 序列去重不等于口袋几何/配体骨架无泄漏,应补 pocket similarity 与 scaffold split
成功条件 Vina Dock < -8.18、QED >0.25、SA >0.59(行 267–275) 单个 Success Rate 汇总三目标 阈值改变可重排模型;应报告三维 Pareto 面与阈值曲线

仓库与论文存在一处需诚实标注的配置差异:HF/README 的训练示例给 epochs 15,而论文附录描述 property predictor 至多 5 epochs;两者可能对应通用默认配置与论文实际运行,但公开材料没有逐 checkpoint 的哈希—配置映射。因此,权重虽公开,训练级复现仍应固定 commit、checkpoint hash 与命令行,而不能只执行 README 默认值。

6. 数据集、基准与指标

CrossDocked2020 把蛋白 pocket 与配体复合物对齐后切分,是 SBDD/SBMO 常用基准。论文主评估有五类场景:标准从头优化、R-group redesign、fragment growing、scaffold hopping、大 pocket OOD;另补联合目标、构象质量和推理耗时。Vina Score 通常在未局部优化构象上评估,Vina Min 经局部最小化,Vina Dock 经过 docking 搜索;三者不是同一变量。QED 是药物相似性的经验组合,SA 是合成可达性的学习/规则代理,Diversity 衡量样本差异,Success 是三个阈值的逻辑与。

这套指标能回答“在相同 docking/代理评分流水线上,哪种生成策略更容易同时过线”,不能回答临床活性、选择性、毒性、ADMET、合成收率或真实结合自由能。尤其 docking 分数常随分子尺寸增加而改善;主实验虽对 atom count 先验有所控制,大 pocket OOD 仍需同时看平均原子数。

7. 完整主结果矩阵

7.1 标准 CrossDocked2020:十五方法、七指标

下表逐项来自核心论文表 1(HTML 行 279–297);斜杠前后为 mean/median。Vina 越负越好,其余越高越好。— 表示论文未报告,不补造。

方法 Vina Score ↓ Vina Min ↓ Vina Dock ↓ QED ↑ SA ↑ Div ↑
Reference -6.36/-6.46 -6.71/-6.49 -7.45/-7.26 0.48 0.73 —
AR -5.75/-5.64 -6.18/-5.88 -6.75/-6.62 0.51 0.63 0.70
GraphBP — — -4.80/-4.70 0.43 0.49 0.79
Pocket2Mol -5.14/-4.70 -6.42/-5.82 -7.15/-6.79 0.57 0.76 0.69
FLAG 45.85/36.52 9.71/-2.43 -4.84/-5.56 0.61 0.63 0.70
DiffSBDD -1.44/-4.91 -4.52/-5.84 -7.14/-7.30 0.47 0.58 0.73
TargetDiff -5.47/-6.30 -6.64/-6.83 -7.80/-7.91 0.48 0.58 0.72
DecompDiff -5.19/-5.27 -6.03/-6.00 -7.03/-7.16 0.51 0.66 0.73
IPDiff -6.41/-7.01 -7.45/-7.48 -8.57/-8.51 0.52 0.59 0.74
MolCRAFT -6.55/-6.95 -7.21/-7.14 -7.67/-7.82 0.50 0.67 0.70
AutoGrow4 — — -8.99/-9.00 0.46 0.76 0.47
RGA — — -8.01/-8.17 0.57 0.71 0.41
DecompOpt -5.75/-5.97 -6.58/-6.70 -7.63/-8.02 0.56 0.73 0.63
TAGMol -7.02/-7.77 -7.95/-8.07 -8.59/-8.69 0.55 0.56 0.69
MolJO -7.52/-8.02 -8.33/-8.34 -9.05/-9.13 0.56 0.78 0.66
MolJO†(top-of-10) -8.54/-8.81 -9.48/-9.09 -10.50/-10.14 0.67 0.79 0.61

MolJO† 的更好数字以 10 倍候选池换来,不标为同预算最优。相对梯度基线 TAGMol,MolJO 的 Success +40.2 pp(+362.16%),SA +0.22(+39.29%),Vina Dock 均值绝对幅度改善 0.46(5.36%);相对 DecompOpt,Success +11.9 pp(+30.20%),Vina Dock 改善 1.42(18.61%),SA +0.05(6.85%)。

7.2 三类约束优化

表 2(行 299–313)只报告 Success Rate。MolJO 对最强基线 MolCRAFT 的绝对/相对提升分别为:redesign +6.3 pp/+27.75%,growing +12.8 pp/+64.32%,hopping +9.0 pp/+61.64%。

方法 R-group redesign ↑ Fragment growing ↑ Scaffold hopping ↑
TargetDiff 18.9% 11.2% 6.2%
TAGMol 19.2% 11.8% 6.9%
MolCRAFT 22.7% 19.9% 14.6%
MolJO 29.0% 32.7% 23.6%

7.3 大 pocket OOD

表 3(行 331–338)显示 MolJO 的 OOD Success 为 64.2%,高于 DecompOpt 52.5%(+11.7 pp,复算 +22.29%),Vina Dock 为 -10.53,SA 0.72。但 MolJO 平均 30.0 个原子,Reference 为 22.8,增加 7.2(+31.58%);DecompOpt 甚至为 32.9。由于 docking 评分对尺寸敏感,这一表支持“在该 atom-count 分布与代理指标下更强”,不能单独证明单位重原子的化学效率更高。

方法 Vina Dock ↓ QED ↑ SA ↑ Success ↑ 平均原子数
Reference -7.45 0.48 0.73 25.0% 22.8
DecompDiff -8.39 0.45 0.61 24.5% 29.4
DecompOpt -9.01 0.48 0.65 52.5% 32.9
MolCRAFT -9.25 0.46 0.62 36.6% 29.4
MolJO -10.53 0.50 0.72 64.2% 30.0

同协议 Success Rate 与搜索预算

MolJO 单配置领先 TAGMol 与 DecompOpt;top-of-10 进一步提升,但不是同预算比较。

单配置/原报 MolJO 系列
TAGMol、DecompOpt、MolJO 与 MolJO top-of-10 的成功率 四个柱分别为 11.1%、39.4%、51.3% 和 70.3%。最后一项使用每个 pocket 十个候选后择优,不是同预算。 0%20% 40%60% 80% TAGMol 11.1% DecompOpt 39.4% MolJO 51.3% MolJO top-of-10 70.3% TAGMolDecompOpt MolJOMolJO† 11.1%39.4% 51.3%70.3% Success Rate;越高越好。MolJO† 为每 pocket 10 个候选择优。
数据:MolJO 论文表 1(arXiv v4,HTML 行 279–297);单位:百分比。

8. 逐层消融:真正的收益来自哪里

表 4(行 343–353)把无梯度/有梯度与三种采样器拆开。下面保留 mean 指标;括号是论文原表给出的相对变化,不另推断统计显著性。

采样/引导 Vina Score ↓ Vina Min ↓ Vina Dock ↓ QED ↑ SA ↑
Vanilla,无梯度 -5.23 -6.30 -7.37 0.46 0.62
Vanilla,有梯度 -5.47(+4.6%) -6.29(-0.2%) -7.49(+2.2%) 0.46(0) 0.62(0)
SDE,无梯度 -6.62 -7.31 -8.22 0.51 0.65
SDE,有梯度 -7.11(+7.4%) -7.76(+6.1%) -8.39(+2.1%) 0.50(-1.9%) 0.68(+4.6%)
B.C.,无梯度 -6.50 -7.03 -7.95 0.49 0.69
B.C.,有梯度(MolJO) -7.52(+15.7%) -8.34(+18.6%) -9.11(+14.6%) 0.56(+14.3%) 0.77(+11.6%)

层层读法是:第一,给 Vanilla 直接加梯度几乎没解决多目标问题,QED/SA 不动,Vina Min 还略退;因此“有梯度”不是充分条件。第二,SDE 的随机性扩大探索,亲和力明显改善,却让 QED 略降。第三,只有 B.C.+梯度在五项上同时改善,说明回退重采样不是装饰,而是把局部方向变成可持续搜索的核心机制。

表 11(行 727–740)进一步拆 joint guidance。在 affinity 目标下,去掉类型梯度,Vina Score 从 -7.74 退到 -7.13;去掉坐标梯度退到 -6.61。坐标梯度贡献更大,但类型梯度仍带来 0.61 的绝对幅度改善;在 QED 目标下,完整模型 QED 0.66,去掉类型梯度降到 0.52,而去掉坐标仍为 0.65。这提供了清楚的因果画像:亲和力更依赖几何,QED 更依赖类型;所谓“joint”不是两个冗余梯度。

9. 超参敏感性、相变与统计显著性

9.1 引导尺度的可见相变

附录表 13(行 765–787)是论文最重要却最容易被摘要忽略的结果。Affinity scale 从 20 提到 50,Vina Min 由 -7.84 改善到 -8.39,Connected 仅从 0.98 降到 0.90;再提到 100,Vina Min 到 -9.87,却使 Connected 0.90→0.55,绝对 -35 pp、复算相对 -38.89%,SA 也从 0.68 降到 0.63。属性优化在 50–100 之间出现明显的“代理更好、结构更坏”相变。

Affinity scale Vina Score ↓ Vina Min ↓ QED ↑ SA ↑ Connected ↑
0.1 -6.28 -7.17 0.50 0.70 0.96
1 -6.24 -7.27 0.50 0.69 0.96
10 -6.69 -7.46 0.51 0.70 0.97
20 -7.03 -7.84 0.51 0.70 0.98
50(主配置) -7.64 -8.39 0.53 0.68 0.90
100 -9.33 -9.87 0.55 0.63 0.55

QED 引导则在 scale 10–50 达到 0.66,scale 100 回落到 0.62、SA 降到 0.60;SA 引导在 scale 20–50 达 0.79,继续放大不再有收益。可操作结论不是“50 永远最佳”,而是每个属性都有不同的饱和/崩塌点,统一 scale 隐含了未经验证的量纲假设。

9.2 B.C. 窗口

论文文字报告 (k>50) 后多指标平衡改善,并选 (k=130),在 (k(50,200]) 内相对稳健(行 744–750)。但论文未报告完整数值表或方差,只有图/文字趋势,因此本报告不从图上抄伪精确值。我们能推出“太短窗口会累积误差、130 在本 200-step 设置可用”,不能推出“130 对所有步数、口袋大小与目标都最优”。

9.3 显著性

附录表 12(行 751–756)在 (N=1000) 上做独立 t-test。MolJO 相对 Vanilla 的 p 值依次为 Vina Score (2.63^{-13})、Vina Min (3.31^{-31})、Vina Dock (2.79^{-35})、SA (8.10^{-115})、QED (1.98^{-26});相对 SDE 为 (2.55{-19})、(6.48{-19})、(7.84{-4})、(2.10{-50})、(1.82^{-12})。表 6还给连续指标 95% CI,例如 MolJO Vina Dock -9.048±0.083、QED 0.556±0.003、SA 0.775±0.003(行 644–657)。

但 Success Rate、三类约束任务、OOD 与 top-of-10 未报告置信区间或配对检验;多指标多比较也未说明校正。因此可以说连续代理指标的改进在该样本上很难由均值噪声解释,不能说所有任务的 11.9 pp、6.3 pp 等差距都已统计显著。

10. 构象质量与系统成本

表 14(行 804–817)显示 MolJO energy 163、clash 6.72、RMSD<2Å 比例 43.5%、connected 97.3%。它的 RMSD 通过率略高 MolCRAFT 的 42.4%,energy 低于 MolCRAFT 196,但仍不及 Reference energy 114,clash 也高于 Reference 5.46。这意味着优化没有把几何彻底破坏,却也没有达到晶体构象分布。

方法 Energy ↓ Clash ↓ RMSD<2Å ↑ Connected ↑
Reference 114 5.46 34.0% 100%
Pocket2Mol 186 6.22 31.3% 96.3%
TargetDiff 1208 10.67 31.0% 90.4%
MolCRAFT 196 6.91 42.4% 96.7%
TAGMol 2058 7.41 37.2% 92.0%
MolJO 163 6.72 43.5% 97.3%

推理耗时表 16(行 839–844)在单 RTX 3090、batch 5、10 个 protein 上给 mean±sd:MolJO 146±11 s,TAGMol 667±69 s,RGA 458±43 s,AutoGrow4 2586±360 s,DecompOpt 11714±1115 s。复算 MolJO 分别快 4.57×、3.14×、17.71×、80.23×。这是实际工程贡献:性能不依赖 30 轮黑盒再训练。不过论文未报告吞吐随 batch、口袋大小、样本数的曲线,也未把 predictor 训练成本、docking 后处理与 top-of-10 选择全部摊入端到端成本。

11. 相关工作:问题来源与技术传递链

11.1 Bayesian Flow Networks:统一参数空间

解决什么。 BFN 用贝叶斯更新把噪声注入和生成写在分布参数空间,对连续变量与离散变量都能给出统一流程(arXiv:2308.07037)。
遗留缺口。 原始 BFN 解决“如何生成”,没有回答 pocket 条件下如何用属性梯度同时改坐标与原子类型。
MolJO 如何回应。 它在 BFN 的中间信念状态上训练能量并求导,使离散类型不需要穿过最终硬采样。
设计如何传递。 “先把异构模态映射到可微信念参数,再做联合引导”可迁移到蛋白序列—结构、材料成分—晶格、反应物—路径的混合生成。

11.2 MolCRAFT:从统一生成到可控优化

解决什么。 MolCRAFT 在连续参数空间联合生成坐标和类型,ICML 2024 正式页报告 reference-level Vina Score -6.59 kcal/mol,并较强基线改善 0.84。
遗留缺口。 它是 pocket-conditioned generator,不直接让用户在推理时指定“更高 affinity/QED/SA”。
MolJO 如何回应。 直接复用预训练 MolCRAFT,把新训练成本压到属性预测器;这使比较更像控制器增量,而非换一个更大骨干。
设计如何传递。 先预训练无条件/条件“可行域模型”,再外挂小型可微 property head,是 AI4Science 中比端到端重训更经济的路线。

11.3 TAGMol:连续坐标可引导,离散类型仍滞后

解决什么。 TAGMol 把目标函数梯度加到扩散采样中,论文表 1报告 Vina Score -7.02、Vina Min -7.95、Vina Dock -8.59、QED 0.55、SA 0.56、Diversity 0.69;它证明无需 RL 也能 test-time 优化。
遗留缺口。 其核心引导显式作用于坐标,类型变化主要依靠生成网络的耦合,且 SA 明显偏低;不同论文的 hit/success 定义也不完全一致。
MolJO 如何回应。 对类型概率直接求梯度,并用 B.C. 防止局部引导积累错误。核心论文同协议重评 TAGMol Success 11.1%,MolJO 51.3%。
设计如何传递。 对混合模态不能只优化“好求导的那一半”;需要明确每一模态对每个目标的消融。

11.4 DecompOpt:强黑盒搜索,但计算极重

解决什么。 DecompOpt 通过分解生成、性质筛选和迭代优化完成 SBDD;原论文选定 Opt Prior 时报告 Success 52.5%、Vina Dock -8.98、Diversity 0.60,并使用 30 rounds。
遗留缺口。 迭代再训练/筛选带来 11714±1115 s 的推理级成本(MolJO 论文表 16复测),多目标权衡也没有统一最佳方案。
MolJO 如何回应。 在单次 200-step 采样轨迹内做可微优化,达到 146±11 s;核心论文自己的统一协议下 DecompOpt Success 为 39.4%,MolJO 51.3%。
设计如何传递。 若科学代理可微或可蒸馏成可微 surrogate,轨迹内控制往往比外循环黑盒搜索更便宜;若 surrogate 不可靠,则黑盒验证仍不可省。

11.5 非线性技术演进树

                         ┌─ TAGMol:扩散采样 + 坐标梯度 ──┐
BFN:参数空间贝叶斯流 ─┤                                ├─ MolJO:坐标+类型联合梯度
                         └─ MolCRAFT:连续参数空间 SBDD ─┘          │
                                                                  ├─ backward correction
黑盒性质优化 ─ DecompOpt:生成→筛选→再训练(30轮) ────────────────┘
                                                                  │
                    ┌─ R-group redesign / growing / hopping ──────┤
                    ├─ affinity + QED + SA 多目标 ────────────────┤
                    └─ 大 pocket OOD / top-of-N 搜索 ─────────────┘
                                      ↓
           下一转折:可微代理的校准、Pareto 权重、湿实验闭环与合成路线约束

演进动力有两次转折。第一次从“生成合法分子”转向“生成时满足目标”,梯度比 RL/黑盒搜索更直接;第二次从“只推坐标”转向“联合控制混合模态”,BFN 参数空间成为接口。下一次转折不会只是再多一个 docking 指标,而应把代理不确定性、合成路径与实验反馈放进闭环。

11.6 跨论文对标

工作 生成骨干/搜索 坐标引导 类型引导 探索机制 多目标 统一协议 Success 核心代价
MolCRAFT BFN 无显式属性梯度 无显式属性梯度 原生采样 否 26.8% 可控性弱
TAGMol Diffusion 显式 隐式耦合 随机扩散 有 11.1% SA 与离散控制弱
DecompOpt 分解生成 + 30轮迭代 黑盒 黑盒 外循环筛选 有 39.4% 11714±1115 s
MolJO BFN + B.C. 显式 显式 滑窗回退重采样 有,梯度平均 51.3% 代理偏置与 scale 相变

12. 真正贡献、可复用设计与迁移潜力

真正贡献一:离散引导的接口。 MolJO 的关键不是又造一个 docking model,而是把 categorical 类型留在概率参数空间直至最后采样,让属性梯度能改“是什么原子”而不只改“原子在哪里”。这一接口适合任何连续—离散混合科学对象。

真正贡献二:B.C. 是优化器而非采样花活。 消融显示 Vanilla+gradient 几乎无效,SDE+gradient 有限,而 B.C.+gradient 五指标齐升。换言之,科学生成中的 test-time guidance 需要显式的错误恢复/探索机制。

真正贡献三:效率可落地。 相对 DecompOpt 80.23× 的单卡耗时差,把“每个新 pocket 做一轮昂贵适配”变成“复用骨干和属性头执行一次受控采样”。这对虚拟筛选规模比单项均值更重要。

可复用迁移有三类。蛋白设计可把氨基酸类型视为离散模态、骨架坐标视为连续模态;晶体/材料可把元素占位与晶格坐标联合引导;反应规划可把离散反应模板与连续条件/能垒联合引导。迁移时必须同时迁移三样东西:可微参数化、属性代理的不确定性估计、以及类似 B.C. 的探索恢复,而不是只复制一条 gradient update。

13. 论文自述限制

  1. 多目标冲突(附录联合目标,行 679–681)。 作者指出 QED+SA 的联合引导会损害 affinity,最终采用简单梯度平均。因为不同目标梯度量纲与夹角未被校准,所以当前联合优化不能保证 Pareto 最优,只能说给出一个经验折中。
  2. 窗口误差与超参依赖(方法式 11–14;附录行 744–750)。 作者指出太小 (k) 会造成误差积累,最终选 130。因为没有跨采样步数/骨干的完整数表,所以窗口稳定性只能限定在当前 200-step CrossDocked 设置。
  3. 评估与潜在滥用(附录构象讨论行 795–797;Impact 行 371–373)。 作者承认 Auto‑Dock poses/RMSD 评估有局限,并指出方法可能被用于有害物质设计、虽需要湿实验资源。因为所有主结论仍停留在计算代理,所以真实安全性与活性都未被验证。

14. 独立批判:每一点都用证据推出边界

  1. [构念效度] 因为主结论依赖 Auto‑Dock Vina、QED 与 SA,且没有生化 assay、自由能微扰或合成实验,所以51.3% Success 只能代表通过三个计算阈值,不能推出真实结合、选择性、毒性或可合成收率。
  2. [代理奖励/分布外优化] 因为 affinity scale 从 50 增至 100 时 Vina Min -8.39→-9.87,却让 Connected 0.90→0.55、SA 0.68→0.63(表 13),所以能量梯度存在典型 reward hacking 区域;继续提高代理分数会离开化学可行域。
  3. [规模混杂] 因为大 pocket OOD 中 MolJO 平均原子数 30.0、Reference 22.8(+31.58%),而 docking 分数会受接触面积/尺寸影响,所以-10.53 的优势不能完全归因于更聪明的化学设计;应补 ligand efficiency、每重原子得分与匹配 atom-count 对照。
  4. [搜索预算公平] 因为MolJO† 由每 pocket 10 个候选择优,Success 51.3→70.3%但 Diversity 0.66→0.61,而基线未统一成相同 oracle 调用数,所以70.3%不能作为同预算 SOTA;应画 success–oracle calls 曲线。
  5. [统计完整性] 因为t-test 与 95% CI 只覆盖部分连续指标,Success、三类约束任务、OOD 和 top-of-10 没有区间,且多重检验未校正,所以可以信连续分数差异很稳定,却不能断言所有成功率差距都超出 pocket 间方差。
  6. [内部一致性/复现] 因为主表 1与附录表 6对 DecompOpt、MolCRAFT 的部分均值不一致(例如 DecompOpt Vina Score -5.75 对 -4.839),README 的训练示例又是 15 epochs、论文 property head 写至多 5 epochs,所以复现者必须固定“哪一张表/哪一份 checkpoint/哪条命令”,不能把公开仓库默认配置自动视为论文配置。
  7. [离散梯度理论] 因为实现为稳定性采用 ((1-)) 近似离散导数,而论文没有给这一替代的偏差界或收敛保证,所以“联合离散引导”目前是有效工程机制,而非已闭合的理论最优更新;需要与 exact simplex Jacobian、Gumbel-softmax、straight-through 做等预算对照。
  8. [外推范围] 因为训练/主测试只来自 CrossDocked2020 的 100,000 poses 与 100 test proteins,序列相似度阈值虽为 30%但未报告 pocket geometry 与配体 scaffold 泄漏,所以不能推出跨靶点家族、共价配体、金属口袋或蛋白柔性场景同样有效。

15. 开放挑战:可直接执行的 head-to-head 实验空白

15.1 理论

  1. 离散梯度估计器对照。 固定 MolCRAFT checkpoint、property head、200 steps、每 pocket 样本数与随机种子,比 ((1-))、exact simplex Jacobian、Gumbel-softmax、straight-through 四种更新;报告属性增益、connected、梯度方差、wall-clock 与最终类型翻转率。若 B.C. 只对某一估计器有效,就能分离“回退”与“梯度近似”的贡献。
  2. B.C. 的预算等价实验。 令所有方法拥有相同 200 次能量/网络调用,比较 B.C. (k{25,50,100,130,200})、SDE、多起点 beam 与 MCMC rejuvenation;主指标用 success–calls AUC,而非只比最后一点。
  3. 多目标 Pareto 机制。 固定总梯度范数,对 simple average、GradNorm、MGDA、PCGrad、uncertainty weighting 做 head-to-head;报告 affinity/QED/SA 三维 hypervolume、连通性与每个目标的最坏退化,验证“平均梯度”是否真是合理折中。

15.2 工程与应用

  1. 代理—物理双重盲测。 在同一 pocket/scaffold split 上,让 MolJO、TAGMol、DecompOpt 各提交固定数量候选,先用未参与训练的 docking 引擎/FEP 重排,再对 top 候选做盲法 assay;比较 Vina 增益转化为真实 (K_d/IC_{50}) 增益的校准曲线。
  2. 尺寸匹配对照。 对每种方法做相同 heavy-atom count、相同 formal charge、相同 rotatable bonds 的匹配采样,报告 Vina per heavy atom、ligand efficiency、QED、SA 与 success;这能直接检验 OOD -10.53 中有多少来自尺寸。
  3. 全成本审计。 在同一 GPU/CPU、相同每 pocket 目标成功数下,比较 MolJO、TAGMol、DecompOpt 的 predictor 训练、采样、docking、top-N 选择、失败重试和 I/O;报告能耗、P50/P95 time-to-valid 与每个成功分子的 GPU-hour。

15.3 新兴方向

  1. 不确定性感知的引导。 给 property head 加 ensemble 或 conformal uncertainty,在高不确定区把 scale 自动衰减;与固定 50、100 做 head-to-head,观察是否能保留 -9 级 docking 同时避免 connected 在 100 scale 崩到 0.55。
  2. 合成路线共同生成。 将 MolJO 与 CGFlow 类“分子+合成路径”共同设计接合,对仅 SA 引导与显式反应路径生成比较:最终 wet-lab 可合成率、步骤数、成本、yield 与属性 Pareto 面。若显式路径只提升 SA 代理而不提升真实合成率,也能及时否证。
  3. 蛋白柔性闭环。 固定配体预算,比较刚性 pocket MolJO、side-chain ensemble、诱导契合 docking 与短 MD/FEP 重评分;按蛋白家族分层报告。该实验能检验 BFN 坐标优化是在学真实互补性,还是适配单一刚性口袋。

16. 近期值得跟踪的 6 篇工作

  1. MolPilot / Piloting Structure-Based Drug Design via Modality-Specific Optimal Schedule(ICML 2025):PMLR|arXiv:2505.07286。它不再主攻属性梯度,而优化混合模态噪声调度;正式材料报告 95.9% PoseBusters pass,适合与 MolJO 组合成“更强骨干 + 更稳引导”。
  2. Compositional Flows for 3D Molecule and Synthesis Pathway Co-design(ICML 2025):arXiv:2504.08051|代码。它把分子和合成路径放进同一个 compositional flow,报告 4.2× sampling efficiency;正面回应 MolJO 的 SA 只是代理而非路线证据。
  3. LDMol: Text-Conditioned Molecule Diffusion Model Leveraging Chemically Informative Latent Space(ICML 2025):PMLR|arXiv:2405.17829。它探索语言条件的化学 latent;与 pocket-conditioned MolJO 互补,可测试自然语言目标是否能转成可靠联合梯度。
  4. Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment(ICML 2025):PMLR|arXiv:2506.10186。它挑战“必须全程等变”的成本假设;值得和 MolJO 的 SE(3) 保证做精度—吞吐 head-to-head。
  5. Multi-Fidelity Latent Active Learning for Molecular Property Discovery(ICML 2025):PMLR。正式页报告平均 binding free energy 改善 50%;其多保真主动学习可作为 MolJO docking surrogate 与昂贵物理验证之间的预算分配器。
  6. Hierarchical Graph Tokenization for Molecule-Language Alignment(ICML 2025):arXiv:2406.14021|OpenReview。它以层级 graph tokens 对齐分子与语言,若把这些 token 接到 MolJO 类型参数,可检验“语义级编辑”是否比逐原子梯度更稳。

17. 结论

MolJO 最值得看的不是 51.3% 这个孤立数字,而是一条可复用的控制链:BFN 把连续坐标与离散类型放到统一可微信念空间,联合梯度负责定向,backward correction 负责从局部误差中恢复。主表、类型/坐标消融和 80.23× 相对 DecompOpt 的耗时差共同支持这一设计确实有效。

最关键的科学判断则更克制:MolJO 把计算代理上的结构化分子优化显著推进了,却没有跨过代理到实验的鸿沟。Scale 100 时连接率从 0.90 崩到 0.55,是一条清楚的红线:当生成器学会讨好评价器,分数继续变好并不等于分子继续变好。下一步应优先做不确定性感知引导、预算等价对照、尺寸匹配和盲法实验闭环,而不是继续只刷新单一 Vina 数字。


脚注与索引状态。 本报告使用的核心数字来自 MolJO arXiv v4 HTML 的表 1–4、6、11–16、算法 1与对应正文/附录行域;相关工作数字来自各自论文正式页或正文。ml-report-index.md 已在本次运行中追加本条记录。HTML 已通过 Warm Editorial 结构验证器(0 error / 0 warning);运行环境虽有 Playwright 包,但 /usr/bin/chromium、/opt/pw-browsers 与 Playwright 缓存均无可执行浏览器,且按要求未运行 playwright install,故本轮无法完成真实 Chromium 的浅色、深色和移动端截图核验。未公开或未找到的项目页、视频、独立聚合条目均已明确标注,不以搜索缺失推断“绝对不存在”。