PDEBench

许可证: MIT · 更新于 2026-08-01

简介

PDEBench 是 2023 年提出的系统性 PDE 基准平台,旨在为数据驱动求解器和物理信息神经网络(PINN)提供统一的评测环境。相比此前零散的基准,它在规模(数 TB)、多样性(20+ 方程类型)、任务难度(低维到高维)上均有量级提升,是当前最接近「PDE 领域 ImageNet」的公开数据集。

收录内容

覆盖以下方程类型:

  • 常微分方程(ODE):随机微分方程、迟滞系统
  • 偏微分方程(PDE):热传导、波方程、对流扩散、Navier-Stokes、浅水方程、Burgers 方程
  • 多物理场耦合:热-流体耦合、反应-扩散系统

数据规格

项目规格
总规模约 5 TB
训练样本每方程 10,000+ 条
分辨率128×128、256×256、512×512
格式HDF5(.h5),含初边值条件与参考解
许可证Apache-2.0

适用场景

  • 代理模型(Surrogate Model)训练与评测
  • PINN / DeepONet / FNO 等方法的消融实验
  • 新型网络架构的公平对比基准
  • 教学:理解不同 PDE 数值方法与数据驱动方法的差异

使用注意

  • 下载数据需访问官方 GitHub Release(或官方提供的 S3 链接),国内访问需代理
  • 官方提供 PyTorch / JAX 数据加载器,也可自行用 h5py 读取
  • 基准论文请引用:@misc{pdebench2023, title={PDEBench: A Massive Benchmark for Data-Driven PDE Solvers}, author={Takamoto et al.}, year={2023}}
标签: 基准PDECFD数据驱动机器学习
相关条目
  • PDEArena · 微软研究院开源的 PDE 基准套件,内置可微求解器与多类时空动力学数据,支持端到端评测神经算子与代理模型的长时滚动稳定性。
  • CFDBench · 专注计算流体力学(CFD)场景的多物理场数据集,覆盖内流、外流、热对流等典型工程场景,含时序与稳态数据。
  • TIDE · 2026 年发布的 3D 湍流 DNS 基准数据集,含 5 个任务与受控泛化切分,专用于评测湍流模型的外推能力。
  • DrivAerNet++ · MIT 发布的超大规模汽车外气动数据集,含 8000 个参数化车型的 CFD 流场、网格与气动系数,是气动代理模型与几何生成设计的主力语料。
  • The Well · 覆盖 16 个物理仿真领域的超大数据集,被称为「物理 ML 的 ImageNet 时刻」,配套 PyTorch API 与预训练基线,支持 HF 流式加载。