«Экстремальные события» по цене в 300 раз ниже: Ai2 выпустила эмулятор погоды HiRO-ACEv2 с сеткой 3 км

Две модели вместо одной: как устроена система

HiRO-ACEv2 — новейшее обновление системы High Resolution Output Ai2 Climate Emulator (HiRO-ACE), которую лаборатория Ai2 представила ранее в 2026 году. Система состоит из двух частей: ACE — ИИ-модель, работающая с грубым разрешением 100 км, и HiRO — модель, которая достраивает эти грубые результаты до детализации 3 км.

Смысл конструкции в экономии: вместо того чтобы физически считать атмосферу на сетке 3 км, система сначала строит дешёвую картину на 100 км, а затем «заостряет» её отдельной нейросетью. Именно вторая часть и переписана в версии v2: главное изменение — новая модель даунскейлинга HiROv2, которая теперь предсказывает несколько полей высокого разрешения сразу вместо одних осадков, то есть способна покрывать гораздо более широкий набор экстремальных погодных явлений, и лучше справляется с редкими событиями.

Эталоном в работе выступает X-SHiELD — высокоразрешающая глобальная атмосферная модель, которая явно моделирует штормы и другие мелкомасштабные погодные процессы. Сравнение с ней и служит в статье мерой качества.

Что удалось воспроизвести из редких явлений

Авторы отдельно подчёркивают: основной фокус был именно на экстремальных событиях, и изменения в обучении HiROv2 улучшили её способность воспроизводить самые интенсивные значения дождя, ветра и давления. Получив огрублённые данные X-SHiELD, модель совпадает со статистикой целевой симуляции высокого разрешения в пределах 3 % — и это касается даже событий, которые встречаются лишь один раз на миллион точек данных.

Кроме распределений, HiROv2 точно оценивает полную энергию тропических циклонов, включая штормы, которых не было в обучающих данных, и порождает ураганы с реалистичным глазом и дождевыми полосами — правда, ветер и давление вокруг глаза остаются у неё несколько более асимметричными, чем в X-SHiELD. Также модель реалистично воспроизводит сильные ветры, стекающие по горным склонам.

Для системы обновили и саму ACE. За десятилетнюю симуляцию её ураганы, средние климатические картины и общие распределения осадков, ветра и давления хорошо совпадают с X-SHiELD — это, по словам авторов, говорит о том, что грубые симуляции ACE служат надёжной основой для получения долговременных данных высокого разрешения.

Год глобальных данных и порядок цен

Собрав HiROv2 и ACE вместе, авторы сгенерировали полный год глобальных данных с сеткой 3 км. Их статистика близко совпала с X-SHiELD, хотя HiRO работала не с настоящими данными X-SHiELD, а с симуляцией, которую ACE провела самостоятельно. На десятилетнем прогоне энергия ураганов HiRO-ACEv2 оказалась сопоставимой с X-SHiELD за тот же период, а отдельные срезы показывали и другие экстремальные события, похожие на те, что даёт эталон.

Экономика здесь и есть главный аргумент: производство глобальных данных высокого разрешения обходится примерно в 300 раз дешевле, чем запуск X-SHiELD. Если нужен только один регион, экономия может быть более чем в сто раз больше — в зависимости от площади области. Авторы также работают над упрощённой версией HiROv2, которая может оказаться ещё более чем в десять раз быстрее текущей.

Чего система пока не умеет

Ограничение названо открыто: поскольку HiRO училась на X-SHiELD, она работает только с той версией ACE, которая настроена под X-SHiELD. Значит, получать данные высокого разрешения для других климатов система пока не может — ни для более тёплого будущего, ни для исторических записей вроде набора ERA5.

Планы на будущее — обучение на новых наборах X-SHiELD, включая симуляцию мира теплее на 4 °C, чтобы система умела работать с разными климатами, и добавление новых выходов: приземной температуры, снега и льда. Последнее открыло бы дополнительные применения, например предсказание снежного покрова.

Как посмотреть результат

Чтобы проверить, что именно даёт HiRO, без запуска инференса авторы собрали веб-демонстрацию: она показывает заранее сгенерированные осадки, ветер и давление на уровне моря от HiROv1 и HiROv2 рядом с X-SHiELD, на котором модели учились. Демонстрация дополняет количественные сравнения в техническом отчёте и позволяет приблизиться к своему региону и оценить качество там.

Для более глубокого разбора модели выложены для загрузки, а учебные ноутбуки описывают, как их запускать; карточка модели опубликована на Hugging Face как allenai/HiRO-ACEv2.