Знание физики укрощает хаос: прогноз экстремальных событий на малых данных

Оригинал: "Knowledge-based Deep Learning for Modeling Chaotic Systems"
arXiv:2209.04259 · 2022-09-09 · CC BY 4.0 · 3 мин
Гибридная нейросеть, обученная на синтетике и физических законах, точнее предсказывает экстремальные явления, когда данных мало.

Контекст

Прогнозирование хаотических систем, таких как погода, эпидемии или океанские течения, остаётся вызовом: классические физические модели требуют точных начальных условий и быстро расходятся с реальностью, а чисто data-driven подходы нуждаются в огромных выборках. Экстремальные события усугубляют проблему: они редки, поэтому данных о них мало, а их динамика нелинейна и чувствительна к шуму. Работа стоит на стыке физически-информированного машинного обучения и переноса знаний: соседние исследования уже показали, что внедрение законов сохранения в нейросети помогает моделировать температуру озёр или потоки в чашке эспрессо, но для хаотических систем с производными высокого порядка такие методы не работали.

Методы

Авторы строят архитектуру KDL на основе LSTM (долгой краткосрочной памяти) с тремя слоями по 50 нейронов. Сначала генерируют синтетический временной ряд из уравнения Льенара с внешним периодическим воздействием (параметры: f=0.2, α=0.45, γ=-0.5, ω=0.642, β=0.5) методом Рунге-Кутты. Для каждого момента времени вычисляют дискретные производные первого и второго порядка как разности соседних значений. Затем обучают сеть предсказывать сразу три величины: x(t), dx/dt и d²x/dt², минимизируя сумму ошибки данных и физической невязки, которая требует, чтобы предсказанные значения удовлетворяли уравнению Льенара. После предобучения веса сети переносятся на реальные данные (Эль-Ниньо, лихорадка денге, осадки на острове Бьёрнёйа), где физическая регуляризация сравнивает левую часть уравнения для реальных и предсказанных рядов.

Результаты

На данных Эль-Ниньо KDL показал наименьшую RMSE (0.428 против 0.433 у LSTM при 20% обучающих данных) и наименьшую физическую невязку (276.654 против 283.340 у LSTM). При увеличении доли обучения до 80% преимущество сохраняется, но сокращается. Для лихорадки денге перенос знаний с синтетики ухудшил результат, поэтому авторы отключили предобучение и оставили только физическую регуляризацию: KDL получил лучшую физическую невязку при 40% и 60% данных, хотя по RMSE уступил LSTM. На осадках Бьёрнёйа (15320 наблюдений) KDL выиграл по RMSE при 20%, 60% и 80% обучающих данных, но по физической невязке уступил резервуарным вычислениям (ESN). В среднем по всем наборам и разбиениям KDL занял первое место по RMSE (ранг 1.12) и MAE (ранг 1.29), обойдя LSTM, FFNN, CNN и ESN.

Значение

Работа показывает, что физическая регуляризация и перенос знаний с синтетических данных позволяют глубоким сетям прогнозировать хаотические экстремальные события даже при малом числе наблюдений. В отличие от соседних работ, где физика вшивалась через жёсткие ограничения или использовалась для эмуляции PDE, здесь производные вычисляются дискретно и включаются в функцию потерь LSTM, что делает метод применимым к реальным временным рядам без точных граничных условий. Это расширяет арсенал физически-информированных нейросетей на задачи с редкими и нерегулярными данными, где классические PINN не работают.

Развитие

Авторы планируют распространить KDL на многошаговое прогнозирование, что важно для практических предупреждений об экстремальных событиях. Возможна адаптация к другим типам нелинейных осцилляторов и включение более сложных физических законов, например, уравнений Навье-Стокса для океанических течений. Перспективно также исследовать, когда перенос знаний с синтетики помогает, а когда вредит (как в случае с денге), чтобы строить адаптивные стратегии обучения.

Влияние

Метод может улучшить раннее предупреждение о климатических аномалиях (Эль-Ниньо), вспышках инфекций и экстремальных осадках в регионах с ограниченными историческими данными. В инженерии он применим для прогноза редких отказов в хаотических системах, например, в энергосетях.

Следующие шаги

Проверить KDL на других наборах данных с экстремальными событиями, например, на финансовых крахах или сейсмической активности. Исследовать автоматический выбор коэффициента регуляризации λ и порога, когда перенос знаний с синтетики следует отключать.

Ключевые проблемы

Работа напрямую связана с давней проблемой прогнозирования хаотических систем при ограниченных данных, где классические методы либо требуют точной модели, либо большого объёма наблюдений. Она также затрагивает вопрос физической согласованности прогнозов, что важно для доверия к моделям в геофизике и эпидемиологии. Открытым остаётся вопрос о том, насколько универсально уравнение Льенара описывает реальные экстремальные события и как выбирать подходящую физическую модель для конкретного набора данных.

Уравнение Льенара, использованное для генерации синтетических данных, было введено французским физиком Альфредом Льенаром в 1928 году для описания колебательных контуров и с тех пор стало классикой теории нелинейных осцилляторов. Интересно, что экстремальные события в таких системах возникают как редкие, но повторяющиеся всплески амплитуды, похожие на гигантские волны-убийцы в океане.

Идея предсказывать хаос с помощью гибридного знания перекликается с романом Айзека Азимова «Основание», где психоистория объединяет статистику и детерминированные законы для прогноза будущего больших систем. В отличие от Азимова, здесь физика вшита в нейросеть, а не в уравнения для масс, но дух гибридного подхода схож.

\frac{d^2x}{dt^2} + \alpha x \frac{dx}{dt} + \gamma x + \beta x^3 = f \sin(\omega t)
Второе уравнение системы, где α, γ, β — параметры нелинейности и затухания, f и ω — амплитуда и частота внешней силы
L_{phy} = RMSE\left(\frac{d^2y_{pred}}{dt^2} + \alpha y_{pred}\frac{dy_{pred}}{dt} + \gamma y_{pred} + \beta y_{pred}^3, f\sin(\omega t)\right)
Среднеквадратичная ошибка между левой частью уравнения Льенара для предсказанных значений и внешним воздействием
L_{phy} = RMSE\left(\frac{d^2y_{real}}{dt^2} + \alpha y_{real}\frac{dy_{real}}{dt} + \gamma y_{real} + \beta y_{real}^3, \frac{d^2y_{pred}}{dt^2} + \alpha y_{pred}\frac{dy_{pred}}{dt} + \gamma y_{pred} + \beta y_{pred}^3\right)
Сравнение левых частей уравнения для реальных и предсказанных рядов, чтобы прогноз удовлетворял той же динамике

Ключевые числа

  • RMSE Эль-Ниньо при 20% данных: 0.428 (KDL) против 0.433 (LSTM)
  • Физическая невязка Эль-Ниньо при 20% данных: 276.654 (KDL) против 283.340 (LSTM)
  • Средний ранг KDL по RMSE: 1.12 (первое место среди 5 моделей)
  • Число наблюдений в данных осадков Бьёрнёйа: 15320 (против 1634 у Эль-Ниньо)
  • Параметр нелинейности β в уравнении Льенара: 0.5
Понятия
моделирование физики с помощью нейронных сетей нейронные сети прогностического обучения большие языковые модели теория хаоса динамические системы глубокое обучение в медицинской визуализации вычислительная гидродинамика динамика сложных систем стохастическая динамическая система сложные системы квантовый хаос обучающие данные математическая модель обучение навык прогноза языковая модель белков стохастическое моделирование обобщение вероятностный прогноз вычислительный анализ реакций