Lune

ICML2026Top-tier venue

Hermite-NGP: Gradient-Augmented Hash Encoding for Learning PDEs

Jinjin He, Zhiqi Li, Sinan Wang, Bo Zhu

2026Year

Abstract

We propose Hermite-NGP, a gradient-augmented multi-resolution hash encoding designed to enable fast and accurate computation of spatial derivatives for neural PDE solvers. Unlike existing NGP-based approaches that rely on automatic differentiation or finite differences and suffer from instability or high cost, Hermite-NGP explicitly stores function values and mixed partial derivatives at hash grid vertices, allowing fully analytic evaluation of gradients, Jacobians, and Hessians via Hermite interpolation. This design preserves the efficiency and spatial adaptivity of NGP while supporting analytic differential operators up to second order. We further introduce a multi-resolution curriculum training strategy analogous to multigrid V-cycles to enable coarse-to-fine optimization. Across a range of 2D and 3D PDE benchmarks, Hermite-NGP achieves up to ∼20×{\sim}20{\times} lower error than prior neural PDE methods, and reduces wall-clock convergence time by 22 – 10×10\times compared to other solvers, with per-epoch training times as low as 3.5ms3.5 \mathrm{ms} for models with up to 1717M parameters.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext fee54d5a-cb69-46a9-9828-9849cd032028

Builds on22

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines