arXiv:2411.06286cs.LGcs.NA2024-11被引 24

用分离变量法加速高维偏微分方程求解的新型神经网络。

SPIKANs: Separable Physics-Informed Kolmogorov-Arnold Networks

  • 将变量分离思想融入物理信息KAN,按维度分治处理。
  • 在高维问题上训练速度提升显著,精度保持不变。
  • 适合需要高效求解高维偏微分方程的研究者。

物理信息神经网络(PINNs)是科学计算中求解偏微分方程(PDEs)的有前景方法。传统PINNs使用多层感知机(MLPs),而近年来的Kolmogorov-Arnold网络(KAN)展现出更快的神经网络缩放和更好的可解释性。将KAN应用于物理信息学习催生了物理信息KAN(PIKANs),可用于求解PDEs。然而,KAN在高维问题中常因采样点数量随维度指数增长而导致训练缓慢。为此,本文提出可分离的物理信息Kolmogorov-Arnold网络(SPIKANs),通过分离变量原理,使每个维度由独立的KAN处理,大幅降低训练复杂度,同时保持精度。在多个基准测试中,SPIKANs展现出优于PIKANs的可扩展性和性能,具备解决复杂高维PDEs的潜力。

原文摘要 · Abstract (English)

Physics-Informed Neural Networks (PINNs) have emerged as a promising method for solving partial differential equations (PDEs) in scientific computing. While PINNs typically use multilayer perceptrons (MLPs) as their underlying architecture, recent advancements have explored alternative neural network structures. One such innovation is the Kolmogorov-Arnold Network (KAN), which has demonstrated benefits over traditional MLPs, including faster neural scaling and better interpretability. The application of KANs to physics-informed learning has led to the development of Physics-Informed KANs (PIKANs), enabling the use of KANs to solve PDEs. However, despite their advantages, KANs often suffer from slower training speeds, particularly in higher-dimensional problems where the number of collocation points grows exponentially with the dimensionality of the system. To address this challenge, we introduce Separable Physics-Informed Kolmogorov-Arnold Networks (SPIKANs). This novel architecture applies the principle of separation of variables to PIKANs, decomposing the problem such that each dimension is handled by an individual KAN. This approach drastically reduces the computational complexity of training without sacrificing accuracy, facilitating their application to higher-dimensional PDEs. Through a series of benchmark problems, we demonstrate the effectiveness of SPIKANs, showcasing their superior scalability and performance compared to PIKANs and highlighting their potential for solving complex, high-dimensional PDEs in scientific computing.

偏微分方程神经网络高维求解物理信息

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。