基于非重叠型区域分解的随机参数偏微分方程算子学习

刘思琪 ,  石晓宇 ,  徐之航 ,  廖奇峰

四川大学学报(自然科学版) ›› 2026, Vol. 63 ›› Issue (4) : 813 -822.

PDF (980KB)
四川大学学报(自然科学版) ›› 2026, Vol. 63 ›› Issue (4) : 813 -822. DOI: 10.19907/j.0490-6756.260122
深度学习方法在多物理耦合建模与计算中的应用

基于非重叠型区域分解的随机参数偏微分方程算子学习

作者信息 +

Nonoverlapping domain decomposition based operator learning for partial differential equations with stochastic parameters

Author information +
文章历史 +
PDF (1003K)

摘要

深度神经算子通过构建数据驱动的代理模型为含随机参数的偏微分方程提供高效求解途径。在大规模高维问题中,直接训练全局神经算子普遍面临计算成本高昂、泛化精度受限的双重瓶颈。本文提出了一种融合非重叠区域分解方法与算子学习的新型计算框架,该框架将全局计算域剖分为若干不重叠的局部子区域,并结合局部Karhunen-Loève(KL)展开实现参数降维。在训练阶段,该框架独立训练局部神经算子,建立局部参数与界面条件到局部解的映射。在预测阶段,该框架引入基于界面约束的优化算法来驱动各局部算子并行、快速地推断全局近似解。相比传统全局神经算子,该方法显著降低了代理模型的拟合难度,能够以更低的计算代价获得更高的预测精度。在二维随机扩散方程的数值实验中,相比全局神经算子,本文方法仅需约20%的网络参数即可获得更优更稳健的逼近效果,且相对误差降低约34%。本文的方法具有局部化结构,天然支持并行计算,可以有效提升大规模参数化偏微分方程的求解效率。

Abstract

Deep neural operators have emerged as a promising paradigm for efficiently solving parametric partial differential equations (PDEs) with random parameters via data-driven surrogate modeling.However, directly training global neural operators for large-scale, high-dimensional problems usually suffers from prohibitive computational costs and limited generalization capabilities.To address these challenges, this paper proposes a novel computational framework that couples nonoverlapping domain decomposition methods (DDM) with operator learning.This approach partitions the global computational domain into several local subdomains and utilizes local Karhunen-Loève (KL) expansions to effectively reduce the parameter dimensionality.In the training phase, each local neural operator is trained independently to learn the mapping from local parameters and interface conditions to the local solution.In the prediction phase, an optimization algorithm based on interface constraints is further proposed, enabling all local operators to infer the global approximation rapidly and in parallel.Compared with the global neural operator, domain decomposition significantly reduces the fitting difficulty of surrogate models and achieves higher prediction accuracy at lower computational cost.A numerical example on a two-dimensional stochastic diffusion equation demonstrates that the proposed local operator requires only about 20% of the network parameters to achieve a better and more robust approximation performance, with the relative error reduced by approximately 34%.Furthermore, its localized structure naturally supports parallel computing, providing a new paradigm for the efficient solution of large-scale parameterized PDEs.

Graphical abstract

关键词

含参偏微分方程 / 非重叠型区域分解 / 算子学习 / 代理模型

Key words

parametric partial differential equation / nonoverlapping domain decomposition / operator learning / surrogate model

引用本文

引用格式 ▾
刘思琪,石晓宇,徐之航,廖奇峰. 基于非重叠型区域分解的随机参数偏微分方程算子学习[J]. 四川大学学报(自然科学版), 2026, 63(4): 813-822 DOI:10.19907/j.0490-6756.260122

登录浏览全文

4963

注册一个新账户 忘记密码

偏微分方程是计算科学与工程领域刻画物理现象的核心数学工具,在流体仿真、大气模拟等领域具有极其广泛的应用。然而,在针对复杂问题进行建模时,需要求解的模型通常带有表示不确定性的随机参数,如果采用传统数值方法(有限元方法1,有限差分方法2等)反复进行计算求解,往往会产生难以承受的计算成本。
目前,通过构建代理模型来加速预测已成为缓解上述计算瓶颈的研究热点。代理模型能够有效融合历史数据和潜在物理信息,以相对低的计算代价获得高精度的近似求解。为此,研究人员发展了广义多项式混沌展开3-4,高斯过程回归5-6和深度神经网络7-8等多种高效建模方法。
近年来,基于神经网络的算子学习方法因其具有逼近复杂映射的能力及处理高维数据的优势而成为极具前景的研究方向。算子学习的核心在于利用深度学习模型拟合输入空间与输出空间之间的映射关系,如学习偏微分方程中随机参数到其解的映射等。其中,深度算子网络(Deep Operator Network, DeepONet)9是一种双网络架构,无需固定网格离散化就能学习函数之间的映射。傅里叶神经算子(Fourier Neural Operator, FNO)10构建了在频域中运行的专有架构,能够将神经网络与特定物理结构离散化解耦。
为了进一步拓展算子模型在不同场景下的适用性,研究人员提出了多种扩展框架。例如,为处理复杂物理边界,可学习形变的傅里叶神经算子11引入了辅助网络,实现规则域到不规则域的映射。面对训练数据稀缺的挑战,内嵌物理信息的神经算子12-13提出将控制方程等先验物理知识融入训练过程的算法,有效提升模型的泛化能力与预测性能。
尽管算子学习方法在求解偏微分方程方面取得了进展,但仍旧面临着一些挑战。首先,训练模型过程需要高昂的计算与存储成本,特别是在处理高维问题时。算子模型通常需要在多次求解高维偏微分方程所得到的大型数据集上进行训练,不仅消耗极大的算力,还会带来巨大的内存需求。其次,基于单一全局计算域训练的算子模型的泛化能力有限,在处理非线性和多尺度问题时目标映射在整个定义域内呈现出显著的差异,进一步加剧使用单一全局算子进行高精度全局逼近的难度。
为应对以上挑战,研究者引入了区域分解方法14-20。这是一种极具前景的破局思路,核心是将大型计算区域剖分为多个局部子域,以缓解整体计算压力。将该思想应用在算子学习中可以将单一的全局映射转化为一组作用于各子域的局部算子。由于局部算子仅需拟合较小区域内的物理演化,这就使需要处理的问题相对变得低频简单,进而使深度学习模型能够更精准地捕捉解的局部特征,展现更好的泛化能力。此外,区域分解方法具备高度的可并行性,局部算子的训练能够充分利用计算资源,实现算法层面的高效扩展。并且,该策略还允许在不同子域内灵活配置异构网络分辨率,为多保真度数据融合提供支持,在保证局部复杂区域求解精度的同时最大化地提升整体计算效率。
本文提出了一种结合深度神经算子与区域分解的新型算子学习框架,以应对含参偏微分方程的计算挑战。数值实验结果表明,该方法能够显著降低算力开销,保证预测精度,为大规模含参偏微分方程的高效逼近提供可行的新方案。

1 问题描述

DRn(n=2,3)为具有多边形边界D的有界连通区域,其中xD表示空间变量。本文考虑如下含随机参数的偏微分方程系统:

Lx,ux,ω;a(x,ω)=fx, xD,Bx,ux,ω;a(x,ω)=gx, xD

其中,LB分别表示含随机参数的微分算子与边界算子,fg分别表示确定性源项与边界条件,系统随机输入a(x,ω)是定义在物理区域D上的随机过程,通常称为随机场,即

a(x,ω):D×ΩR,

其中,Ω是定义在完备概率空间Ω,F,P中的样本空间,Fσ代数,P为概率测度。

在实际计算中,难以直接处理无限维随机场,通常需要采用有限维随机变量对其进行参数化表示与降维。Karhunen-Loève (KL)展开21是一种常用的随机场降维方法,通过有限项截断有效刻画随机场中的主导不确定性。对均值μax和协方差函数Ca(x,x')的随机场a(x,ω),其d阶截断KL展开可表示为:

a˜x,ω=μax+j=1dλjψjxξjω

其中,λj,ψjxj=1d为协方差算子的前d个特征对,并按特征值大小降序排列,ξjωj=1d是一组互不相关的随机变量。使用截断KL展开式(2),近似替代原随机场a(x,ω)后,无限维随机参数偏微分方程系统(1)可转化为如下的有限维参数化问题:

Lx,ux,ξ;a˜x,ξ=fx, xD,Bx,ux,ξ;a˜x,ξ=gx, xD,

其中,ξ=ξ1,,ξdTΓ=Γ1××ΓdRd为由随机变量组成的有限维参数向量。针对上述有限维参数化偏微分方程问题,本文结合区域分解思想与深度学习算子方法构建局部化神经算子框架。该方法能够利用区域分解在处理大规模问题中的优势,在降低计算成本的同时为参数化偏微分方程问题构建高精度代理模型。

2 区域分解与局部参数化

考虑具有多边形边界D的物理区域DRn(n=2,3)。将全局区域D划分为ND个互不重叠的子区域D(1),,D(ND),满足D=i=1NDD(i),且对于任意ijD(i)D(j)=,记各子区域D(i)的边界为D(i)。对于每个子区域D(i),定义如下边界集合与邻接关系。

1) 子区域D(i)D(j)之间的交界面定义为jD(i)D(i)D(j),其中ij,显然有jD(i)=iD(j)

2) 子区域D(i)位于全局物理边界上的部分定义为exD(i)D(i)D,称为外部边界。相应地,子区域的内部界面部分定义为inD(i)DiexDi

3) 与子区域D(i)相邻的子区域指标集定义为Λ(i)j|j1,,ND,ji,D(i)D(j),该集合表示所有与D(i)共享公共界面的相邻子区域编号。在ND>1且区域划分不存在孤立子区域的情况下,有Λ(i)

4) 所有相邻子区域的有序编号对构成集合Λi,j|i1,,ND,jΛ(i)

图1给出了包含两个非重叠型子区域的区域分解示例。此时,有D=D(1)D(2)Λ(1)=2Λ(2)=1并且Λ=1,2,(2,1)

针对每个子区域D(i),定义局部随机场a(i)x,ω为全局随机场ax,ω在该子区域上的限制,即

a(i)x,ωax,ωIDix

其中,IDix是子区域D(i)的区域指示函数,当xDi时取值为1,否则为0。基于上述区域分解,全局含随机参数的偏微分方程问题(1)可转化为定义在各子区域D(i)上的局部偏微分方程系统:

L(i)x,uix,ω;a(i)x,ω=f(i)x,xDi,B(i)x,uix,ω;a(i)x,ω=g(i)x,xexDi,Cijuix,ujx=0, xjD(i),jΛi

其中,L(i)=L|DiB(i)=B|Di分别表示限制在子区域Di上的局部微分算子与局部边界算子,f(i)g(i)分别表示相应的局部源项与外部边界条件,界面算子Cij用于描述相邻子区域在公共界面jD(i)上的耦合条件,约束局部解的连续性以及通量连续性,从而保证各局部问题组合后能够形成一致的全局解。

在求得各子区域上的局部解uix,ω后,全局解ux,ω可以通过局部解的组装得到,即

ux,ω=i=1NDuix,ωIDix

显然,局部随机场a(i)x,ω继承了全局随机场ax,ω在子区域Di上的统计性质。具体而言,其均值函数可表示为μa(i)x=μaxIDix,并且在D(i)×D(i)上,其协方差函数满足Ca(i)x,x'=Ca(x,x')IDixIDix'。对每个子区域Di独立地应用截断KL展开1522,可得到局部随机场的有限维表示

a˜(i)x,ξ(i)=μa(i)x+j=1d(i)λj(i)ψj(i)xξj(i)ω

其中,ξ(i)=ξ1(i),,ξd(i)(i)T为局部随机变量组成的参数向量,其分量可定义为

ξj(i)ω=1λj(i)D(i)a(i)x,ω-μa(i)xψj(i)xdx,    j=1,2,,di

因此,局部偏微分方程系统(3)可进一步转化为以下有限维参数化问题:

L(i)x,uix,ξ(i);a˜(i)x,ξ(i)=f(i)x,xDi,B(i)x,uix,ξ(i);a˜(i)x,ξ(i)=g(i)x,xexDi,Cijuix,ujx=0,xjD(i),jΛi

引入区域分解并在各子区域上进行局部展开的核心目的在于降低参数空间的有效维数。如文献[22]所述,KL展开中特征值的衰减速率与随机场的相对相关长度la,D密切相关。该相对相关长度是场自身相关长度la与物理区域特征尺度lD的比值(即la,DlalD)。相对相关长度越大,KL特征值衰减越迅速,因而只需保留较少的KL展开项即可有效刻画随机场中的主要不确定性。在执行非重叠区域分解后,局部子区域D(i)的几何特征尺度通常显著小于全局区域特征尺度,因而子区域上局部随机场的相对相关长度会相应增大,使局部KL特征值衰减更快。因此,相较于全局随机场的KL展开,局部随机场所需保留的KL展开项的数量d(i)会减少。

综上所述,通过将全局问题分解为一系列复杂度更低的局部子区域问题,并在各子区域上进行局部参数化表示,可以有效降低代理模型的构建难度。同时,该局部化框架适合并行计算,有助于在保证近似精度的前提下提升大规模参数化偏微分方程的求解效率。

3 算子学习

算子学习是科学机器学习领域的重要研究方向,核心目标是构建能够逼近函数空间之间映射关系的深度学习模型。考虑输入函数a:DaRna和输出函数u:DuRnu,并分别用𝒜(Da;Rna)𝒰(Du;Rnu)表示其所属的函数空间。定义算子G: 𝒜𝒰,用于刻画输入函数空间和输出函数空间之间的映射关系,即Ga=u。以含随机参数的偏微分方程系统(1)为例,从随机输入场ax,ω到系统响应解ux,ω的映射关系可由参数化神经算子Gθ近似表示为:

GθaGa=u,

其中θ表示神经网络参数,最优参数θ*可通过最小化预测解与真实解的误差得到,即

θ*=argminθΩCux,ω,Gθax,ωdP(ω),

其中,C:𝒰×𝒰R为非负代价泛函,用于度量预测解Gθa,ω与真实解u,ω之间的误差。

在实际计算中,通常给定由高保真数值求解器生成的训练样本集ak,ukk=1N,其中ak表示第k个输入样本,uk表示对应的高保真解。基于蒙特卡罗方法23对积分进行近似,得到神经算子的经验损失函数如下:

θ=1Nk=1NCuk,Gθak

因此,神经算子的训练过程可视为在给定样本集上求解如下优化问题:

θ*=argminθ θ

通过上述训练,神经算子能够学习从输入函数a到输出函数u的映射关系。训练完成后,对于新的随机输入,可直接通过前向推理得到对应系统响应的近似解,从而避免对每个新样本求解原始偏微分方程,提高参数化偏微分方程问题的预测效率。

3.1 深度算子网络

深度算子网络9是一类具有较强表达能力的神经算子模型,其核心架构包含两个子网络:主干网络和分支网络。主干网络以空间坐标xDu作为输入,输出一个p维特征向量

tx=t1x,t2x,,tpxTRp

分支网络以输入函数的离散化观测值a˜作为输入,输出对应的p维系数向量

ba˜=b1a˜,b2a˜,,bpa˜TRp

对于给定的输入函数样本a˜和空间坐标x,神经算子Gθ的近似预测值可由分支网络与主干网络输出向量的内积得到,即

Gθa˜x=k=1pbka˜tkx+b0,

其中,b0为可训练的偏置项,θ表示网络中所有可训练参数。

在深度算子网络结构中,主干网络学习定义域Du上的一组基函数t1,t2,,tp,而分支网络则根据输入函数a˜预测对应的展开系数b1,b2,,bp。由于空间坐标x作为独立输入传入主干网络,训练完成后的模型可以在定义域内任意空间位置进行查询,从而具有一定的无网格预测能力。该特性使深度算子网络能够灵活适配不同空间采样点上的预测需求,并为参数化偏微分方程的快速求解提供有效的代理模型。根据分支网络和主干网络的组合方式,深度算子网络可进一步分为非堆叠式架构和堆叠式架构,本文采用非堆叠式的网络架构,如图2所示。

3.2 局部算子学习

本文将区域分解方法引入算子学习框架,为每个子区域独立构建局部神经算子。对每个子区域D(i)及其相邻子区域指标集合Λ(i),令ϕi,j(x)表示定义在公共界面jD(i)上的未知界面函数,其中jΛ(i)。则子区域D(i)的内部界面条件函数βi(x)可以定义为

βix=jΛ(i)ϕi,j(x)IjD(i)x,  xinDi

在局部算子训练过程中,为保证后续全局组装时相邻子区域解及其通量在界面上的连续性,局部界面条件函数βi(x)需要与随机场参数a(i)共同作为神经算子的输入。针对此类多输入函数映射问题,本文采用多输入算子网络架构24,即为每一个输入函数分别构建独立的分支网络。对于输入函数a˜β˜,局部神经算子的输出可表示为

Gθa˜,β˜x=k=1pbk(1)a˜bk2(β˜)tkx+b0

其中,bk1(a˜)bk2(β˜)分别为两个分支网络输出的第k个分量,tkx为主干网络输出的第k个分量,b0为可训练偏置项。

考虑定义在子区域D(i)上的局部偏微分方程系统(3),记相应的局部算子为Giai,βi=u(i)。在获取样本集a˜i,k,β˜i,k,ui,kk=1N后,局部神经算子Gθ(i)可通过最小化如下经验损失函数进行训练:

(i)θ=1Nk=1NCui,k,Gθ(i)a˜i,k,β˜i,k,

其中,a˜iβ˜i分别为输入函数ai和界面条件函数βi的离散化表示,C为度量预测解与参考解之间误差的非负代价泛函。

通过上述训练方式,每个局部神经算子能够学习从局部随机参数及界面条件到局部解的映射关系。由于不同子区域的局部算子可独立训练,该结构可以降低全局算子学习的拟合复杂度,天然具有并行计算优势。

3.3 基于优化的子域耦合

在各子区域的局部神经算子训练完成后,全局解的重构仍依赖于未知界面条件。因此,需要确定各子区域界面条件函数βi,以保证相邻子区域在公共界面上的解连续性。为此,本文采用基于优化的界面耦合策略,对界面条件进行求解。记β˜=β˜1,,β˜NDT为所有子区域界面条件离散化表示的集合。为了使相邻子区域在公共界面上满足连续性条件,构建如下界面损失函数:

interfaceβ˜=1β˜+2β˜,1β˜=(i,j)ΛjD(i)Gθ(i)a˜i,β˜ix-Gθ(j)a˜j,β˜jx2dx,2β˜=(i,j)ΛjD(i)Gθ(i)a˜i,β˜ixnx-Gθ(j)a˜j,β˜jxnx2dx,

其中n是界面处的法向量,上述界面积分可采用蒙特卡罗积分方法进行数值近似。

在最小化界面损失函数时,所有已训练完成的局部神经算子Gθi(i=1,,ND)的网络权重均保持固定,仅将界面参数集合β˜作为待优化变量。该优化问题可采用梯度下降等标准优化算法进行求解。界面参数的初始化通过对测试数据中前100个样本在接口节点上的精确解求均值得到。由于原始界面自由度较高,可进一步采用主成分分析对界面参数进行低维表示。

求得最优界面条件β˜*=β˜1,*,,β˜ND,*T后,各子区域的最优预测解可表示为

ui,*x=Gθ(i)a˜i,β˜i,*x

最后,全局预测解upred(x)可通过各局部预测解的组装得到,即

upredx=i=1NDui,*xIDix

其中𝕀Dix是子区域D(i)的区域指示函数。该组装方式能够在非重叠区域分解框架下将各局部预测结果组装为全局近似解。

4 数值算例

本节以一个含随机参数的二维扩散方程为例验证所提出的非重叠型区域分解算子学习方法的有效性,并将其与全局神经算子进行对比,以展示该方法在降低计算成本与提升预测精度方面的优势。

考虑如下带随机输入的二维扩散方程:

-Kx,ωux,ω=fx, xD,ux,ω=gx, xD,

其中,计算区域为D=-1,1×-1,1, 空间坐标记为x=(x1,x2),源项设置为

fx=2+0.6x1x2+0.3sin (πx1)sin πx2

狄利克雷边界条件定义为

gx=2+0.4x13x2+0.3sin (πx1)+0.3sin (πx2)

系统的不确定性来源于扩散系数,其定义为Kx,ω=0.5ax,ω+0.5。其中,随机场ax,ω的均值函数为μax=1,协方差函数如下:

Cax,x'=exp-x-x'2l2,  x,x'D

本实验中相关长度取为l=0.5。如图3所示,全局物理区域D被均匀划分为4个互不重叠的子区域:

D1=-1,0×-1,0,D2=-1,0×0,1,
D(3)=0,1×-1,0,D(4)=0,1×0,1 

针对全局随机场和各局部子区域随机场,分别进行KL展开。图4对比展示了全局区域和局部子区域KL展开的前20个特征值。由于4个子区域尺寸完全相同,图中仅展示其中一个子区域的局部KL特征值。可以看出,当子区域尺寸减小后,局部随机场的相对相关长度增大,局部KL展开的特征值衰减更为迅速。因此,与全局KL展开相比,局部KL展开仅需保留较少的特征项即可刻画随机场中的主要不确定性。

在构建局部神经算子的训练数据集时,需要为每个子区域Di准备局部随机场样本和内部界面条件参数。首先,对每个子区域的局部随机场执行KL展开,得到截断后的局部随机场特征对λj(i),ψj(i)xj=1d(i)。实验中统一设置局部KL展开截断项数为d(i)=12 i=1,2,3,4,全局随机场的KL展开截断项数设置为d=34。随后,从标准正态分布𝒩(0,Id(i))中采样N=2500个随机变量样本ξi,kk=1N,并结合式(4)生成对应的局部随机场样本a˜i,k(x,ξ(i))k=1N。局部神经算子所需的内部界面条件样本由少量全局样本导出。为降低界面参数维度并提取主要特征,首先选取n个界面有限元节点上的取值构成向量βix1,,βixnT。随后,对该向量应用主成分分析法25进行降维,提取其主成分系数作为最终的离散化输入β˜i,由此构成界面条件参数样本β˜i,kk=1N。接着,采用有限元方法26-27生成高保真参考解,并构造对应的局部训练样本集ui(x,a˜i,k,β˜i,k)k=1N。数值计算使用均匀方形网格,网格尺寸取h=132,并基于双线性基函数实现空间离散。同时,采用相同方式额外生成Ntest=500个样本作为测试集。最后,构建神经网络的输入:局部随机场的离散化表示a˜i和界面条件函数的离散化表示β˜ia˜i由其KL展开系数ξ(i)给出。为统一符号,将随机变量ξ(i)记为a˜i,以便与界面特征输入β˜i对应。

在每个网络结构中,分支网络和主干网络均使用等宽全连接结构,并使用双曲正切激活函数σx=tanh x=ex-e-xex+e-x。模型采用Adam优化器,初始学习率设定为0.001,总迭代次数为2000次,每训练200次后学习率衰减为原来的一半。为减少随机初始化带来的误差,所有网络均采用不同随机种子独立训练5次,并统计平均结果。在构建全局神经算子的训练数据集时,仅需在完整计算域内构造全域随机场参数,无需额外引入界面条件参数。具体而言,首先对全域随机场执行KL展开,截断项数设置为d=34,并从标准正态分布中采样N=2500 组随机样本,结合式(2)生成全局随机场实现样本。随后,采用有限元方法求解得到全域高保真参考解,有限元网格、基函数等数值离散设置均与局部算子保持一致。同样划分Ntest=500组测试样本,全局神经算子的输入由全局随机场的KL展开系数表征。为保证实验条件统一,全局神经算子沿用与子域网络相同的网络结构、激活函数、优化器、学习率衰减策略以及迭代次数,并重复5次独立训练以降低随机误差。

网络预测精度通过预测解u˜i和参考解uref(i)之间的L2相对误差进行评估,定义如下:

Elocal=u˜ix-uref(i)xDi,2uref(i)xDi,2,

其中, D,2表示定义在区域D上的L2范数,即f(x)D,2=Df2(x)dx12。在实际误差计算时,由于连续积分难以解析计算,本文采用蒙特卡罗数值积分对其进行近似。

为全面评估局部神经算子耦合重构全局解的能力,重新采样Ntest=1000个全局随机场的实现akk=1Ntest,计算对应的有限元解urefkk=1Ntest作为参考解。利用局部KL展开构造局部算子输入a˜i,kk=1Ntest,在固定局部神经算子网络参数后,采用3.3节所述的优化算法求解最优内部界面条件β˜i,kk=1Ntest,之后通过式(5)组装得到全局预测解upred(x),并计算全局L2相对误差:

Eglobal=upred(x)-uref(x)D,2uref(x)D,2

表1给出了全局神经算子与局部神经算子在5次独立实验中的平均性能对比。结果表明,相较于全局神经算子,局部神经算子在使用更小规模网络架构的情况下,能够将全局相对误差从约0.973%降低至约0.638%,相对误差降低约34%。同时,其误差标准差由0.154%降低至0.142%,表明模型预测结果更稳定、波动更小。此外,为确保对比实验的客观性,本文统一了子域网络与全局网络的训练超参数(包括批次规模与迭代次数)。虽然网络参数量不一致,但是子域网络需要额外考虑界面参数,因此二者训练耗时基本相近。在硬件资源开销方面,单个子域模型在训练阶段和预测阶段的显存占用峰值分别为47.26 MB和46.21 MB,而全局模型训练阶段和预测阶段的显存占用峰值分别为81.41 MB和66.13 MB。在推理速度方面,子域网络对单样本预测时间约为0.000 124 s,远优于全局网络预测时间的0.002 06 s。以上结果表明,局部神经算子在保证预测精度的同时具备更低的资源占用与更高的推理效率。

图5展示了测试集中某一随机样本的随机场及其对应的有限元参考解。为了更加清晰地展示结果,随机场的色标范围设置为[-1.0,4.0],有限元参考解的色标范围设置为[1.0,3.0]。

针对该特定样本,全局神经算子与基于区域分解的局部神经算子代理模型的预测结果及逐点误差分布如图6所示,该结果基于界面变量优化的非重叠区域耦合方法得到。局部神经算子训练完成后,在界面优化阶段网络参数保持固定,仅将各子区域接口处的边界输入作为待优化变量,通过最小化相邻子区域间的不连续性来实现区域之间的耦合。界面参数的初始值由测试数据中前100个样本在接口节点上的高保真解均值给出。通过主成分分析进行降维后,实验中每个子区域的待优化界面变量的维度为12。界面优化时采用Adam方法,学习率设为0.1,使用阶梯式学习率衰减策略,每100次迭代将学习率乘以0.9,界面优化总共执行1000次迭代,未设置提前停止条件。在NVIDIA GeForce RTX 4090D GPU上,对单个测试样本进行1000次界面优化的耗时约为11.348 s。可以看到,全局神经算子与局部神经算子得到的预测解在视觉上均与高保真参考解具有较好的一致性,两种方法的预测解的色标范围均设置为[1.0,3.0],逐点误差图像的色标范围均设置为[-0.1,0.1]。进一步比较逐点误差分布可以发现,得益于区域分解策略对子区域局部空间特征的精细化捕捉与精准拟合,本文所提出的局部神经算子代理模型能够显著降低区域内的误差,获得更高精度、更稳定的全局近似解。这充分体现了区域分解与局部神经算子耦合策略在提升预测精度方面的优势。

综上,二维随机扩散方程算例验证了所提出的局部神经算子的有效性和稳定性。实验结果表明,该计算框架具备良好的泛化能力与工程拓展潜力。从应用场景层面分析,该方法可进一步推广至多尺度扩散、介质热传导、地下水渗流、结构力学形变分析以及反应堆多物理场耦合等各类参数化偏微分方程求解问题。在实际工程场景中,物理场普遍存在参数随机扰动、介质非均匀分布等复杂特征。本文方法依靠较少的网络参数即可实现高精度、稳健的数值逼近,能够适配此类复杂参数化PDE的求解需求。同时,将该框架推广至实际复杂工程问题仍存在部分技术难点:后续面向不规则复杂几何计算域、多子域耦合求解以及强非线性偏微分方程问题时,需重点优化算子边界约束处理方式、完善多区域信息交互机制,并改进非线性项的拟合能力,以此进一步提升模型在复杂工程多物理场计算中的适配性与稳定性。

5 结论

本文提出了一种融合非重叠区域分解与深度神经算子的代理模型新框架,通过全局剖分、局部KL展开实现高维随机参数的降维,并提出基于界面连续性约束的优化策略,实现了参数化偏微分方程全局解的精准重构与快速推理。二维随机扩散方程的数值结果验证了该方法的有效性:相较于全局神经算子,基于区域分解的单个局部算子模型仅需约20%的网络参数量,即可实现更高精度的预测结果,且对物理场局部特征具备更敏锐的刻画能力。该框架有效缓解了传统全局算子在高维问题中计算负担重、拟合能力有限的缺陷,为复杂高维参数化偏微分方程代理模型的构建提供了一种高效、可扩展的全新技术路径。针对本文方法,后续仍存在进一步优化与完善的空间:未来可通过设计更稳健高效的界面优化算法,进一步提升多子域拼接精度;同时可深度融合控制方程、边界条件等物理先验信息,改善模型在小样本场景下的建模可靠性,以更好地适配高维复杂偏微分方程的求解场景。

参考文献

[1]

Brenner S CScott L R.The mathematical theory of finite element methods [M].New York: Springer, 2008.

[2]

Morton K WMayers D F.Numerical solution of partial differential equations: An introduction [M].Cambridge: Cambridge University Press, 2005.

[3]

Xiu DKarniadakis G E.The Wiener-Askey polynomial chaos for stochastic differential equations [J].SIAM J Sci Comput200224(2): 619-644.

[4]

Xiu DKarniadakis G E.Modeling uncertainty in flow simulations via generalized polynomial chaos [J].J.ComputPhys2003187(1): 137-167.

[5]

Xing W WTriantafyllidis VShah A Aet al.Manifold learning for the emulation of spatial fields from computational models [J].JComputPhys2016326: 666-690.

[6]

Raissi MPerdikaris PKarniadakis G E.Numerical Gaussian processes for time-dependent and nonlinear partial differential equations [J].SIAM J Sci Comput201840(1): A172-A198.

[7]

Karniadakis G EKevrekidis I GLu Let al.Physics-informed machine learning [J].Nat Rev Phys20213(6): 422-440.

[8]

Zhu YZabaras N.Bayesian deep convolutional encoder–decoder networks for surrogate modeling and uncertainty quantification [J].J ComputPhys2018366: 415-447.

[9]

Lu LJin PPang Get al.Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators [J].Nat Mach Intell20213(3): 218-229.

[10]

Li ZKovachki NAzizzadenesheli Ket al.Fourier neural operator for parametric partial differential equations [PP/OL].(2021-05-17)[2025-10-09]

[11]

Li ZHuang D ZLiu Bet al.Fourier neural operator with learned deformations for PDEs on general geometries [J].J Mach Learn Res202324(388): 1-26.

[12]

Wang SWang HPerdikaris P.Learning the solution operator of parametric partial differential equations with physics-informed DeepONets [J].Sci Adv20217(40): eabi8605.

[13]

Li ZZheng HKovachki Net al.Physics-informed neural operator for learning partial differential equations [J].ACM/IMS J Data Sci20241(3): 1-27.

[14]

Toselli AWidlund O.Domain decomposition methods-algorithms and theory [M].Berlin:Springer, 2004.

[15]

Xu ZLiao QLi J.Domain-decomposed Bayesian inversion based on local Karhunen-Loève expansions [J].J Comput Phys2024504: 112856.

[16]

Liao QWillcox K.A domain decomposition approach for uncertainty analysis [J].SIAM J Sci Comput201537(1): A103-A133.

[17]

Xu ZXia YLiao Q.A domain-decomposed VAE method for Bayesian inverse problems [J].Int J Uncertain Quantif202414(3): 67-95.

[18]

Sun QXu XYi H.Domain decomposition learning methods for solving elliptic problems [J].SIAM J Sci Comput202446(4): A2445-A2447.

[19]

Li SXia YLiu Yet al.A deep domain decomposition method based on Fourier features [J].J Comput Appl Math2023423: 114963.

[20]

Li KTang KWu Tet al.D3M: A deep domain decomposition method for partial differential equations [J].IEEE Access20198: 5283-5294.

[21]

Ghanem R GSpanos P D.Stochastic finite elements: A spectral approach [M].New York:Courier Corporation, 2003.

[22]

Chen YJakeman JGittelson Cet al.Local polynomial chaos expansion for linear differential equations with high dimensional random inputs [J].SIAM J Sci Comput201537(1): A79-A102.

[23]

Caflisch R E.Monte Carlo and quasi-Monte Carlo methods [J].Acta Numer19987: 1-49.

[24]

Jin PMeng SLu L.MIONet: Learning multiple-input operators via tensor product [J].SIAM J Sci Comput202244(6): A3490-A3514.

[25]

Jolliffe I TCadima J.Principal component analysis: a review and recent developments [J].Philos Trans R Soc A Math Phys Eng Sci2016374(2065): 20150202.

[26]

Elman H CRamage ASilvester D J.IFISS: A computational laboratory for investigating incompressible flow problems [J].SIAM Rev201456(2): 261-273.

[27]

Elman H CSilvester D JWathen A J.Finite elements and fast iterative solvers: With applications in incompressible fluid dynamics [M].Oxford: Oxford University Press, 2014.

基金资助

国家自然科学基金(12071291)

AI Summary AI Mindmap
PDF (980KB)

0

访问

0

被引

详细

导航
相关文章

AI思维导图

/