【PINN】DeepXDE学习训练营(7)——operator-ADR_solver.py

灵犀拾荒者

于 2025-04-28 10:00:00 发布

阅读量1.8k

点赞数 62

分类专栏： Py PINN 文章标签：学习神经网络 PINN

本文链接：https://blog.csdn.net/2303_77200324/article/details/147414545

版权

Py 同时被 2 个专栏收录

73 篇文章

订阅专栏

PINN

28 篇文章

订阅专栏

一、引言

随着人工智能技术的飞速发展，深度学习在图像识别、自然语言处理等领域的应用屡见不鲜，但在科学计算、工程模拟以及物理建模方面，传统的数值方法仍然占据主导地位。偏微分方程（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，也通过端到端训练实现数据匹配，增强实际适用性。

五、一维ADR方程求解器

这段代码实现了一个一维对流-扩散-反应(Advection-Diffusion-Reaction, ADR)方程的数值求解器。下面是详细的代码注解：

import matplotlib.pyplot as plt
import numpy as np


def solve_ADR(xmin, xmax, tmin, tmax, k, v, g, dg, f, u0, Nx, Nt):
    """Solve 1D
    u_t = (k(x) u_x)_x - v(x) u_x + g(u) + f(x, t)
    with zero boundary condition.
    """
    # 创建空间和时间网格
    x = np.linspace(xmin, xmax, Nx)  # 空间离散点
    t = np.linspace(tmin, tmax, Nt)  # 时间离散点
    h = x[1] - x[0]  # 空间步长
    dt = t[1] - t[0]  # 时间步长
    h2 = h**2  # 空间步长的平方，用于计算

    # 构建有限差分算子
    D1 = np.eye(Nx, k=1) - np.eye(Nx, k=-1)  # 中心差分算子，用于一阶导数
    D2 = -2 * np.eye(Nx) + np.eye(Nx, k=-1) + np.eye(Nx, k=1)  # 中心差分算子，用于二阶导数
    D3 = np.eye(Nx - 2)  # 内部点的单位矩阵
    
    # 计算扩散系数
    k = k(x)
    # 构建扩散项矩阵
    M = -np.diag(D1 @ k) @ D1 - 4 * np.diag(k) @ D2
    # 边界条件处理，只考虑内部点
    m_bond = 8 * h2 / dt * D3 + M[1:-1, 1:-1]
    
    # 计算对流系数
    v = v(x)
    # 构建对流项矩阵
    v_bond = 2 * h * np.diag(v[1:-1]) @ D1[1:-1, 1:-1] + 2 * h * np.diag(
        v[2:] - v[: Nx - 2]
    )
    # 合并扩散和对流项
    mv_bond = m_bond + v_bond
    # 构建右侧项系数矩阵
    c = 8 * h2 / dt * D3 - M[1:-1, 1:-1] - v_bond
    # 计算源项
    f = f(x[:, None], t)

    # 初始化解矩阵
    u = np.zeros((Nx, Nt))
    # 设置初始条件
    u[:, 0] = u0(x)
    
    # 时间推进循环
    for i in range(Nt - 1):
        # 计算非线性反应项
        gi = g(u[1:-1, i])
        # 计算非线性反应项的导数
        dgi = dg(u[1:-1, i])
        # 构建反应项矩阵
        h2dgi = np.diag(4 * h2 * dgi)
        # 构建系统矩阵
        A = mv_bond - h2dgi
        # 构建右侧向量的第一部分（源项和反应项）
        b1 = 8 * h2 * (0.5 * f[1:-1, i] + 0.5 * f[1:-1, i + 1] + gi)
        # 构建右侧向量的第二部分（前一时间步的贡献）
        b2 = (c - h2dgi) @ u[1:-1, i].T
        # 求解线性系统得到下一时间步的解
        u[1:-1, i + 1] = np.linalg.solve(A, b1 + b2)
    
    return x, t, u


def main():
    # 设置第一个测试案例的参数
    xmin, xmax = -1, 1  # 空间区间
    tmin, tmax = 0, 1  # 时间区间
    k = lambda x: x**2 - x**2 + 1  # 扩散系数（注意：x**2 - x**2 = 0，所以实际上是常数1）
    v = lambda x: np.ones_like(x)  # 对流系数（常数1）
    g = lambda u: u**3  # 非线性反应项
    dg = lambda u: 3 * u**2  # 非线性反应项的导数
    f = (
        lambda x, t: np.exp(-t) * (1 + x**2 - 2 * x)
        - (np.exp(-t) * (1 - x**2)) ** 3
    )  # 源项
    u0 = lambda x: (x + 1) * (1 - x)  # 初始条件
    u_true = lambda x, t: np.exp(-t) * (1 - x**2)  # 真实解（用于验证）

    # 第二个测试案例的参数（被注释掉了）
    # xmin, xmax = 0, 1
    # tmin, tmax = 0, 1
    # k = lambda x: np.ones_like(x)
    # v = lambda x: np.zeros_like(x)
    # g = lambda u: u ** 2
    # dg = lambda u: 2 * u
    # f = lambda x, t: x * (1 - x) + 2 * t - t ** 2 * (x - x ** 2) ** 2
    # u0 = lambda x: np.zeros_like(x)
    # u_true = lambda x, t: t * x * (1 - x)

    # 设置网格分辨率
    Nx, Nt = 100, 100  # 空间和时间的网格点数
    
    # 求解ADR方程
    x, t, u = solve_ADR(xmin, xmax, tmin, tmax, k, v, g, dg, f, u0, Nx, Nt)

    # 计算并打印数值解与真实解的最大误差
    print(np.max(abs(u - u_true(x[:, None], t))))
    
    # 绘制数值解
    plt.plot(x, u)
    plt.show()


if __name__ == "__main__":
    main()

代码功能总结
这段代码实现了一个一维对流-扩散-反应(ADR)方程的数值求解器，具体功能如下：

1. 求解的方程：
u_t = (k(x) u_x)_x - v(x) u_x + g(u) + f(x, t)

其中：
- u_t 表示 u 对时间的偏导数
- (k(x) u_x)_x 是扩散项
- v(x) u_x 是对流项
- g(u) 是非线性反应项
- f(x, t) 是源项
2. 数值方法：
- 空间上使用有限差分法离散
- 时间上使用隐式格式推进
- 对非线性项采用线性化处理
3. 主要步骤：
- 构建空间和时间网格
- 构建有限差分算子
- 处理扩散、对流和反应项
- 时间循环求解线性系统
- 与真实解比较误差并可视化结果
4. 测试案例：
- 代码提供了两个测试案例（第二个被注释掉了）
- 每个案例都有对应的真实解，用于验证数值解的准确性
这种数值求解器在流体力学、热传导、物质扩散等物理过程的模拟中非常有用，是科学计算和偏微分方程数值解的重要组成部分。