近日,中国人民大学高瓴人工智能学院长聘副教授孙浩团队在Nature子刊《自然-机器智能》(Nature Machine Intelligence)上发表了题目为“Encoding physics to learn reaction-diffusion process”的论文。《自然-机器智能》期刊出版高质量的原创研究和评论,涉及机器学习、机器人和人工智能的广泛主题,探索和讨论这些领域对其他科学学科以及社会和工业的许多方面产生的重大影响。这是我院首次在该高水平期刊上发表论文。
该论文提出了一种新的物理知识嵌入深度学习架构,即PeRCNN,用于偏微分方程(PDE)正反问题求解、非线性时空动力系统建模和控制方程发现,旨在改善复杂时空动力系统基于稀疏和噪声数据建模的准确性和可解释性。该论文还建立了一个强制编码物理结构的循环卷积神经网络框架,有效解决了网络对训练数据的依赖性,突破了神经网络模型不可解释的瓶颈,同时提升了模型的外推和泛化性,在各类反应扩散系统(RD)问题场景实现成功验证。
图1 :PeRCNN模型图
一般而言,由于先验知识有限,且系统变量的非线性过程难以用明确的偏微分方程(PDE)公式描述,复杂时空动力系统建模和仿真一直是一项具有挑战性的任务。常见的机器学习方法则需要依赖大量训练数据,还存在可解释性差、泛化性弱、误差不可控等基础科学问题。将先验物理知识(例如控制方程)作为约束加入深度学习,在一定程度上会增强模型的可解释性、缓解数据依赖性。但是,现有的基于物理信息的深度学习方法,通常需要通过软约束来施加物理定律,模型表现很大程度上取决于超参数的恰当设置。因此,发展新的知识嵌入学习模型,从数据中捕捉潜在的时空动态演化机制,是亟需解决的问题。
图2:实验结果(PDE正问题)
为此,该文章提出了一种新的学习模型,即物理编码递归卷积神经网络(PeRCNN),如图1所示。PeRCNN的一个主要优点是可以将先验物理知识结构编码到网络中,采用了时空学习范式,旨在建立一个通用且鲁棒学习模型,能够保证所得到的网络严格服从给定的先验物理知识(例如PDE结构、初始和边界条件),使得网络具有可解释性。模型通过循环卷积神经网络对给定的物理结构进行编码,从而提高在稀疏和噪声数据情况下学习复杂时空动力学的能力。
图3:实验结果(PDE反问题)
通过大量数值实验,结果展示了PeRCNN对反应扩散型偏微分方程的正向(图2)和反向(图3)问题分析的有效性。与几个基线模型的比较表明,所提出的物理编码学习范式具有独特的外推能力、对数据噪声或稀疏数据的泛化能力和鲁棒性。在数据驱动仿真实验中,PeRCNN取得了SOTA结果;PeRCNN的平坦误差传播曲线(图4),也证明了PeRCNN显著的外推和泛化能力,这意味着该方法不仅能够准确地预测复杂时空动力系统的演变,还能够捕捉模型背后的物理机制,并且在应对新的工况时能够展现出一定的通用性。
图4:实验结果(数据驱动建模,误差传播和外推图示)
此外,该工作还成功地将 PeRCNN 与稀疏回归算法相结合,解决潜在PDE 发现问题(图5),可进一步从学习模型中提取控制底层物理机制的解析表达式。耦合方案使得模型能够迭代优化网络参数,并对发现的PDE结构和系数进行微调,获得最终PDE的简约表达形式,从稀疏带噪声噪声的测量数据中准确可靠地发现潜在的物理定律。
图5:控制偏微分方程发现流程图
虽然本文证明了PeRCNN在各种反应扩散系统上的有效性,但该模型在理论上适用于其他类型的时空动力系统,例如,文章的补充材料中展示了带对流项的二维Burgers方程、雷诺数为1000的Kolmogorov湍流(NS方程)。
这项研究成果为复杂时空动力系统数据驱动建模领域带来了新进展,为科学家和工程师提供了更强大的工具来理解和预测自然和工程现象。这种结合了深度学习和先验物理的方法,有望在多个学科领域应用中发挥重要作用,包括流体力学、生物化学、环境科学、工程学、材料科学等。让我们拭目以待,期待这一新方法的进一步发展和应用,它将为我们揭示更多关于复杂时空动力系统的奥秘,并为未来的科学和技术发展带来新的突破。
主要作者介绍:孙浩,中国人民大学高瓴人工智能学院长聘副教授、博导,国家海外高层次青年人才;美国哥伦比亚大学取得工程力学博士、麻省理工学院博士后。曾任美国匹兹堡大学、美国东北大学终身序列助理教授、博导。主要从事科学智能计算理论与方法研究,在Nature Machine Intelligence、Nature Communications等期刊/会议上发表论文60余篇,主持国家高层次人才计划、国家自然科学基金委(重大研究计划培育项目、面上项目)、美国科学基金委(ECI、SCC、DRRG)、北京市自然科学基金委(面上项目)等研究项目十余项(3000余万元)。入选福布斯北美“30位30岁以下精英榜(科学类)”、“美国十大华人杰出青年”,荣获DeepTech“中国智能计算科技创新人物”。
论文信息:Chengping Rao, Pu Ren, Qi Wang, Oral Buyukozturk, Hao Sun*, Yang Liu*. Encoding physics to learn reaction-diffusion processes. Nature Machine Intelligence (2023), DOI: 10.1038/s42256-023-00685-7
论文链接:https://www.nature.com/articles/s42256-023-00685-7
检测到您当前使用浏览器版本过于老旧,会导致无法正常浏览网站;请您使用电脑里的其他浏览器如:360、QQ、搜狗浏览器的速模式浏览,或者使用谷歌、火狐等浏览器。
下载Firefox