Два танцора стоят друг напротив друга, но не видят лиц. Один — атмосфера, быстрая и шумная, забывающая свои движения уже через пару недель. Второй — океан, медленный, тяжёлый, помнящий каждое касание месяцами. Чтобы танец состоялся, им нужно научиться вести друг друга, глядя только на руки партнёра. Именно такую пару создали исследователи, собрав модель Ola: две нейросети, которые обмениваются лишь несколькими числами, но вместе воспроизводят один из самых капризных ритмов планеты — Эль-Ниньо.
Секрет танца спрятан под водой. Верхний слой океана, до 300 метров, хранит память о прошлых ветрах и отдаёт её через температуру поверхности. Атмосфера чувствует эту температуру и меняет ветер, который снова толкает океан. В физических моделях эту связь описывают уравнения, но они десятилетиями накапливают ошибки в тропиках — например, холодный язык, который портит прогнозы. Ola пошла другим путём: нейросеть для атмосферы обучалась на 36 годах реанализа ERA5, нейросеть для океана — на 22 годах данных UFS-replay, а потом их соединили в цикл. Океанская сеть видит восемь уровней глубины, от полуметра до 300 метров, и потому различает термоклин — слой резкого перепада температуры, где прячется будущее Эль-Ниньо.
Проверка длилась пять лет, с 2017 по 2022. Ola воспроизвела и Эль-Ниньо, и его холодную сестру Ла-Нинью, причём амплитуда колебаний температуры в области Niño 3.4 оказалась близка к наблюдениям. Вдоль экватора модель сгенерировала волны Кельвина, бегущие на восток со скоростью около двух метров в секунду, и волны Россби, ползущие на запад в пяти-десяти градусах от экватора. Под поверхностью аномалии температуры во время Эль-Ниньо смещались к востоку на глубине 50 метров, как и в реальном океане. Единственный заметный промах случился зимой 2021–2022 годов, когда модель не удержала третью Ла-Нинью подряд, но с этим не справилось большинство оперативных систем.
Раньше было неясно, сможет ли нейросеть выучить связанную динамику океана и атмосферы, включая подповерхностные процессы, которые управляют Эль-Ниньо. Ola показала: да, сможет. Отличие от соседних работ принципиальное: те либо предсказывали Эль-Ниньо по статистическим индексам, либо использовали только температуру поверхности океана, а Ola впервые включила в обучаемый контур вертикальную структуру верхних 300 метров. Это не просто ещё один эмулятор, а проверка гипотезы, что нейросеть способна выучить связанную динамику океана и атмосферы, включая волны, которые переносят тепло и определяют фазу Эль-Ниньо.
Сейчас модель стабильна лишь на шесть месяцев, дальше в высоких широтах накапливаются дрейфы. Но уже сегодня прототип можно использовать для быстрых ансамблевых сезонных прогнозов в исследовательских целях: прогноз на шесть месяцев Ola делает меньше чем за минуту на одной видеокарте A100, тогда как классической модели GFDL-SPEAR для того же расчёта нужны часы на кластере. Два танцора, которые учились только по рукам, уже вышли на сцену.
Прогноз на шесть месяцев Ola делает меньше чем за минуту на одной видеокарте A100, тогда как классической модели GFDL-SPEAR для того же расчёта нужны часы на кластере.