摘要
相场建模是一种有效的方法,用于捕捉材料的中尺度形态和微观结构演化,但计算成本高昂。因此,需要快速且具有通用性的替代模型,以减轻材料优化和设计等计算密集型过程的成本。由于尖锐相边界的存在导致物理现象具有内在的不连续性,这使得替代模型的训练变得繁琐。本文开发了一个框架,将卷积自动编码器架构与深度神经算子(DeepONet)相结合,以学习两相混合物的动态演化,并加快预测微观结构演化的求解时间。我们利用卷积自动编码器在低维潜在空间中提供微观结构数据的紧凑表示。在潜在空间中训练好DeepONet后,它可用于在插值任务中替代高保真相场数值求解器,或在外推任务中加速数值求解器。
(带有卷积自动编码器的DeepONet示意图:蓝色为编码器卷积层,绿色为最大池化层,橙色为解码器转置卷积层,步骤一为自动编码器训练,步骤二为潜在空间中DeepONet训练)
研究方法
用卷积自动编码器将高维微观结构数据映射到低维潜在空间,学习数据的紧凑表示;DeepONet 在潜在空间中学习微观结构的动态演化,通过训练两个网络,实现对微观结构演化的高效预测。研究自动编码器潜在维度大小、激活函数、训练样本数量、损失函数等因素对模型性能的影响,确定最佳模型参数;采用增加早期时间步权重的训练策略,使模型更好地捕捉多尺度特征。
(框架结构详细信息;对潜在维度大小、网络架构及非线性激活函数的详细研究)
(算法)
数据集
使用 Sandia 的内部多物理场相场建模代码 MEMPHIS,通过拉丁超立方抽样独立采样相分数ϕA、相迁移率MA和MB,共生成 500 个三元组(ϕA,MA,MB) 。在二维方形域Ω=[0,1]×[0,1]上进行模拟,离散为 512×512 个网格点,时间步长Δt=1×10−4 ,模拟 50,000,000 步,每 500,000 步保存一次微观结构状态,每个模拟案例共保存 100 个时间帧。
将 512×512 的域图像下采样为四个 256×256 的图像,再用三次插值法降分辨率至 128×128 。从 500 个微观结构演化样本中生成 2000 个 128×128 分辨率的样本,其中 1600 个用于训练,400 个用于测试。使用相对L2范数衡量模型预测误差,分析不同因素对模型性能的影响。
(真实和预测微结构间的L2范数误差:a图为训练时间给定不同时间步时的误差;b图为训练阶段对未来时间步的预测误差,以评估泛化能力)
结果
增加潜在维度可提高模型预测精度,tanh 和 sin 激活函数的 DeepONet 模型性能优于 ReLU 激活函数的模型;增加训练样本数量可提高模型准确性,1200 个样本训练的模型与 1600 个样本训练的模型精度相似;基于L2损失训练的自动编码器模型性能更好。
模型在早期时间步预测误差较大,随着演化过程变慢,预测准确性提高;在预测未见过的时间帧时,误差逐渐增加,但较大潜在维度的模型预测误差较小,ld=100或ld=196时,预测误差小于 6% 。
向模型输入添加高达 10% 的高斯白噪声,相对L2范数增加不明显,表明模型对噪声具有鲁棒性。时间间隔增大,预测误差增加,但 DeepONet 预测的计算效率不受时间间隔影响。
将自动编码器 - DeepONet 框架与高保真相场求解器集成的混合模型,在预测 10 个时间帧时,相比单独使用高保真求解器速度提升 29% ,且不损失准确性。
(将微观结构状态信息转换为潜在空间中的向量)
结论与展望
与基于统计函数的方法相比,该框架通过卷积自动编码器学习低维潜在空间,避免了复杂的后处理步骤;可扩展到三维微观结构数据;完全数据驱动,对边界条件变化不敏感。未来可通过添加物理约束提高模型准确性,融合不同来源数据实现多保真度建模,进一步提升框架性能。
原文链接:https://doi.org/10.1038/s41524-022-00876-7
期刊:npj computational materials
开源代码:https://github.com/vivek-brown/NPJCOMPUMATS-03747
补充材料:https://doi.org/10.1038/s41524-022-00876-7
欢迎关注我的新公众号:倒三角冲浪家
在此更新力学、工程类交叉学科等最新文献,以及其他趣味题材文章