用多项式混沌展开实现算子学习,兼具高精度与不确定性量化能力
Polynomial Chaos Expansion for Operator Learning
- 基于多项式混沌展开构建算子学习的数学框架,将问题转化为求解系数方程组
- 在多个PDE问题上实现高精度预测,计算效率优于传统深度学习方法
- 无需额外开销即可实现不确定性量化,适合需可信推断的科学计算场景
算子学习(OL)作为科学机器学习的重要工具,用于近似无限维函数空间间的映射,尤其适用于求解偏微分方程(PDE)的解算子。尽管当前进展主要依赖深度神经网络如DeepONet和FNO,本文首次将多项式混沌展开(PCE)引入算子学习。我们建立了PCE在纯数据驱动与物理信息融合两种场景下的数学框架,将算子学习转化为求解PCE系数的线性系统。该框架天然支持不确定性量化(UQ),仅通过后处理系数即可获得,不增加计算成本。在多种PDE问题上的数值实验表明,该方法在算子学习与不确定性量化任务中均表现优异,兼具高精度与计算高效性。
原文摘要 · Abstract (English)
Operator learning (OL) has emerged as a powerful tool in scientific machine learning (SciML) for approximating mappings between infinite-dimensional functional spaces. One of its main applications is learning the solution operator of partial differential equations (PDEs). While much of the progress in this area has been driven by deep neural network-based approaches such as Deep Operator Networks (DeepONet) and Fourier Neural Operator (FNO), recent work has begun to explore traditional machine learning methods for OL. In this work, we introduce polynomial chaos expansion (PCE) as an OL method. PCE has been widely used for uncertainty quantification (UQ) and has recently gained attention in the context of SciML. For OL, we establish a mathematical framework that enables PCE to approximate operators in both purely data-driven and physics-informed settings. The proposed framework reduces the task of learning the operator to solving a system of equations for the PCE coefficients. Moreover, the framework provides UQ by simply post-processing the PCE coefficients, without any additional computational cost. We apply the proposed method to a diverse set of PDE problems to demonstrate its capabilities. Numerical results demonstrate the strong performance of the proposed method in both OL and UQ tasks, achieving excellent numerical accuracy and computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。