用可微分特征网格加速微分方程求解,速度提升5-20倍。
DInf-Grid: A Neural Differential Equation Solver with Differentiable Feature Grids
- 采用径向基函数插值的可微分网格结构,支持高阶导数计算。
- 多分辨率共位置网格设计,提升高频解建模与全局梯度稳定性。
- 适用于图像重建、波动场与布料模拟等物理场建模任务。
我们提出一种新型可微分网格表示方法DInf-Grid,用于高效求解微分方程(DEs)。现有基于坐标的神经网络(如正弦网络)虽灵活但计算开销大且训练慢;而基于网格的隐式表示(如Instant-NGP、K-Planes)虽训练快,却受限于线性插值,无法计算高阶导数,难以求解微分方程。DInf-Grid结合特征网格效率与径向基函数插值的无限可微性,克服上述限制。通过引入共位置的多分辨率分解,有效捕捉高频解并稳定全局梯度计算。该表示以微分方程本身作为损失函数进行隐式训练,实现对物理场的精确建模。在泊松方程图像重建、赫姆霍兹方程波场模拟、基尔霍夫-洛夫边界值布料模拟等任务中验证,相比基于坐标MLP的方法提速5-20倍,可在数秒至数分钟内完成求解,同时保持相当精度与紧凑性。
原文摘要 · Abstract (English)
We present a novel differentiable grid-based representation for efficiently solving differential equations (DEs). Widely used architectures for neural solvers, such as sinusoidal neural networks, are coordinate-based MLPs that are both computationally intensive and slow to train. Although grid-based alternatives for implicit representations (e.g., Instant-NGP and K-Planes) train faster by exploiting signal structure, their reliance on linear interpolation restricts their ability to compute higher-order derivatives, rendering them unsuitable for solving DEs. Our approach overcomes these limitations by combining the efficiency of feature grids with radial basis function interpolation, which is infinitely differentiable. To effectively capture high-frequency solutions and enable stable and faster computation of global gradients, we introduce a multi-resolution decomposition with co-located grids. Our proposed representation, DInf-Grid, is trained implicitly using the differential equations as loss functions, enabling accurate modelling of physical fields. We validate DInf-Grid on a variety of tasks, including the Poisson equation for image reconstruction, the Helmholtz equation for wave fields, and the Kirchhoff-Love boundary value problem for cloth simulation. Our results demonstrate a 5-20x speed-up over coordinate-based MLP-based methods, solving differential equations in seconds or minutes while maintaining comparable accuracy and compactness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。