Learning Geometric-Aware Quadrature Rules for Functional Minimization
2508.05445v1
math.NA, cs.LG, cs.NA, 65D32, 65N12, 68T07
2025-08-09
Авторы:
Costas Smaragdakis
Резюме на русском
**Резюме**
В статье предлагается метод QuadrANN, основанный на Graph Neural Network (GNN), для адаптивного обучения квадратурных правил на неравномерных точечных облаках. Это решение направлено на улучшение точности численного интегрирования в моделях машинного обучения для решения частичных дифференциальных уравнений (ПДУ) с использованием вариационных принципов. Обучение QuadrANN направлено на извлечение оптимальных весов квадратурных правил из геометрических особенностей точечных облаков. Архитектура модели использует глубокий механизм передачи сообщений, который в первом слое кодирует локальные геометрические признаки, включая положения абсолютные и относительные, а также меру локальной плотности. Затем в последующих слоях включается контекстное видение. Это позволяет QuadrANN генерировать динамические, пермутационно-независимые квадратурные правила, которые адаптируются к локальной плотности точек и форме общего домена. Эксперименты показали, что QuadrANN существенно уменьшает дисперсию при оценке интегралов по сравнению с квази-Монте-Карло методами. Оно достигает этого благодаря тому, что точечные облака подстраиваются под критические места домена, где интегранды имеют сингулярности. Такая улучшенная стабильность в критических областях ключевая для оптимизации энергетических функционалов и повышения эффективности современных нейросетевых вариационных решений.
Abstract
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.