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.
Ссылки и действия
Дополнительные ресурсы: