当前位置：首页 > news >正文

【PINN】DeepXDE学习训练营(5)——function-mf_dataset.py

news 来源：原创 2025/6/20 0:11:50

一、引言

随着人工智能技术的飞速发展，深度学习在图像识别、自然语言处理等领域的应用屡见不鲜，但在科学计算、工程模拟以及物理建模方面，传统的数值方法仍然占据主导地位。偏微分方程（Partial Differential Equations, PDEs）作为描述自然界中众多复杂现象的重要数学工具，在物理、化学、工程、金融等领域具有广泛应用。然而，伴随着高维度、多变量、复杂边界条件等挑战，传统数值求解方法面临效率低、适应性差等困境。

近年来，深度学习的崛起为科学计算带来了全新的解决思路。其中，以深度偏微分方程（Deep PDE）为代表的研究方向，通过结合神经网络与偏微分方程的理论，成功开发出高效、灵活的求解方案。这种方法不仅可以克服传统方法的局限，还能应对高维、复杂几何等问题。

作为深度偏微分方程领域的开源工具库，DeepXDE（Deep Learning for Differential Equations）由lululxvi团队精心开发，凭借其强大的功能、易用的接口和丰富的示例，受到学术界与工业界的广泛关注。本文将系统介绍DeepXDE的基本内容与应用价值，深入探讨其核心技术原理，分享环境配置与运行技巧，并结合实际案例进行分析，最后对未来发展趋势进行展望。

二、DeepXDE的用途

DeepXDE，一个基于TensorFlow和PyTorch的深度学习微分方程求解库，应运而生。它提供了一个简洁、高效且易于使用的框架，使得研究人员和工程师能够利用深度学习技术求解各种类型的微分方程，包括常微分方程（ODEs）、偏微分方程（PDEs）、积分微分方程（IDEs）以及分数阶微分方程（FDEs）。

DeepXDE旨在提供一站式的深度学习框架，用于高效求解各种偏微分方程，包括但不限于：

1. 传统偏微分方程求解

定常和非定常问题：热传导方程、波动方程、拉普拉斯方程、扩散方程等。
线性和非线性方程：支持线性边界条件，也能处理非线性、非局部问题。

2. 高维偏微分方程

在高维空间中，传统数值方法面临“维数灾难”。DeepXDE利用神经网络天然的高维逼近能力，有效解决高维PDE，如贝尔曼方程、多体问题等。

3. 复杂几何和边界条件

支持任意复杂的几何区域、非均匀边界条件，极大扩展了求解的适用范围。

4.参数逆问题和数据驱动建模

整合数据，使模型在已知部分信息的基础上进行参数识别、反演问题求解。

5. 动态系统和时间演化

支持带有时间变量的演化问题，模拟动态过程。

6. 结合有限元、有限差分等方法

虽然核心为神经网络，但兼容各种数值方法，提供灵活的求解策略。

7. 教育科研与工程实践

丰富的案例与接口帮助科研人员快速验证理论，工程师实现快速设计优化。

总结而言，DeepXDE不仅是一个纯粹的数学工具，更是工程实践中的“聪明助手”，帮助用户以信赖深度学习的方式突破传统技术瓶颈，实现创新性的科学计算。

三、核心技术原理

DeepXDE的核心思想是利用神经网络作为逼近器，通过构造损失函数，使网络能在满足偏微分方程边界条件的前提下逼近真实解。以下详细阐释其原理基础。

1. 神经网络逼近偏微分方程解

假设待求解的偏微分方程可以写成：

$\mathcal{N}[u](x)=0,x\in\Omega,$

配合边界条件

$\mathcal{B}[u](x)=g(x),x\in\partial \Omega,$

这里， $\mathcal{N}$ 代表微分算子， $\mathcal{B}$ 代表边界条件算子。

DeepXDE利用深度神经网络 𝑢𝜃(𝑥) 作为解的逼近，参数为 𝜃 。通过自动微分（AutoDiff），网络可以自然求出 𝑢𝜃 的各阶导数，从而在网络定义的每个点上计算微分方程的残差。

2. 损失函数设计

训练模型的目标是最小化残差，使神经网络逼近满足偏微分方程的解。损失函数由两部分组成：

方程残差部分：

$\mathcal{L}_{\mathit{PDE}}=\frac{1}{N_{f}}\sum_{i=1}^{N_{f}}|\mathcal{N}[u_{ \theta}](x_{f}^{(i)})|^{2},$

其中， $x{_{f}}^{(i)}$ 为采样点，用于评估微分残差。

边界条件部分：

$\mathcal{L}_{\mathit{BC}}=\frac{1}{N_{b}}\sum_{i=1}^{N_{b}}|\mathcal{B}[u_{ \theta}](x_{b}^{(i)})-g(x_{b}^{(i)})|^{2},$

结合整体目标函数：

$\mathcal{L}(\theta)=\lambda_{f}\mathcal{L}_{\mathit{PDE}}+\lambda_{b} \mathcal{L}_{\mathit{BC}},$

这里 $\lambda {_{f}}$ 、 $\lambda {_{b}}$ 为调节系数。

3. 自动微分（AutoDiff）技术

深度学习框架如TensorFlow或PyTorch提供自动微分功能，方便快速计算神经网络输入的微分，自动应用链式法则求导，极大简化偏微分方程的数值差分表达。

4. 训练优化方法

利用成熟的梯度下降（SGD）、Adam等优化算法，通过反向传播调节神经网络参数，使损失函数达到最小。

5. 样本生成和采样策略

采样点生成：采用随机采样、拉丁超立方（Latin Hypercube Sampling）或网格采样来选取训练点。
自适应采样：在训练过程中，根据误差分布调整采样点，提高训练效率。

6. 复杂边界与几何的处理

采用非结构化的几何描述和SDF（Signed Distance Function）结合，保证不同几何形状的灵活支持。

7. 逆问题与数据融合

在已知数据集上引入数据损失，使模型不仅满足PDE，也通过端到端训练实现数据匹配，增强实际适用性。

五、代码详解

DeepXDE 多保真度数据集训练示例代码注解

这段代码展示了如何使用 DeepXDE 库来训练一个多保真度神经网络模型，用于处理不同精度级别的数据。下面是详细的代码注解：

"""Backend supported: tensorflow.compat.v1"""
# 这行注释说明了支持的后端框架，仅支持 TensorFlow 1.x 兼容模式import deepxde as ddefname_lo_train = "../dataset/mf_lo_train.dat"
fname_hi_train = "../dataset/mf_hi_train.dat"
fname_hi_test = "../dataset/mf_hi_test.dat"
# 定义三个数据文件的路径：
# - 低保真度训练数据文件
# - 高保真度训练数据文件
# - 高保真度测试数据文件data = dde.data.MfDataSet(fname_lo_train=fname_lo_train,fname_hi_train=fname_hi_train,fname_hi_test=fname_hi_test,col_x=(0,),col_y=(1,),standardize=True,
)
# 创建多保真度数据集对象：
# - 指定低保真度训练数据文件
# - 指定高保真度训练数据文件
# - 指定高保真度测试数据文件
# - col_x=(0,) 表示输入特征在数据文件的第0列
# - col_y=(1,) 表示输出标签在数据文件的第1列
# - standardize=True 表示对数据进行标准化处理（均值为0，标准差为1）activation = "tanh"
# 设置激活函数为双曲正切函数initializer = "Glorot uniform"
# 设置权重初始化方法为 Glorot 均匀分布初始化（也称为 Xavier 初始化）regularization = ["l2", 0.01]
# 设置 L2 正则化，权重为 0.01，用于防止过拟合net = dde.nn.MfNN([1] + [20] * 4 + [1],[10] * 2 + [1],activation,initializer,regularization=regularization,
)
# 创建多保真度神经网络 (MfNN) 模型：
# - 第一个参数 [1] + [20] * 4 + [1] 定义了主网络（高保真度）的结构：
#   输入层1个节点，4个隐藏层各20个节点，输出层1个节点，即 [1, 20, 20, 20, 20, 1]
# - 第二个参数 [10] * 2 + [1] 定义了分支网络（低保真度）的结构：
#   2个隐藏层各10个节点，输出层1个节点，即 [10, 10, 1]
# - 使用 tanh 激活函数
# - 使用 Glorot uniform 初始化方法
# - 应用 L2 正则化model = dde.Model(data, net)
# 将数据集和神经网络组合成一个完整的模型model.compile("adam", lr=0.001, metrics=["l2 relative error"])
# 编译模型：
# - 使用 Adam 优化器
# - 学习率设为 0.001
# - 评估指标为 L2 相对误差losshistory, train_state = model.train(iterations=80000)
# 训练模型，迭代 80000 次，并返回损失历史和训练状态dde.saveplot(losshistory, train_state, issave=True, isplot=True)
# 保存并绘制训练过程中的损失和指标变化：
# - issave=True 表示保存训练历史到文件
# - isplot=True 表示生成可视化图表

代码功能总结
这段代码实现了以下功能：

1. 从文件中加载多保真度数据集，包括低保真度训练数据、高保真度训练数据和高保真度测试数据
2. 构建一个多保真度神经网络模型，包括一个主网络（处理高保真度数据）和一个分支网络（处理低保真度数据）
3. 使用 Adam 优化器训练模型，迭代80000次
4. 保存训练历史并生成可视化图表
多保真度学习是一种特殊的机器学习方法，它可以同时利用低成本但不太准确的数据（低保真度）和高成本但更准确的数据（高保真度）来训练模型。这种方法在计算成本高昂的领域（如物理模拟、工程设计等）特别有用，因为它可以减少对昂贵高保真度数据的依赖，同时保持较高的预测精度。

六、总结与思考

DeepXDE作为深度偏微分方程求解的先进工具，展现出强大的学术研究与工程应用潜力。其基于自动微分的深度学习框架，使得复杂偏微分方程在高维、多几何场景下的求解变得更为高效、灵活。相比传统数值方法，DeepXDE具有架构简单、扩展性强、支持数据融合等优点，极大地拓展了偏微分方程的应用边界。

然而，深度学习方法仍面临一些挑战，比如训练的不稳定性、超参数调优的复杂性、理论基础的逐步完善等。未来，随着硬件性能的提升、算法的不断创新，DeepXDE有望在更高维度、更复杂的物理场景中表现出更强的竞争力。

在科学研究中，DeepXDE不仅是验证创新理论的实验平台，更是推动工程实践创新的桥梁。从基础数学模型到端到端的数据驱动建模，深度偏微分方程代表了科学计算的未来方向。我们应积极探索其潜力，推动其在实际问题中的落地，为解决复杂系统的大规模仿真提供更强的工具。

【作者声明】

本文为个人原创内容，基于对DeepXDE开源项目的学习与实践整理而成。如涉及引用他人作品，均注明出处。转载请注明出处，感谢关注。

【关注我们】

如果您对神经网络、群智能算法及人工智能技术感兴趣，请关注【灵犀拾荒者】，获取更多前沿技术文章、实战案例及技术分享！