Цифровой двойник атмосферы: как нейросеть ускорила климат в 350 раз ✛

Оригинал: "CAMulator: Fast Emulation of the Community Atmosphere Model"
arXiv:2504.06007 · 2025-04-08 · CC BY 4.0 · 2 мин
Нейросеть CAMulator повторяет физику атмосферы CAM6, сохраняя массу, влагу и энергию, и работает в 350 раз быстрее.

Переводчик на важных переговорах не просто быстро переводит слова, но следит, чтобы смысл не потерялся. Так и CAMulator: он берёт сложную климатическую модель CAM6 и переводит её на язык нейросети, которая работает в сотни раз быстрее, при этом не теряя физической сути. Раньше такие «переводчики» либо упрощали картинку, либо забывали про законы сохранения, и их прогнозы рассыпались при долгом счёте. CAMulator стал первым, кто держит и полное разрешение, и строгие правила физики.

Скорость поражает: 480 лет климата за один день на одной видеокарте. Обычная модель CAM6 на суперкомпьютере выдаёт лишь 14 лет за тот же срок на десяти процессорных узлах.

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

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

Несмотря на это, CAMulator ловит главные климатические «жесты»: Эль-Ниньо, Североатлантическое колебание и тихоокеанско-североамериканскую моду. Корреляция с оригиналом достигает 0.935 для осадков во время Эль-Ниньо.

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

CAMulator обучался всего три дня, но способен симулировать 480 лет климата за один день на одном графическом процессоре. Для сравнения, CAM6 на суперкомпьютере Derecho выдаёт 14 лет в день на 10 процессорных узлах.

Понятия
климатическая модель ERA5 связанная модель общей циркуляции связь океана и атмосферы реанализ NOAA CPC OISST v2.1 ансамблевый прогноз температура поверхности моря атлантическая мультидекадная осцилляция IRI климат морская волна тепла компьютерное моделирование ла-нинья глобальное потепление объём тёплой воды сезонный цикл энсо аналоговый прогноз