Chemeleon2:强化学习引导生成模型发现新型晶体结构

超能小量子
2026-07-14 20:43:57
人工智能
材料科学
论文精读与讲座笔记
本帖最后由 超能小量子 于 2026-7-14 20:48 编辑


本研究将强化学习引入晶体生成模型,用奖励函数替代似然目标,让模型从"模仿已知材料"转向"主动探索未知空间",在 mSUN(亚稳、唯一、新颖结构比例)指标上实现了从 15.9% 到 61.3% 的跃升。




晶体材料一直是技术革命的底层推动力——从半导体到电池,从催化剂到能源材料,每一种新晶体的发现都可能开启新的应用范式。然而,晶体材料的设计空间是典型的组合爆炸问题:元素组成、化学计量比、晶格参数、原子坐标、空间群、局域配位环境……这些维度相互耦合,使得穷举搜索在计算上不可行。


高通量计算和材料数据库(如 Materials Project)极大扩展了已知晶体的版图,但它们仍然是在"已知附近"打转——无论是 DFT 筛选还是数据挖掘,都很难系统性地跳出已有知识的引力范围。


生成式 AI 的出现改变了这个局面。VAE、GAN、自回归模型、扩散模型等方法陆续被用于生成无机晶体结构,试图从数据库中学习材料的隐含分布,然后从中采样新晶体。问题在于,这种做法存在一个根本性的目标错位​:模型学习的是训练数据的高概率区域,而材料科学家真正想要的,恰恰是那些低概率、但可能稳定且有用的未知结构。


Chemeleon2 正是为了解决这个错位而设计的。它的核心思路很简单:不让模型学分布,而是让模型学"什么值得生成"​​。具体做法是把晶体生成建模为一个强化学习问题——扩散模型是智能体,奖励函数是环境,新颖性、稳定性和多样性是评分标准。经过 RL 后训练,模型从被动的分布采样器升级为主动的探索工具。




背景:为什么似然目标不够用


要理解 Chemeleon2 的动机,得先看清楚似然训练的局限性。


扩散模型的标准训练目标是最大化训练数据的似然。在数学上这是优雅的,但在材料发现场景下会产生一个棘手的后果:模型生成的样本会天然地聚集在训练集中的高密度区域。换句话说:模型更倾向于生成"看起来像已知材料"的晶体,而不是去寻找那些数据库中缺席、但物理上可能存在的全新结构。


材料发现的特殊性加剧了这个问题。已知数据库(如 ICSD、Materials Project)的覆盖严重不均衡——某些元素组合和化学配比被充分研究,而另一些区域则几乎是空白。这种不均衡不是随机的;它反映了人类研究者的偏好(热门体系更容易被研究)和实验可及性的差异(某些体系更容易合成)。如果生成模型忠实复现了数据库的分布,它也会忠实复现这种偏好的扭曲。


此外,多样性、新颖性、热力学稳定性这三种目标之间存在天然张力。越靠近已知结构越可能稳定,越远离则越新颖但越可能不稳定。传统的条件生成方法(如 Classifier-Free Guidance)可以对某一属性做弱控制,但无法在多目标之间做显式的权衡优化。


所以问题不是生成模型不够好,而是训练目标本身和发现目标不一致​。这个洞察是 Chemeleon2 的出发点。




方法:把生成变成强化学习


Chemeleon2 的技术架构分两层:生成层和优化层。


生成层:潜在扩散模型作为策略网络


直接对晶体做 RL 会面临一个实际问题:晶体表示太复杂了。原子类型是离散的,晶格矩阵和原子坐标是连续的,策略概率计算困难,采样成本也高。Chemeleon2 的做法是先用一个变分自编码器(VAE)将晶体结构压缩到连续潜在空间,再在这个潜在空间中做扩散生成。


这样做的优势很明显:潜在空间是连续的,可以直接用策略梯度方法优化;扩散模型的去噪过程天然对应着一个多步决策过程,每一步去噪就是一个动作,整个扩散链就是一条轨迹。因此,把扩散模型视为策略网络在数学上是自然的。



图1|强化学习引导晶体材料探索框架。


优化层:GRPO + 多目标奖励


RL 阶段是整篇论文的核心创新。Chemeleon2 使用组相对策略优化(Group Relative Policy Optimization, GRPO)来更新扩散模型参数。


GRPO 的核心思路是:对同一个输入条件生成一组候选晶体(比如 K=4 个),然后在组内做相对归一化——每个候选的奖励减去组内均值,再除以组内标准差,得到组内相对优势。和普通的 REINFORCE 相比,GRPO 大幅降低了优势估计的方差。论文中的训练曲线也验证了这一点:GRPO 的奖励提升更平滑,梯度方差更低。



图2|策略优化算法示意图。


奖励函数由三个组件构成:


创造性奖励:通过连续结构距离度量生成晶体与参考数据库中最近邻晶体的差异,鼓励生成唯一且不同于已知结构的候选。这里使用连续距离而不是简单的"是否完全匹配"二元判断,避免了优化过程中的陡峭梯度问题。


稳定性奖励:基于能量凸包上方能量(Energy Above Hull),鼓励热力学上可行或亚稳的结构。考虑到 DFT 计算太昂贵,Chemeleon2 使用机器学习力场(MLFF)做快速近似评估,这使得 RL 训练中的大量试错变得可行。


多样性奖励:显式惩罚模式坍缩,推动模型覆盖更广泛的组成和化学空间。消融实验表明,去掉多样性奖励后,模型会反复生成少数相似结构,仅通过微小几何扰动来赚取奖励——这正是材料生成中典型的奖励黑客行为。


这三个奖励的权重决定了模型在探索和利用之间的平衡点。论文没有给出固定的最优权重,而是展示了不同权重配置下的帕累托前沿,说明这个框架的可调性本身就是一种优势。




结果:从 15.9% 到 61.3% 的跃升


核心指标:mSUN


论文定义了一个核心评估指标——mSUN(metastable, unique, and novel)​,即同时满足三个条件:亚稳(能量凸包上方能量 < 0.1 eV/atom)、唯一(与其他生成结构不重复)、新颖(不同于训练集中的任何结构)。这个指标直接衡量一个生成模型对材料发现的实用价值——多生成没有用的已知结构,少生成不稳定的虚构结构。


在 Alex-MP-20 数据集上的从头生成基准测试中,每个模型生成 10,000 个晶体,直接对未做几何优化的原始结构评估:


mSUN 提升了 45.4 个百分点,相对提升 285.5%。这个数字的含义是:每生成 100 个晶体,有效新结构的数量从约 16 个增加到约 61 个,接近 **4 倍的发现效率提升**​。


唯一性从 99.4% 降到 88.7%,提示 RL 模型仍存在轻微的模式坍缩倾向——模型找到了一些"高奖励区域"后会集中采样。但考虑到 mSUN 的巨大提升,这个代价是值得的,而且多样性奖励已经显著缓解了这一问题。


在较小的 MP-20 数据集上的结果趋势一致,说明该方法不是对单一数据集的过拟合。


新颖性—稳定性困境


材料生成中有一个经典困境:越靠近已知材料越稳定,越远离越新颖。传统生成模型受训练分布限制,很难同时提升这两者。


Chemeleon2 在这个二维图上明显拓展了帕累托前沿。t-SNE 结构嵌入可视化显示,基线模型虽然覆盖较广,但有效候选稀疏;RL 模型则形成了富含 mSUN 结构的局部簇,表明模型学会了在"有前途"的区域深耕,而不是均匀撒网。


一个有趣的发现是:RL 模型倾向于先识别热力学有利的结构原型(比如钙钛矿、尖晶石等),再通过多元素、多阴离子替换引入组成复杂性。换句话说,模型学会的策略是"在稳定骨架之上增加化学新颖性"​​,而不是凭空想象全新结构类型。这种策略在材料科学上也是合理的——已知结构类型通常已经经过自然和实验的筛选。


组成空间的迁移


另一个重要发现是 RL 如何改变了模型的元素偏好。基线 Chemeleon2 偏向生成含氧、硫族和卤素的化合物,这恰好是 Materials Project 等数据库中占比最高的类别。RL 后,过渡金属比例从 26.2% 增加到 35.6%,非金属比例从 27.2% 降至 16.6%。Br、I、Se 等更可极化元素的出现频率也增加了。


这种迁移是奖励设计的结果:稳定性奖励偏好具有多氧化态和可调配位环境的过渡金属,创造性奖励推动模型探索数据库中欠表示的元素组合。模型还生成了较多三元和四元化合物,其中相当一部分具有较低凸包上方能量,提示这些复杂组成中存在真实的合成候选。


论文也坦率指出,目前的模型生成的是有序周期结构,不能很好区分离散化合物和固溶体。某些多金属体系可能更适合被解释为无序固溶体,而不是确定的有序化合物。这是当前模型的一个已知盲区。


性质引导生成:带隙 3 eV 的靶向设计


除从头生成外,论文还做了一个更有应用场景的实验——目标带隙引导设计​,以 3 eV 为靶点。


这一实验比较了三种条件生成方式:CFG、LoRA+CFG、RL。CFG 生成的带隙分布很宽,无法有效集中到目标附近,而且严重破坏了 mSUN(有效结构比例接近零)。LoRA+CFG 略有改善但不明显。RL 优化后的模型则生成了 DFT 验证带隙明显集中在 3 eV 附近的结构,同时保持 mSUN 为 45.3%。


在 512 个初始样本中,有 82 个结构同时满足 mSUN 和 2.7–3.3 eV 的目标带隙。经过 DFT 几何弛豫后,mSUN 和亚稳比例还进一步提升了,说明模型生成的结构已经接近平衡构型。


这个结果的意义在于:RL 不仅能引导模型生成想要的属性,而且不会在属性优化过程中牺牲化学有效性​。这是传统条件生成方法无法做到的。



图3|强化学习用于性质引导晶体生成。




讨论与判断


Chemeleon2 的真正贡献不在于具体的 mSUN 数字,而在于它提出了一个范式层面的思路转变​:生成式 AI 在科学发现中的角色,不应该是一个被动的数据分布模仿者,而应该是一个可以被科学目标直接驱动的主动探索者。


这个思路的通用性很高。论文中奖励函数的模块化设计意味着,只要有一个可以量化评估的科学目标(无论是稳定性、带隙还是其他功能性质),就可以作为 RL 的奖励信号来引导生成。这为 AI 驱动的材料逆向设计提供了一套可插拔的框架。


但局限也值得认真对待:


第一,结构新颖性的精确定义仍然是一个开放问题。对于低对称结构、轻微晶格畸变和复杂配位环境,简单的结构距离不足以做出可靠判断,"看似新实则等价"或"看似已知实则不同"的误判难以避免。


第二,有序结构的假设限制了模型的适用范围。真实材料中的固溶体、位点无序、缺陷、非化学计量组成都无法被当前模型处理。这在多元素体系中尤其突出——模型生成的某些"有序结构"可能在实验上更接近无序固溶体。


第三,稳定性评估的可靠性取决于力场精度。MLFF 虽然快,但对于高阶多元素体系中参考相图稀疏的区域,能量估计可能被高估。论文也提醒了这一点——最终的验证仍然需要 DFT 弛豫,甚至实验合成。


另外,从工程角度看,Chemeleon2 训练一个 RL 后训练的模型需要先生成大量候选再做评估反馈,计算开销不小。论文没有详细报告训练成本,但对于实际部署来说,这是一个需要考量的因素。


综合来看,Chemeleon2 打动人的地方在于它回答了一个非常根本的问题:当生成模型的训练目标和科学发现目标不一致时,该怎么办?它的回答是:别让模型学分布,让模型学目标。这个回答简洁、有效,而且可复用。





论文信息:


Park, H., Walsh, A. Guiding generative models to uncover diverse and novel crystals via reinforcement learning. Nature Machine Intelligence (2026).


DOI: https://doi.org/10.1038/s42256-026-01262-4


80
0
0
0
关于作者
相关文章
  • 量子计算解决湍流非线性难题:专用量子计算机实现复杂流场高效模 ...
    湍流作为典型的非线性复杂系统,始终是经典计算“逐点推进”逻辑的算力天花板。北京大 ...
    了解详情 
  • JCTC | AI赋能分子晶体结构预测:基于类似物模板策略的高效采样 ...
    分子晶体结构预测(CSP)是材料发现领域的重要计算工具,但其核心瓶颈在于计算成本高昂,这主要 ...
    了解详情 
  • Kaiwu-PyTorch-Plugin v0.2.0发布 | 支持Q-Diffusion,生成任务 ...
    摘要给扩散模型装上“能量导航”;蛋白质生成和单细胞分析任务更精准更新概览Kaiwu-Py ...
    了解详情 
  • PocketXMol原子级生成式AI大模型发布:统一3D小分子与多肽药物设 ...
    人工智能在分子结构预测与药物设计领域正经历着深刻的变革,但现有的主流模型往往受限于特定的分 ...
    了解详情 
领取成功
本月5个550bit真机配额已发放给您,配额将在2个月后到期,请及时使用哦~
活动中心
联系我们
二维码
返回顶部
返回
活动中心

完成任务,轻松获取真机配额

×
每日必做
新手任务
长期任务
其他任务
快速回复 返回顶部 返回列表
玻色有奖小调研
填写问卷,将免费赠送您1个1000bit真机配额
(单选) 您是从哪个渠道得知我们的?*
您是从哪个社交媒体得知我们的?*
您是通过哪个学校的校园宣讲得知我们的呢?
取消

提交成功

真机配额已发放到您的账户,可前往【云平台】查看

量子AI开发者认证

考核目标

开发者能够成功搭建Kaiwu-PyTorch-Plugin项目基础环境,并成功运行示例代码,根据示例提示,输出指定的值并填写至相应的输入框中。

通过奖励

5个一年效期的1000量子比特真机配额

专属「量子AI开发者」社区认证标识

开发者权益

每月固定权益:5个550量子比特真机配额
前往考核

第一步

按照README提示成功安装Kaiwu-PyTorch-Plugin库环境依赖
前往GitHub

第二步

运行 community-assessment 分支下的 run_rbm.py 代码示例

第三步

理解示例代码,手动打印并填写如下数值:

正相采样的状态

负相采样的状态

正相的能量值

负相的能量值

*

提交答案

开发者权益

每月固定权益:5个550量子比特的真机配额

恭喜您完成考核

您将获得量子AI开发者认证标识及考核奖励

1000 bit*5

配额

Quantum AI Developer Certification

Assessment Objectives

Developers should successfully set up the basic environment for the Kaiwu-PyTorch-Plugin project, run the QBM-VAE sample code, and calculate the correct FID value based on the random seed value provided by the system.

Pass Rewards

10 quotas for 550-qubit real quantum machines with a one-year validity period

Exclusive "Quantum AI Developer" Community Certification Badge

Developer Benefits

Fixed Monthly Benefits: 5 quotas for 550-qubit real quantum machines
Proceed to Assessment

Step 1

Install the environment dependencies for the Kaiwu-PyTorch-Plugin library according to the README instructions
Go to GitHub

Step 2

Replace the Seed Value

Your seed value is

Step 3

Enter the FID Value You Calculated

*

Submit Answer

Developer Benefits

Fixed Monthly Benefits: 5 quotas of 550-qubit real machines

Congratulations on Completing the Assessment

You will receive the Quantum AI Developer Certification Badge and Assessment Rewards

550bit*10

Quotas