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

## Контекст Исследования в области машинного обучения (ML) применяются в системах симуляции транзиентных численных задач, основанных на численных решениях транзиентных параболических и эллиптических частных дифференциальных уравнений (PDE). Однако существуют ограничения, включая необходимость выполнения классических симуляций для обучения моделей и высокие требования к вычислительным ресурсам. Из-за этого, методы ML пока не смогли доказать значительное преимущество перед существующими классическими методами, такими как системы Жордана-Галеркина. Мы рассмотрим альтернативный подход, где ML-модель взаимодействует с классическим решателем, обучаясь на его внутренней информации. Это позволяет добиться однократной ускорения решения PDE. Наша работа сфокусирована на случае линеаризации PDE при помощи предиктора-корректора PCG. ## Метод Мы предлагаем PCGBandit — новый подход, который использует онлайн-обучение для построения адаптивных предикторов-корректоров в процессе решения PDE. На каждом шаге решения системы линейных уравнений, вызываемой PCG, мы собираем входные данные (в том числе историю решений) и используем их для настройки конфигурации системы. Бандитное решение используется для распределения ресурсов между различными предикторами-корректорами в ходе решения. Метод PCGBandit может быть интегрирован в существующий классический код OpenFOAM. Мы проводим эксперименты на наборе тестовых задач в области жидкостей и магнитогидродинамики (MHD). ## Результаты Мы сравнили PCGBandit с классическими методами на наборе тестовых задач в области жидкостей и MHD. Результаты показали, что PCGBandit значительно ускоряет решение PDE в первых нескольких итерациях, без дополнительных вычислительных затрат в процессе. Мы также провели эксперименты для доказательства обоснованности нашего подхода, включая измерение затрат вычислений и оценку точности решения. Наши результаты показывают, что PCGBandit эффективно обучается в реальном времени, достигая быстрых улучшений в производительности решения PDE. ## Значимость Мы видим применение PCGBandit в системах симуляции, где требуется быстрое решение PDE. Например, в проблемах распространения тепла, гидродинамики или магнитогидродинамики, где каждый процент ускорения решения может привести к значительным экономиям во времени и ресурсах. Наш подход может быть использован для улучшения существующих систем, таких как OpenFOAM, не требуя полного перестроения их архитектуры. Кроме того, это может привести к новым возможностям в области разработки эффективных алгоритмов для решения PDE в сценариях с ограниченными вычислительными ресурса