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.

Ссылки и действия