Ai2 выпустила HiRO-ACEv2: климатический ИИ в 3 км примерно в 300 раз дешевле X-SHiELD

Ai2 выпустила HiRO-ACEv2: климатический ИИ в 3 км примерно в 300 раз дешевле X-SHiELD

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

Основной упор Ai2 сделала на точное воспроизведение экстремальной погоды. Когда HiROv2 получает огрублённые данные X-SHiELD (глобальной атмосферной модели высокого разрешения, которая напрямую рассчитывает штормы и другие мелкомасштабные явления), она совпадает со статистикой целевой симуляции с точностью до 3%, даже для событий, которые встречаются раз на миллион точек данных. Модель точно оценивает суммарную энергию тропических циклонов, в том числе штормов, которых не видела при обучении, создаёт ураганы с реалистичными глазом и дождевыми полосами и реалистично воспроизводит сильные ветры, стекающие по горным склонам. Оговорка авторов: ветер и давление вокруг глаза урагана остаются несколько более «перекошенными», чем в X-SHiELD.

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

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

Ограничение: поскольку HiRO училась на X-SHiELD, она работает только с версией ACE, настроенной под X-SHiELD, и пока не может выдавать данные высокого разрешения для других климатов, например, для более тёплого будущего или исторической записи вроде набора ERA5. Дальше планируется обучение на новых наборах данных X-SHiELD, включая мир на 4°C теплее, и добавление выходных величин, температуры у поверхности, снега и льда (что открыло бы, например, прогноз снежного покрова). Для наглядности есть веб-демо с предрассчитанными осадками, ветром и давлением на уровне моря от HiROv1 и HiROv2 рядом с X-SHiELD; модели доступны для скачивания, к ним приложены обучающие ноутбуки.

Ключевые факты

  • HiRO-ACEv2 состоит из ACE (нейросеть на 100 км) и HiROv2 (модель, уточняющая результат до 3 км); HiROv2 теперь предсказывает несколько полей сразу, а не только осадки.
  • По оценке Ai2, глобальные данные в 3 км обходятся примерно в 1/300 стоимости X-SHiELD; для одного региона экономия может быть более чем стократно больше.
  • Получив огрублённые данные X-SHiELD, HiROv2 совпадает с её статистикой с точностью до 3%, в том числе для событий раз на миллион точек данных.
  • Система воспроизводит энергию тропических циклонов, но ветер и давление вокруг глаза урагана остаются более «перекошенными», чем в X-SHiELD.
  • Пока работает только с версией ACE, настроенной под X-SHiELD: данные для других климатов, например тёплого будущего или ERA5, получить нельзя.

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

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

Кому это важно

В первую очередь исследователям климата и экстремальной погоды, которым нужно много версий высокоразрешённых данных за 10 лет и более. Авторы отдельно называют региональные расчёты: если нужна только одна область, экономия может быть ещё заметно больше. Дальнейшие планы по снегу и льду (например, для прогноза снежного покрова) расширят круг применений.

Как это применить

Модели доступны для скачивания, а обучающие ноутбуки Ai2 объясняют, как их запускать. Чтобы просто оценить результат без запуска, есть веб-демо: оно показывает предрассчитанные осадки, ветер и давление на уровне моря от HiROv1 и HiROv2 рядом с X-SHiELD и позволяет приблизить свой интересующий регион. Данные пока годятся только для того климата, под который настроена версия ACE, согласованная с X-SHiELD.

Можно ли доверять

Источник, блог самих разработчиков, и оценки качества даны ими же; количественные сравнения вынесены в технический отчёт, на который ссылается текст. Важная оговорка: совпадение «до 3%» измерено относительно целевой симуляции X-SHiELD при подаче ей огрублённых данных X-SHiELD, а не относительно наблюдений реального мира. Стоимость указана только как отношение к X-SHiELD, абсолютных цифр в тексте нет.

Риски и подводные камни

Система обучена на X-SHiELD и работает только с версией ACE, настроенной под неё, поэтому пока не умеет выдавать данные высокого разрешения для других климатов, тёплого будущего или исторической записи вроде ERA5. Ураганы выглядят реалистично, но ветер и давление вокруг глаза остаются более «перекошенными», чем в X-SHiELD. Упрощённая версия HiROv2 «более чем в десять раз быстрее» и обучение на новых данных, включая мир на 4°C теплее, планы без указанных сроков.