用可微分方法精准计算相平衡,让机器学习符合热力学规律。
Differentiable Thermodynamic Phase-Equilibria for Machine Learning
- 结合离散枚举与掩码软最大值,实现热力学一致的相平衡计算。
- 在二元液-液平衡数据上优于现有替代方法,误差更低。
- 适合需要物理一致性建模的化学工程场景,如相变预测。
精确预测相平衡仍是化学工程中的核心挑战。近年来,将热力学结构融入神经网络的物理一致机器学习方法在活度系数建模中表现出色。然而,将其扩展到基于极值原理(如液-液平衡)的平衡数据仍具难度。本文提出DISCOMAX,一种可微分相平衡计算算法,可在训练和推理时保证热力学一致性,仅依赖用户指定的离散化。该方法在前向传播中传递真实平衡态,在反向传播中使用掩码软最大值聚合,并采用直通梯度估计器,实现神经网络 extit{gE}-模型的物理一致端到端学习。该方法与统计热力学具有类比关系,已在二元液-液平衡数据上验证,性能优于现有基于代理模型的方法,同时为从各类平衡数据中学习提供通用框架。
原文摘要 · Abstract (English)
Accurate prediction of phase equilibria remains a central challenge in chemical engineering. Physics-consistent machine learning methods that incorporate thermodynamic structure into neural networks have recently shown strong performance for activity-coefficient modeling. However, extending such approaches to equilibrium data arising from an extremum principle, such as liquid-liquid equilibria, remains difficult. Here we present DISCOMAX, a differentiable algorithm for phase-equilibrium calculation that guarantees thermodynamic consistency at both training and inference, only subject to a user-specified discretization. The method combines discrete enumeration of feasible phase states with masked softmax aggregation in the backward pass, with the propagation of the true equilibrium state in the forward pass, using a straight-through gradient estimator to enable physics-consistent end-to-end learning of neural \gls{gE}-models. We show that this approach bears analogy to statistical thermodynamics, and we evaluate it on binary liquid-liquid equilibrium data where it outperforms existing surrogate-based methods, while offering a general framework for learning from different kinds of equilibrium data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。