Океан как паутина: нейросеть предсказывает Эль-Ниньо

Оригинал: "The World as a Graph: Improving El Niño Forecasts with Graph Neural Networks"
arXiv:2104.05089 · 2021-04-11 · CC BY 4.0 · 2 мин
Нейросеть сама нашла невидимые связи в океане и научилась предсказывать тёплое течение на полгода вперёд.

Океан похож на гигантскую паутину. Если тронуть одну нить, задрожит далёкий угол. Нейросеть из этой работы научилась плести такую паутину, чтобы предсказывать Эль-Ниньо.

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

Новая модель представила океан как граф: 1345 точек на карте, соединённых невидимыми линиями. Каждая точка хранила температуру воды за три месяца. Связи не задавались заранее — модель сама решала, какие точки должны обмениваться информацией, и оставляла только восемь самых сильных на каждую точку. Затем по этим связям распространялись сигналы, как слухи по сети знакомств.

Граф — это просто набор точек и линий между ними. Здесь точки — участки океана, а линии — связи, по которым тепло передаётся на расстояния.

Результат оказался лучше прежних подходов. Для прогноза на шесть месяцев корреляция с наблюдениями составила 0.780, тогда как у свёрточной нейросети — 0.762. Для сильного Эль-Ниньо 2015 года модель дала корреляцию 0.948 по году. Самое важное: выученная паутина совпала с тем, что физики уже знали об океане. На коротких сроках ключевые точки находятся там, где измеряют температуру. К шести месяцам влияние уходит на запад Тихого океана, а к 23 месяцам — в Индийский и Атлантический океаны.

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

Модель потребовала в 12 раз меньше нейросетей, чем предыдущий рекорд: одна сеть на горизонт вместо двенадцати для каждого сезона. При этом обучение занимало всего 33–50 минут на четырёх GPU.

В романе Кима Стэнли Робинсона «Сорок знаков дождя» климатологи бьются над предсказанием внезапных климатических сдвигов. Graphiño — шаг к тому, чтобы такие прогнозы стали рутиной, а не драмой.

Понятия
графовая нейронная сеть прогноз нейронные сети эль-ниньо весенний барьер предсказуемости энсо графы индекс Эль-Ниньо ла-нинья климатическая сеть ERSST большие языковые модели модель Джина–Тиммермана NOAA PMEL TAO распространение экваториальных волн NOAA CPC многомерный индекс ЭНСО IRI обобщение генеративно-состязательные сети