arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.13397cs.LGphysics.comp-ph
收藏

使用哈希编码的高效物理信息神经网络

Efficient physics-informed neural networks using hash encoding

King Abdullah University of Science and Technology(阿卜杜拉国王科技大学)

Xinquan Huang, Tariq Alkhalifah

AI总结:

本文提出将多分辨率哈希编码引入物理信息神经网络,并用有限差分替代自动微分以处理导数不连续性,在Burgers、Helmholtz和Navier-Stokes方程上实现了约10倍的训练效率提升。

摘要:

物理信息神经网络(PINNs)因其对偏微分方程(PDE)解的函数表示具有灵活性和准确性,在科学计算中引起了广泛关注。然而,其训练成本限制了其作为经典数值方法真正替代方案的实际应用。因此,我们提出将多分辨率哈希编码引入PINNs以提高训练效率,因为这种编码为神经网络提供了局部感知(多分辨率)的坐标输入。借鉴神经表示场社区(NeRF)的思想,我们研究了计算此类哈希编码神经网络相对于输入坐标的导数的鲁棒性,而这通常是PINN损失项所需要的。我们提出用导数的有限差分计算替代自动微分,以解决此类导数的不连续性问题。我们还分享了哈希编码超参数的合适取值范围,以获得鲁棒的导数。我们在三个问题上测试了所提出的方法,包括Burgers方程、Helmholtz方程和Navier-Stokes方程。与原始PINN实现相比,所提出的方法在效率上实现了约10倍的提升。

英文摘要:

Physics-informed neural networks (PINNs) have attracted a lot of attention in scientific computing as their functional representation of partial differential equation (PDE) solutions offers flexibility and accuracy features. However, their training cost has limited their practical use as a real alternative to classic numerical methods. Thus, we propose to incorporate multi-resolution hash encoding into PINNs to improve the training efficiency, as such encoding offers a locally-aware (at multi resolution) coordinate inputs to the neural network. Borrowed from the neural representation field community (NeRF), we investigate the robustness of calculating the derivatives of such hash encoded neural networks with respect to the input coordinates, which is often needed by the PINN loss terms. We propose to replace the automatic differentiation with finite-difference calculations of the derivatives to address the discontinuous nature of such derivatives. We also share the appropriate ranges for the hash encoding hyperparameters to obtain robust derivatives. We test the proposed method on three problems, including Burgers equation, Helmholtz equation, and Navier-Stokes equation. The proposed method admits about a 10-fold improvement in efficiency over the vanilla PINN implementation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑