用图神经网络自动学习点云积分权重,提升PDE求解稳定性。
Learning Geometric-Aware Quadrature Rules for Functional Minimization
- 设计GNN模型QuadrANN,从点云几何学直接学习积分权重。
- 相比标准方法,积分方差降低,在奇异区域更稳定。
- 适合需要高精度能量泛函优化的mesh-free PDE求解场景。
在基于变分原理的无网格机器学习求解偏微分方程(PDE)中,非均匀点云上的精确数值积分仍具挑战。传统蒙特卡洛方法无法处理非均匀点云,而现代神经网络可处理置换不变输入,从而为任意点云构建积分规则。本文提出QuadrANN,一种图神经网络(GNN)架构,直接从点云几何结构学习最优积分权重。模型首层编码局部几何特征,包括绝对与相对位置及显式密度度量;后续层引入全局上下文向量。该设计使QuadrANN生成的积分规则具备置换不变性,并能自适应局部点密度与整体域形状。我们在一系列挑战性测试中验证方法,涵盖凸与非凸域上的积分,以及热方程和福克-普朗克方程的解估计。所有测试中,QuadrANN通过将点云映射至关键区域更密集,显著降低积分估计方差,优于标准拟蒙特卡洛方法。这种在域内关键区域的增强稳定性对能量泛函优化至关重要,从而提升了基于深度学习的变分求解器性能。
原文摘要 · Abstract (English)
Accurate numerical integration over non-uniform point clouds is a challenge for modern mesh-free machine learning solvers for partial differential equations (PDEs) using variational principles. While standard Monte Carlo (MC) methods are not capable of handling a non-uniform point cloud, modern neural network architectures can deal with permutation-invariant inputs, creating quadrature rules for any point cloud. In this work, we introduce QuadrANN, a Graph Neural Network (GNN) architecture designed to learn optimal quadrature weights directly from the underlying geometry of point clouds. The design of the model exploits a deep message-passing scheme where the initial layer encodes rich local geometric features from absolute and relative positions as well as an explicit local density measure. In contrast, the following layers incorporate a global context vector. These architectural choices allow the QuadrANN to generate a data-driven quadrature rule that is permutation-invariant and adaptive to both local point density and the overall domain shape. We test our methodology on a series of challenging test cases, including integration on convex and non-convex domains and estimating the solution of the Heat and Fokker-Planck equations. Across all the tests, QuadrANN reduces the variance of the integral estimation compared to standard Quasi-Monte Carlo methods by warping the point clouds to be more dense in critical areas where the integrands present certain singularities. This enhanced stability in critical areas of the domain at hand is critical for the optimization of energy functionals, leading to improved deep learning-based variational solvers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。