Scalable h-adaptive probabilistic solver for time-independent and time-dependent systems

2508.09623v1 stat.ML, cs.LG 2025-08-15
Авторы:

Akshay Thakur, Sawan Kumar, Matthew Zahr, Souvik Chakraborty

Резюме на русском

#### Контекст Решение частных дифференциальных уравнений (PDE) широко распространено в науке и инженерии, так как они лежат в основе моделей многих физических и технологических процессов. Однако решение PDE часто связано с высокими вычислительными затратами, особенно при работе с высокомерностными проблемами или большими масштабах. Эпистемическая неопределенность, возникающая из-за дискретизации, также представляет собой значительную сложность. Базируясь на подходе проблематической нумерики, в рамках которого решения PDE становятся вероятностными, можно уменьшить эту неопределенность и улучшить точность решения. Однако существующие методы сталкиваются с проблемами масштабируемости, потому что их вычислительный оверхед растет требовательно к количеству точек для решения. #### Метод Мы предлагаем усовершенствования в проблематической нумерике, которые уменьшают вычислительный оверхед и позволяют масштабировать решение PDE. В качестве первого усовершенствования мы предлагаем алгоритм стохастического дуального спуска, который снижает требования к вычислительной сложности от кубических до линейных относительно количества точек для решения. Второй вклад — кластерная стратегия активного обучения, которая адаптивно выбирает точки для решения, чтобы максимизировать информационный выигрыш при минимальной вычислительной затрате. Эти техники включены в $h$-адаптивный подход, где адаптивность достигается через учет локальных характеристик решения и применение эвристик, уменьшающих вычислительный оверхед. #### Результаты Мы проверили алгоритм на нескольких классических PDE, включая дву- и трехмерные уравнения Эллипса, а также временно-зависимое уравнение Параболы. Наши результаты показали, что $h$-адаптивный подход эффективно уменьшает неопределенность и позволяет получить точные решения с меньшими вычислительными затратами по сравнению с традиционными методами. Мы также показали, что стратегия активного обучения приводит к повышению точности и позволяет эффективно управлять ресурсами. #### Значимость Предлагаемый подход имеет широкие применения в областях, где требуется решение высокомерностных PDE, таких как моделирование теплообмена, химических процессов и динамики жидкостей. Он позволяет решать проблемы, которым ранее трудно было подойти из-за высоких требований к вычислительным ресурсам. Одним из ключевых преимуществ является уменьшение вычислительного оверхеда, что позволяет эффективно использовать ресурсы и увеличивает масштабируемость систем. Это делает исследование влияния PDE на реальные процессы более доступным и практичны

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.

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