用机器学习修复粗网格求解器的误差,实现低算力高精度模拟。
RECAST: A Machine-Learning Framework for Correction and Super-Resolution of Coarse-Grid PDE Solvers
- 在时间步进中嵌入学习修正,再重建精细网格状态。
- 相比未修正的粗网格解,平均相对误差降低50%至92%。
- 适用于未知参数和长时程模拟,适合科学计算加速场景。
粗网格数值求解器可大幅降低时变偏微分方程(PDE)模拟的计算成本,但因分辨率不足常导致解的轨迹与空间保真度下降。我们提出RECAST(Recurrent Error Correction And Super-resolution of coarse-grid Trajectories),一种机器学习框架,旨在恢复丢失的精度同时保持粗网格演化特性。RECAST将学习到的误差修正嵌入数值时间步进过程,并从修正后的粗网格历史中重建对应精细网格状态。我们在六个一维PDE系统上评估该框架,涵盖传输、扩散、色散、反应和波动力学,使用8-16倍粗化空间网格及从未见初始条件出发的1000步闭环滚动预测。测试结果表明,RECAST始终紧密贴合精细网格参考解,相较对应未修正粗网格求解器,时间平均相对误差减少约50%-92%。额外实验显示其对未见PDE参数值具有泛化能力;与当代粗网格修正架构对比,RECAST在5000步滚动预测中表现出更低误差和更优长期一致性。这些结果证明,RECAST的学得修正与重建能力可在显著粗化网格的前提下维持解的保真度,为机器学习加速多维数值模拟提供概念验证路径。
原文摘要 · Abstract (English)
Coarse-grid numerical solvers can substantially reduce the computational cost of time-dependent PDE simulation, but under-resolution often degrades both the trajectory and the spatial fidelity of the solution. We introduce RECAST (Recurrent Error Correction And Super-resolution of coarse-grid Trajectories), a machine-learning framework designed to restore this lost accuracy while retaining coarse-grid evolution. RECAST combines learned correction within the numerical time-stepping loop with reconstruction of the corresponding fine-grid state from the corrected coarse history. We evaluate the framework on six one-dimensional PDE systems spanning transport, diffusion, dispersion, reaction, and wave dynamics, using spatial grids coarsened by factors of 8-16 and 1000-step closed-loop rollouts from unseen initial conditions. Across the test cases, RECAST remains closely aligned with the fine-grid reference solutions and reduces time-averaged relative error by approximately 50-92% compared with the corresponding uncorrected coarse-grid solvers. Additional tests show generalization to unseen PDE parameter values, while comparison with a contemporary coarse-correction architecture shows that RECAST achieves lower error and better long-horizon agreement with the fine-grid reference over 5000-step rollouts. These results demonstrate that the learned correction and reconstruction capabilities of RECAST can enable substantially coarser PDE evolution without the corresponding loss of solution fidelity, providing a proof-of-concept route toward machine-learning acceleration of higher-dimensional numerical simulations across science and engineering.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。