---
title: 'PDEZoo: 2D Elliptic PDE Benchmark'
url: https://www.emergentmind.com/topics/pdezoo
type: topic
---

# PDEZoo: 2D Elliptic PDE Benchmark

Searching arXiv for PDEZoo and closely related benchmark papers to ground the article.
PDEZoo is a large-scale benchmark for **elliptic PDE boundary-value problems** in \(2\)D. Each instance is a complete problem \((\Omega,\mathcal{E},f,g,u)\) consisting of a domain \(\Omega\), an elliptic operator family \(\mathcal{E}\), a forcing term \(f\), a Dirichlet boundary condition \(g\), an analytic solution \(u\), and multi-budget Monte Carlo Walk-on-Spheres approximations \(\{\hat u_B\}\). It was introduced together with the hybrid solver MC\(^2\), and is described as “the largest standardized elliptic PDE benchmark to date: 2M PDEs spanning five elliptic families and unlimited geometric compositions, with analytic ground truth and multi-budget Monte Carlo trajectories” [2605.09288].

## 1. Scope and defining problem class

PDEZoo is organized around **elliptic PDE boundary-value problems** on bounded domains \(\Omega\subset\mathbb{R}^2\), with all benchmark instances defined on subsets of \([-1,1]^2\). The benchmark’s general tuple is
\[
\mathcal{P} = (\Omega,\mathcal{E},f,g,u),
\]
where \(u\) is known analytically, \(f\) is derived from \(u\), and \(g=u|_{\partial\Omega}\). The paper presents, for context, the broader elliptic template
\[
\begin{cases}
\nabla \cdot (\alpha(x)\nabla u(x)) + \omega(x)\nabla u(x) - \sigma(x)u(x) = -f(x), & x \in \Omega, \\
u(x) = g(x), & x \in \partial\Omega ,
\end{cases}
\]
while emphasizing that PDEZoo itself uses five explicit elliptic families with **Dirichlet boundary conditions** [2605.09288].

The benchmark is restricted to \(2\)D elliptic problems, and this restriction is central to its design. A stated motivation is that existing PDE benchmarks largely emphasize time-dependent parabolic or hyperbolic systems on fixed rectangular grids, whereas PDEZoo targets elliptic problems with irregular geometries, analytic ground truth, and explicit finite-compute solver trajectories. A plausible implication is that PDEZoo is intended not merely as a dataset of converged solutions, but as infrastructure for studying **solver behavior under compute constraints** [2605.09288].

## 2. Elliptic families and manufactured solutions

PDEZoo covers five elliptic PDE families. For all of them, the analytic solution is first constructed, after which the forcing is derived exactly from the relevant operator. The boundary condition is then obtained by restriction to \(\partial\Omega\), so \(g=u|_{\partial\Omega}\) [2605.09288].

| Family | PDE | Derived forcing or parameter |
|---|---|---|
| Laplace | \(\Delta u = 0\) | \(f\equiv 0\) |
| Poisson | \(\Delta u = f\) | \(f=\Delta u\) |
| Yukawa | \(\Delta u - \lambda u = f\) | \(\lambda \sim \mathcal{U}[0.5,50.0]\), \(f=\Delta u-\lambda u\) |
| Biharmonic | \(\Delta^2 u = f\) | \(f=\Delta^2 u\) |
| Helmholtz | \(\Delta u + k^2 u = f\) | \(k \sim \mathcal{U}[0.5,10.0]\), \(f=\Delta u+k^2u\) |

The benchmark uses a method-of-manufactured-solutions pipeline. For **Poisson, Yukawa, Biharmonic, and Helmholtz**, \(u\) is assembled from a general smooth function family. For **Laplace**, \(u\) is assembled from **harmonic atoms**, so \(\Delta u=0\) analytically. The construction samples a random number of atoms
\[
n \sim \mathrm{Uniform}\{2,\dots,6\},
\]
then forms
\[
u(x,y)=\sum_{i=1}^n a_i\,\phi_i(x,y;\theta_i).
\]
The general pool contains **20 atom types**, and the harmonic pool contains **12 harmonic atoms**. “Hard atoms” are over-sampled using `hard_atom_extra = 4`, which increases their probability from roughly \(25\%\) to approximately \(63\%\) [2605.09288].

This design makes the forcing \(f\) exact with respect to the benchmarked PDE family. It also separates benchmark error from reference-solver discretization error, because the “ground truth” field is obtained by direct evaluation of an analytic expression rather than by numerically solving the PDE [2605.09288].

## 3. Geometry, signed distance functions, and domain composition

A distinctive feature of PDEZoo is its geometry model. Domains are represented by **signed distance functions (SDFs)** satisfying
- \(\mathrm{sdf}(x)>0\) inside the domain,
- \(\mathrm{sdf}(x)=0\) on the boundary,
- \(\mathrm{sdf}(x)<0\) outside the domain [2605.09288].

PDEZoo defines **8 primitive domain families**: disk, square, rectangle, ellipse, annulus, triangle, hexagon, and stadium. On top of these primitives it builds composed domains using Boolean operations. The SDFs for the compositions are
\[
\mathrm{sdf}_{\mathrm{union}}(x)=\max(\mathrm{sdf}_A(x),\mathrm{sdf}_B(x)),
\]
\[
\mathrm{sdf}_{\mathrm{inter}}(x)=\min(\mathrm{sdf}_A(x),\mathrm{sdf}_B(x)),
\]
\[
\mathrm{sdf}_{\mathrm{diff}}(x)=\min(\mathrm{sdf}_A(x),-\mathrm{sdf}_B(x)).
\]
The paper states that these composed SDFs are **conservative underestimates** of the true boundary distance, which is crucial for the correctness of Walk-on-Spheres because every WoS ball must remain entirely inside \(\Omega\) [2605.09288].

The procedural generator samples either a primitive or a composed domain. For the training distribution, each primitive is selected with probability \(1/11\), while composed domains are selected with probability \(3/11\). Compositions use two primitives drawn from disk, rectangle, ellipse, and triangle, with random parameters, rotations, offsets, and one of union, intersection, or difference [2605.09288].

Instance generation is filtered by **8 quality filters**, with up to **40 retries**. These include finite-value checks for \(u\) and \(f\), domain non-degeneracy, a median-SDF threshold for composed domains, amplitude constraints on \(\mathrm{std}(u)\) and \(\mathrm{std}(f)\), bounded boundary values, and a requirement that all ground-truth values be finite on the \(256\times256\) evaluation grid. This suggests that PDEZoo is engineered not only for diversity, but also for numerical regularity across a very large procedural corpus [2605.09288].

## 4. Multi-budget Walk-on-Spheres trajectories and dataset organization

PDEZoo discretizes all fields on a fixed **\(256\times256\)** grid over \([-1,1]^2\). Its canonical stochastic solver is **Walk-on-Spheres (WoS)**, described as mesh-free, geometry-agnostic, and unbiased up to a small \(\varepsilon\)-shell termination bias. The WoS configuration uses
- grid \(256\times256\),
- shell threshold \(\varepsilon = 10^{-4}\),
- max steps per walk \(128\) for training and \(256\) for evaluation,
- walk batch size \(256\) [2605.09288].

The benchmark’s scale is split into a large training corpus and a smaller but broader test corpus:
- **Training split**: **2,000,000 instances**, containing only Laplace, Poisson, and Yukawa, with WoS budgets
  \[
  \mathcal{B}_{\mathrm{train}}=\{1,2,4,8,16,32\}.
  \]
- **Test split**: **5,000 instances**, spanning all five families, with **20 budgets** from
  \[
  B=1 \text{ to } B=131{,}072.
  \]
Helmholtz and Biharmonic are test-only families, so they function as **OOD evaluation** targets [2605.09288].

For each budget \(B\), the field package stores three \(256\times256\) float32 arrays:
- `noisy`: the WoS estimate \(\hat U_B\),
- `clean`: the analytic field \(U^\star\),
- `mask`: the interior-domain indicator [2605.09288].

Metadata are stored separately in JSONL format. Each entry includes the case identifier, PDE family, family parameters such as \(\lambda\) or \(k\), a **PyTorch expression string** for \(u(x,y)\), domain parameters, atom types and parameters, and hardness-related statistics. This makes the benchmark both grid-based and regenerable from symbolic descriptions [2605.09288].

The paper also defines difficulty tiers using WoS MSE at \(B=32\):
- Easy: MSE \(<10^{-2}\), about \(20\%\),
- Medium: \(10^{-2}\le \mathrm{MSE}<1\), about \(41\%\),
- Hard: \(1\le \mathrm{MSE}<10^2\), about \(30\%\),
- Very Hard: \(\mathrm{MSE}\ge 10^2\), about \(8\%\) [2605.09288].

## 5. Finite-compute benchmarking and relation to MC\(^2\)

PDEZoo was introduced alongside **MC\(^2\)**, a hybrid WoS–neural-network solver that uses a low-budget Monte Carlo field as a structured estimator and learns a single-pass correction. In the benchmark, MC\(^2\) is trained on PDEZoo instances using low-budget WoS fields \(\hat U_B\), the source term field \(f\), and the domain mask, with the analytic field \(U^\star\) as target. The training regime uses a random subset of **100k** training instances and budgets \(B\in\{4,8,16,32\}\); evaluation commonly uses a fixed budget such as \(B=8\) [2605.09288].

The principal metrics reported on PDEZoo are masked **MSE**, **PSNR**, **SNR**, and **LPIPS**. Because the test split includes multiple WoS budgets for the same instance, the benchmark supports direct **accuracy–compute** comparisons. The paper reports that WoS MSE follows the expected \(\mathcal{O}(1/N)\) slope over budget on sampled test cases, making the benchmark suitable for studying how much learned correction can replace brute-force Monte Carlo compute [2605.09288].

Within that evaluation protocol, MC\(^2\) using only \(B=8\) WoS is reported to achieve **PSNR \(\approx 44\)–\(45\) dB overall**, comparable to WoS at \(B\approx16{,}384\)–\(32{,}768\). The paper characterizes this as solving PDEs approximately **\(1000\times\)** faster than pure WoS at matched accuracy. It also reports zero-shot generalization from training families to held-out **Biharmonic** and **Helmholtz** test problems [2605.09288].

PDEZoo is nevertheless method-agnostic in design. The benchmark is used to evaluate raw WoS at many budgets, classical denoisers, **FNO**, **PINO**, **DiffusionPDE**, and **CoCoGen**. The stated purpose is broader than benchmarking a single method: to support reproducible study of **finite-compute elliptic PDE solving** [2605.09288].

## 6. Position within the broader PDE benchmark ecosystem

PDEZoo occupies a different niche from several other recent PDE resources. **APEBench** is a benchmark for **autoregressive neural emulators** of time-dependent PDEs, built around a differentiable pseudo-spectral solver and covering **46 distinct PDEs across 1D, 2D, and 3D**; it emphasizes rollout metrics and temporal generalization rather than elliptic boundary-value solving under finite Monte Carlo budgets [2411.00180]. **VisualPDE** is an online interactive solver for a broad class of **1D and 2D** PDE systems, designed for rapid exploration and sharing rather than standardized large-scale benchmarking with analytic elliptic ground truth [2308.01245]. The diagnostic suite in “A Diagnostic Software Suite for Auditing Learned PDE Simulators” provides architecture-independent, post hoc diagnostics such as semigroup consistency, generator discrepancy, energy behavior, integral balance, admissibility constraints, perturbation response, and scaling-law consistency for learned **evolution operators**, which is complementary to PDEZoo’s focus on elliptic field accuracy and finite-compute Monte Carlo trajectories [2606.18200].

Within that landscape, PDEZoo’s defining characteristics are its restriction to **2D elliptic PDEs**, its use of **analytic ground truth** rather than numerically converged reference solves, its **SDF-based geometric diversity**, and its release of **multi-budget stochastic solver outputs**. The paper also notes several limitations: the benchmark is confined to \(2\)D elliptic problems, it is tied to Walk-on-Spheres and related Green’s-function machinery, and it relies on manufactured solutions rather than directly modeling application-specific physical datasets. This suggests that PDEZoo is best understood as a specialized benchmark for **elliptic solver evaluation under controlled geometry and compute variation**, rather than a universal PDE benchmark [2605.09288].

Source: https://www.emergentmind.com/topics/pdezoo