Symmetry-Block Neural PDE Solver
Implementation & benchmark of arXiv:2607.25192 — Parallelisation of Discrete Exterior Calculus via Representation Theory on Curved and Three-Dimensional Meshes
Source paper: Parallelisation of Discrete Exterior Calculus via Representation Theory on Curved and Three-Dimensional Meshes arXiv:2607.25192 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Build an implicit or unrolled graph neural operator on a symmetric simplicial mesh, and perform every symmetry-compatible linear solve in a fixed representation-theoretic basis rather than the original edge/face basis. The same basis can be reused for Poisson, Maxwell, diffusion, and learned linear combinations of DEC operators, yielding parallel independent blocks and lower peak memory without changing the discretized solution.
Formulas
Mathematical statement
Let G be a finite group acting by mesh automorphisms and let V be a cochain space, such as vertex, edge, or face cochains. The action is represented by permutation or orthogonal matrices rho(g):V to V. The paper's central equivariance relations are [rho(g),d]=0 and [rho(g),star]=0, meaning rho(g)d=d rho(g) and rho(g)star=star rho(g) for every g in G. Therefore any operator L assembled from d, the Hodge star star, their adjoints, and sums or products of these operators also satisfies [rho(g),L]=0. Representation theory decomposes V into invariant irreducible or isotypic subspaces; choosing an orthogonal symmetry-adapted basis Q gives Q^T L Q block diagonal. A practical projector onto the isotypic component of an irreducible representation alpha is P_alpha=(d_alpha/|G|) sum over g in G of chi_alpha(g)^* rho(g), where |G| is the group order, d_alpha is the irrep dimension, and chi_alpha is its character. Numerically compute bases for the ranges of these projectors and concatenate them into Q. In a multiplicity-aware basis, an intertwining operator has the stronger form Q^T L Q=direct-sum over alpha of (I_{d_alpha} tensor A_alpha), where A_alpha acts on the multiplicity space.
Implementation notes
(1) Exact integration point: use a symmetric 2-D or 3-D simplicial mesh in a neural PDE solver whose update contains an implicit step (I+tau L_theta)u_{t+1}=b_t, where L_theta=sum_j theta_j L_j is a learned linear combination of fixed DEC matrices such as the Hodge Laplacian, curl-curl, mass, or stiffness matrices. This can be an unrolled diffusion/Maxwell graph network or an implicit neural state-space model. Do not transform arbitrary pointwise nonlinearities; transform only the linear solve and linear message-passing operators guaranteed to commute with the group. (2) Preprocessing pseudocode: enumerate mesh automorphisms g in G, build their cochain permutation matrices rho(g), calculate character projectors P_alpha=(d_alpha/|G|) sum_g conjugate(chi_alpha(g)) rho(g), eigendecompose each symmetric projector, orthonormalize its range, and concatenate the columns into Q. For every fixed L_j, verify max_g ||rho(g)L_j-L_jrho(g)||_F/||L_j||_F < 1e-6. At runtime compute b_tilde=Q^T b, assemble L_tilde_theta=sum_j theta_j Q^T L_j Q, solve each diagonal block (I+tau L_tilde_theta,alpha)u_tilde_alpha=b_tilde_alpha independently, concatenate the block solutions, and return u=Q u_tilde. (3) Computed from the mathematics: the projectors, basis Q, block partition, and independent block solves. Estimated empirically: wall-clock speedup, GPU occupancy, factorization reuse, and conditioning as theta changes. If training introduces a learned matrix M, parameterize it as Q blockdiag(M_alpha) Q^T or as a linear combination of commuting basis matrices. (4) First cheap experiment: use a hexagonal periodic triangulated torus with D6 symmetry, a 10k-vertex mesh, and a 6- to 12-step implicit graph-diffusion model trained to predict one-step heat-equation evolution from random initial conditions. Compare sparse conjugate-gradient solves in the original basis against batched block solves with identical parameters and numerical tolerance. Measure training loss versus wall-clock time, peak memory, solve residual, and throughput as mesh size grows. Success is at least 2x wall-clock improvement or 1.5x lower peak memory at equal residual and identical prediction error; a secondary signal is near-linear scaling when blocks are distributed across GPU streams or CPU workers.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗
Stage 2 · Practical benchmark result: Significant win ✓
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
How to read this: The implementation beat the baseline, but the claimed mechanism did not reproduce at benchmark scale. The gain may come from another effect; treat the cause as unresolved.
Stage 1 — Mechanism check agent confidence 8/10
Built a symmetry-adapted implicit diffusion solver for a cyclic mesh using character projectors and the Fourier representation basis. All three quantitative mechanism predictions were confirmed across n={16,32,64,128,256}, theta={0,0.1,1,10}, and tau={0.01,0.1,1}: commutator error was 0, transformed off-block error was 4.03e-14, and relative solution disagreement was 3.67e-14, with residual below 4.39e-14. The algebraic mechanism is real, but this toy does not establish a consistent runtime win: median transformed-solve time was 0.223 ms versus 0.697 ms for sparse solving, while timings varied by size because dense transforms and small-problem overhead dominate.
- Agent confidence
- 8/10
- Baseline
- SciPy sparse direct solve: median 0.697 ms; maximum relative residual 4.39e-14.
- Idea
- Fourier symmetry-basis scalar block solve: median 0.223 ms; maximum relative solution error 3.67e-14; maximum transformed off-block ratio 4.03e-14; maximum commutator ratio 0.0; projector resolution, idempotence, and orthogonality errors 1.08e-14, 6.80e-15, and 1.90e-15.
Limitations: This is a 1D cyclic graph toy, not the proposed 2D/3D symmetric simplicial torus or a neural PDE model. It uses complex one-dimensional irreducible blocks and dense Fourier transforms, and does not measure GPU execution, peak memory, batched block parallelism, training loss, FLOPs, or scaling to 10k vertices. Timing is a small single-right-hand-side comparison and is not a robust production speedup estimate.
How to run: python3 symmetry_block_solver.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.25192)
Beats the tuned baseline by 100.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Symmetric_periodic_poisson
- Model
- matched_mlp_implicit_poisson
- Paired seeds
- 8
- Baseline mean
- 1542.9739
- Idea mean
- 0.1642
- Effect (Δ)
- -1542.8097 (−100.0%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±51.1%
- Mechanism
- Not confirmed ✗
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
The custom task is a small 8x8 periodic Poisson regression rather than a 2-D/3-D simplicial mesh or large neural PDE operator. No wall-clock speedup, peak memory, GPU occupancy, distributed block solves, Maxwell/DEC face or edge cochains, or scaling to 10k vertices was measured. The signature's failed equivariance check also means the benchmark does not establish that the unconstrained neural trunk learns the claimed group symmetry.
How to run: python3 bench_symmetry_pde.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Worked ✓
Agent confidence: 7/10
Built a local symmetric periodic Poisson PDE track and a matched end-to-end MLP pair differing only in the final linear solve: original-basis dense solve versus FFT symmetry-basis solve. Across 8 paired seeds, the idea had lower test MSE, with delta_mean=-1542.8097 and permutation p=0.0081, satisfying the benchmark's significant-win criterion. The mechanism signature was not confirmed: trained-model translation-equivariance error was 0.4863 and the recorded commutator check was 8.0, so the result supports a strong task-performance signal but not full learned symmetry preservation.
- Baseline
- Best lr=0.003; mean test MSE 1542.9739, std 881.7730; sweep means lr=0.001: 3678.5545, lr=0.003: 1542.9739, lr=0.01: 61191.1179.
- Idea
- Best shared lr=0.01; mean test MSE 0.16419, std 0.00518; sweep means lr=0.001: 0.33431, lr=0.003: 0.22666, lr=0.01: 0.16419. All 8 paired seeds favored the idea; delta_mean=-1542.8097, permutation p=0.0081.
Benchmark result
Beats the tuned baseline by 100.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Symmetric_periodic_poisson
- Model
- matched_mlp_implicit_poisson
- Paired seeds
- 8
- Baseline mean
- 1542.9739
- Idea mean
- 0.1642
- Effect (Δ)
- -1542.8097 (−100.0%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±51.1%
- Mechanism
- Not confirmed ✗
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
The custom task is a small 8x8 periodic Poisson regression rather than a 2-D/3-D simplicial mesh or large neural PDE operator. No wall-clock speedup, peak memory, GPU occupancy, distributed block solves, Maxwell/DEC face or edge cochains, or scaling to 10k vertices was measured. The signature's failed equivariance check also means the benchmark does not establish that the unconstrained neural trunk learns the claimed group symmetry.
How to run: python3 bench_symmetry_pde.py
Artifacts
- bench_report.json 4.0 KB View Raw
- bench_symmetry_pde.py 4.5 KB View Raw
- poisson_symmetry_track.py 1.1 KB View Raw
- report.md 1.7 KB View
- report_bench_2026-08-31T172449.md 6.1 KB View
- results.json 19.3 KB View Raw
- symmetry_block_solver.py 4.5 KB View Raw