用自适应多输出网络加速燃烧等刚性化学反应模拟。
AMORE: Adaptive Multi-Output Operator Network for Stiff Chemical Kinetics
- 设计自适应损失函数,自动平衡不同变量和样本的误差。
- 通过解析映射保证质量分数总和为1,精度更高。
- 适用于多种模型,可提升燃烧仿真计算效率。
刚性系统的时间积分是燃烧、高超音速飞行及其他反应输运系统中的主要计算瓶颈。此类系统的时间尺度远小于其他物理过程,导致显式方法需极小时间步长,或隐式方法计算量巨大。为此,我们提出AMORE:自适应多输出算子网络,包含能预测多个热化学状态的算子及自适应损失函数,确保可靠学习。该算子从初始条件预测全部热化学状态。我们设计了两种自适应损失函数,分别考虑各状态变量与样本的误差。构建的主干网络自动满足单位分解(Partition of Unity)。为精确满足质量分数总和为1的约束,提出一个可逆解析映射,将n维物种质量分数向量转换至(n-1)维空间。还将自适应损失扩展至两步训练中深度算子网络(DeepONet)的主干与分支。另采用软最大函数在预测阶段直接保证质量分数和为1。通过两个案例验证有效性:合成气(12个状态)、GRI-Mech 3.0(54个物种中24个活跃状态)。所提深度算子网络将成为未来湍流燃烧模拟中计算加速的核心组件。AMORE为通用框架,此处亦展示其在FNO上的应用。
原文摘要 · Abstract (English)
Time integration of stiff systems is a primary source of computational cost in combustion, hypersonics, and other reactive transport systems. This stiffness can introduce time scales significantly smaller than those associated with other physical processes, requiring extremely small time steps in explicit schemes or computationally intensive implicit methods. Consequently, strategies to alleviate challenges posed by stiffness are important. While neural operators (DeepONets) can act as surrogates for stiff kinetics, a reliable operator learning strategy is required to appropriately account for differences in error between output variables and samples. Here, we develop AMORE, Adaptive Multi-Output Operator Network, a framework comprising an operator capable of predicting multiple outputs and adaptive loss functions ensuring reliable operator learning. The operator predicts all thermochemical states from given initial conditions. We propose two adaptive loss functions within the framework, considering each state variable's and sample's error to penalize the loss function. We designed the trunk to automatically satisfy Partition of Unity. To enforce unity mass-fraction constraint exactly, we propose an invertible analytical map that transforms the $n$-dimensional species mass-fraction vector into an ($n-1$)-dimensional space. We extend the proposed adaptive loss functions to trunk and branch training in two-step training of DeepONet with multiple outputs. We implemented another unity mass fraction constraint exactly using a softmax function on the predicted mass fraction. We demonstrate efficacy and applicability of our models through two examples: syngas (12 states), GRI-Mech 3.0 (24 active states out of 54). The proposed DeepONet will be a backbone for future CFD studies to accelerate turbulent combustion simulations. AMORE is a general framework, and here, we also demonstrate it for FNO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。