PCGBandit: One-shot acceleration of transient PDE solvers via online-learned preconditioners

## Контекст Исследования в области машинного обучения (ML) для ускорения вычислительных методов в научных задачах набирают обороты. Одним из основных направлений является ускорение моделирования транзиентных численных задач, основанных на частных дифференциальных уравнениях (PDE). Данные задачи характеризуются сложной структурой и необходимостью решения многократно связанных линейных систем. Проблема заключается в том, что традиционные методы требуют больших объемов данных и высокой вычислительной сложности, что ставит перед методами машинного обучения высокие требования. Мы рассматриваем альтернативный подход, включающий использование данных, полученных из самого классического решения, для однократного ускорения симуляции. ## Метод Метод, разработанный в рамках данного исследования, называется **PCGBandit**. Он основывается на применении интеллектуальных алгоритмов построения адаптивных конфигураций для линейного стохастического алгоритма решения транзиентных PDE. Алгоритм использует обучаемый подход на основе методов онлайн-обучения, используя данные, полученные при работе классического решения, такого как **preconditioned conjugate gradient (PCG)**. Алгоритм оптимизирует последовательность конфигураций (например, preconditioners), которые используются для ускорения решения линейных систем. Эта архитектура включает в себя интеграцию с популярным open-source фреймворком OpenFOAM для моделирования транзиентных PDE. ## Результаты Эксперименты проводились на наборе тестовых задач, включающих моделирование газовой динамики и магнитогидродинамики (MHD). Использовались реальные данные, полученные при моделировании транзиентных PDE в OpenFOAM. Результаты показали, что PCGBandit существенно ускоряет моделирование, сохраняя точность и уменьшая вычислительные затраты по сравнению с базовым классическим подходом. Особенно эффективен алгоритм при работе с сложными сценариями, где требуется быстрое реагирование на изменения условий задачи. ## Значимость Метод PCGBandit может быть применен в различных областях, где используется транзиентное моделирование PDE, таких как инженерное моделирование, аэродинамика, электромагнетизм и моделирование теплопроводности. Он обеспечивает однократное ускорение решения линейных систем, что может привести к сокращению времени вычислений на долгих симуляциях. Также, этот подход является гибким и может быть интегрирован с другими методами машинного обучения для улучшения производительности. ## Выводы Разработанный подход PCGBandit продемонстрировал способность однократно ускорять решение транзиентных PDE без снижения точности. Он может быть использован в реальных задачах, где важно эффекти