Океан и атмосфера учатся говорить на одном языке ✛

Оригинал: "Coupled Ocean-Atmosphere Dynamics in a Machine Learning Earth System Model"
arXiv:2406.08632 · 2024-06-12 · CC BY 4.0 · 3 мин
Нейросетевая модель Ola впервые воспроизводит Эль-Ниньо, моделируя океан и атмосферу как единую систему.

Контекст

Прогноз на сезон вперёд всегда упирался в океан. Атмосфера забывает своё начальное состояние за пару недель, а вот верхний слой океана хранит память месяцами и медленно отдаёт её через температуру поверхности. Именно поэтому Эль-Ниньо, это тёплое пятно в экваториальной части Тихого океана, служит главным источником предсказуемости на срок от месяца до года. Физические модели уже полвека учатся связывать океан и атмосферу, но до сих пор страдают от систематических ошибок вроде холодного языка в тропиках. Рядом с этой работой лежат попытки предсказывать Эль-Ниньо по данным и строить гибридные модели, но никто до Ola не включал в обучаемый контур подповерхностную температуру океана.

Методы

Ola состоит из двух нейросетей, обученных отдельно, а затем соединённых в цикл. Атмосферная часть с шагом 24 часа обучалась на реанализе ERA5 с 1980 по 2016 год, океаническая с шагом 48 часов — на данных UFS-replay с 1994 по 2016 год. Океанская сеть получает от атмосферы приземный ветер, температуру и давление, а атмосферная — температуру поверхности океана. Внутри океанской сети восемь уровней по глубине от полуметра до 300 метров, что позволяет видеть термоклин, тот самый слой резкого перепада температуры, где прячется память о будущем Эль-Ниньо. Для ансамблевых прогнозов использовали метод лагированных ансамблей: запускали модель каждые шесть часов в первые три дня месяца и собирали 12 траекторий.

Результаты

За пять лет проверки с 2017 по 2022 год Ola воспроизвела и Эль-Ниньо, и Ла-Нинью, причём амплитуда колебаний температуры в области Niño 3.4 оказалась близка к наблюдениям, а холодное смещение, свойственное GFDL-SPEAR, почти отсутствовало. Вдоль экватора модель сгенерировала волны Кельвина, бегущие на восток со скоростью около двух метров в секунду, и волны Россби, ползущие на запад в пяти-десяти градусах от экватора. Под поверхностью аномалии температуры во время Эль-Ниньо смещались к востоку на глубине 50 метров, как и в реальном океане. Единственный заметный промах случился зимой 2021–2022 годов, когда модель не удержала третью Ла-Нинью подряд, но с этим не справилось большинство оперативных систем.

Значение

Ola показала, что машинное обучение может обойти проклятие параметризаций, из-за которого физические модели десятилетиями накапливали ошибки в тропиках. Отличие от соседних работ принципиальное: те либо предсказывали Эль-Ниньо по статистическим индексам, либо использовали только температуру поверхности океана, а Ola впервые включила в обучаемый контур вертикальную структуру верхних 300 метров. Это не просто ещё один эмулятор, а проверка гипотезы, что нейросеть способна выучить связанную динамику океана и атмосферы, включая волны, которые переносят тепло и определяют фазу Эль-Ниньо.

Развитие

Сейчас модель стабильна лишь на шесть месяцев, дальше в высоких широтах накапливаются дрейфы. Увеличение обучающей выборки, добавление океанических течений и более глубоких уровней может продлить прогноз до года и дальше. Гибридные схемы, где нейросеть обучается на тысячелетних симуляциях климатических моделей, а затем дообучается на коротких рядах наблюдений, выглядят самым реалистичным путём к сезонным и межгодовым прогнозам.

Влияние

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

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

Систематически сравнить разные архитектуры связанных моделей на общих данных, чтобы понять, какие эмпирические выборы влияют на долгосрочную стабильность, и обучить версию с более полным вектором состояния океана.

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

Для климатического моделирования остаётся открытым вопрос, как избежать дрейфа средней температуры тропиков при переходе к связанным моделям. Ola даёт первый намёк, что отказ от параметризаций конвекции и пограничного слоя может решить проблему двойной внутритропической зоны конвергенции, которая мучает физические модели с 1980-х годов.

Взгляд машины знаний

Этот раздел адресован автору работы и его коллегам. Соседние работы нашёл смысловой поиск по нашему архиву — не по совпадению слов, а по смыслу. Мы не рецензируем: мы показываем поле вокруг.

Что сделано в работе

Работа представляет модель Ola на архитектуре Spherical Fourier Neural Operator, состоящую из атмосферной и океанической нейросетей, обученных на ERA5 и UFS-replay. Модель воспроизводит Эль-Ниньо и Ла-Нинью, включая подповерхностную структуру температуры, волны Кельвина и Россби, и показывает меньшее холодное смещение по сравнению с GFDL-SPEAR. Для сезонных прогнозов используется метод лагированных ансамблей.

В чём сила

Включение в обучаемый контур вертикальной структуры верхних 300 метров океана позволяет модели улавливать динамику термоклина, что критично для предсказания Эль-Ниньо. Это отличает Ola от предыдущих подходов, которые использовали только температуру поверхности океана или статистические индексы.

Куда можно двигаться

Что можно попробовать

  • Проверить, улучшает ли включение данных о подповерхностной температуре прогноз экстремальных осадков в тропиках.
  • Использовать Ola для инициализации региональных моделей с высоким разрешением, чтобы улучшить локальные прогнозы.
  • Оценить, как модель воспроизводит дальние связи (телеконнекции), например, влияние Эль-Ниньо на погоду в средних широтах.

Почему это может быть важно

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

Прогноз на шесть месяцев Ola делает меньше чем за минуту на одной видеокарте A100, тогда как классической модели GFDL-SPEAR для того же расчёта нужны часы на кластере.

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

  • разрешение: 0.25° по широте и долготе
  • глубина океана: до 300 метров, 8 уровней
  • скорость прогноза: 6 месяцев менее чем за 1 минуту на A100
  • период обучения атмосферы: 1980–2016 (ERA5)
  • период обучения океана: 1994–2016 (UFS-replay)
Учёные
Ludwig BoltzmannHenri PoincaréJohn von NeumannJosiah Willard Gibbs
Понятия
связь океана и атмосферы модель Зебиака–Кейна энсо NOAA CPC североатлантическая осцилляция климатическая модель буйковые станции TAO/TRITON южная осцилляция эль-ниньо многомерный индекс ЭНСО весенний барьер предсказуемости относительный они диполь индийского океана ла-нинья ERA5 индекс Эль-Ниньо климатическая сеть ансамблевый прогноз температура поверхности моря прогноз ERSST