Scalable h-adaptive probabilistic solver for time-independent and time-dependent systems
2508.09623v2
stat.ML, cs.LG
2025-08-18
Авторы:
Akshay Thakur, Sawan Kumar, Matthew Zahr, Souvik Chakraborty
Резюме на русском
## Контекст
Решение частных дифференциальных уравнений (PDEs) широко используется в различных областях науки и техники, от физики и химии до инженерии и биологии. Однако, действительно точное решение этих уравнений может быть чрезвычайно сложным, особенно при работе с высокой размерностью или большим количеством данных. Одним из основных исходных мотивов для этого исследования является необходимость создания системы, которая может эффективно решать PDEs в сложных условиях с минимальными ресурсами. Более того, многие методы требуют значительного количества вычислений, что может привести к высоким затратам на вычислительные ресурсы, особенно при работе с большими данными или высокой точностью. Таким образом, целью этой работы является разработка метода, который может решать PDEs быстро и точно, при этом учитывая параметры проблемы и масштабируемость.
## Метод
Предложенный метод основывается на применении проблемы в пространстве коллокации с использованием Gaussian Process Regression (GPR). Это позволяет получать решения для PDEs без необходимости использовать меш (grid-based), что уменьшает требования к памяти и увеличивает скорость вычислений. Однако, основная проблема в этой методике заключается в высокой стоимости вычислений при работе с большим количеством коллокационных точек. Для решения этой проблемы, авторы предлагают два ключевых усовершенствования. Во-первых, разработана стратегия стохастического двустороннего дезонтации (stochastic dual descent), которая уменьшает вычислительные затраты для каждой итерации, переходя от кубической к линейной стоимости в зависимости от количества точек коллокационных точек. Во-вторых, введена адаптивная стратегия кластеризации, которая позволяет выбирать только те точки коллокации, которые дадут максимальную информацию, с минимальным вычислительным вкладом. Эти инновации позволяют создать новый метод, который может эффективно решать PDEs на больших масштабах.
## Результаты
Для оценки эффективности нового способа были использованы различные типы PDEs, включая одномерные и многомерные дифференциальные уравнения, а также проблемы временной зависимости (time-dependent). Эксперименты показали, что новый метод значительно эффективнее существующих методов, особенно при работе с большим количеством точек. Было показано, что эксперименты на простых 2D и 3D PDEs дали результаты, совпадающие с существующими методами, но с режекцией значительно меньших вычислительных затрат. В частности, на проблемах временной зависимости, новый метод показал существенно более высокую скорость вычислений и лучшую точность в сравнении с традиционными подходами.
## Значимост
Abstract
Solving partial differential equations (PDEs) within the framework of
probabilistic numerics offers a principled approach to quantifying epistemic
uncertainty arising from discretization. By leveraging Gaussian process
regression and imposing the governing PDE as a constraint at a finite set of
collocation points, probabilistic numerics delivers mesh-free solutions at
arbitrary locations. However, the high computational cost, which scales
cubically with the number of collocation points, remains a critical bottleneck,
particularly for large-scale or high-dimensional problems. We propose a
scalable enhancement to this paradigm through two key innovations. First, we
develop a stochastic dual descent algorithm that reduces the per-iteration
complexity from cubic to linear in the number of collocation points, enabling
tractable inference. Second, we exploit a clustering-based active learning
strategy that adaptively selects collocation points to maximize information
gain while minimizing computational expense. Together, these contributions
result in an $h$-adaptive probabilistic solver that can scale to a large number
of collocation points. We demonstrate the efficacy of the proposed solver on
benchmark PDEs, including two- and three-dimensional steady-state elliptic
problems, as well as a time-dependent parabolic PDE formulated in a space-time
setting.
Ссылки и действия
Дополнительные ресурсы: