Nvidia расширяет набор Earth-2 тремя новыми моделями с открытым исходным кодом для расширенных симуляций погоды и климата

Кратко

Nvidia добавила три новых модели с открытым исходным кодом в свою линейку Earth-2, предоставляя инструменты для прогнозирования погоды, данных и климатического моделирования.

Технологическая компания NVIDIA объявила о трех новых моделях с открытым исходным кодом в рамках семейства Earth-2, предназначенных для упрощения разработки приложений для прогнозирования погоды на всем метеорологическом рабочем процессе, включая данные, прогнозирование, текущие прогнозы, снижение масштаба и связанные задачи.

Разработчики могут использовать программное обеспечение с открытым исходным кодом NVIDIA, такое как Earth2Studio для построения конвейеров вывода и Physics Nemo для обучения моделей, чтобы быстро создавать модели погоды и климата.

Линейка Earth-2 объединяет ряд ускоренных инструментов и моделей, объединяя возможности, которые ранее располагались по разным частям системы искусственного интеллекта для погоды и климата. Его полностью открытая структура позволяет разработчикам настраивать и совершенствовать симуляции в соответствии со своими требованиями, используя собственные данные и инфраструктуру для создания управляемых, суверенных прогнозов погоды и климата.

С этим подходом Earth-2 предоставляет передовой набор моделей погоды и климата с открытым исходным кодом, поддерживаемых доступной экосистемой программного обеспечения, что позволяет создавать индивидуальные, полностью управляемые возможности прогнозирования.

NVIDIA расширяет набор моделей Earth-2 Weather AI на Hugging Face с моделями StormScope, Atlas и HealDA

Выпущенная на Hugging Face, Earth-2 Nowcasting представляет архитектуру модели под названием StormScope, которая использует генеративный ИИ для создания прогнозов высокого разрешения на короткий срок. Система генерирует прогнозы локальных штормов и опасных погодных условий на километровом масштабе за до шести часов вперед, предоставляя результаты за считанные минуты.

Путем моделирования динамики штормов напрямую и прогнозирования спутниковых и радарных данных, Earth-2 Nowcasting способен превосходить традиционные физические модели в краткосрочных прогнозах осадков. Текущая версия обучена на геостационарных спутниковых наблюдениях за континентальной частью США, хотя методология может быть адаптирована для других регионов с сопоставимым спутниковым покрытием.

Параллельно, Earth-2 Medium Range, также доступная на платформе, построена на новой архитектуре модели Atlas, предназначенной для среднесрочного прогнозирования на срок до 15 дней. Система прогнозирует более семидесяти атмосферных переменных, включая температуру, ветер, давление и влажность, используя трансформер с латентным диффузионным механизмом, который моделирует постепенные изменения в атмосфере, чтобы сохранить структурную целостность и минимизировать ошибки. Тестирование показывает, что Earth-2 Medium Range превосходит другие открытые модели прогнозирования, включая GenCast, по наиболее часто используемым индустриальным переменным.

Кроме того, скоро на Hugging Face появится Earth-2 Global Data Assimilation с архитектурой HealDA, которая генерирует начальные атмосферные условия в тысячах точек по всему миру, включая температуру, скорость ветра, влажность и давление воздуха. Производя эти снимки за секунды на GPU, а не за часы на традиционных суперкомпьютерах, HealDA обеспечивает более быстрое и точное инициализацию для прогнозов. В сочетании с Earth-2 Medium Range он образует полностью ИИ-управляемый конвейер, способный предоставлять очень точные прогнозы в масштабах.

Эти дополнения расширяют существующий набор моделей погоды и климата NVIDIA, включающий FourcastNet3, CorrDiff, cBottle и DLESym. В поддержку этой экосистемы NVIDIA Earth2Studio предоставляет открытый фреймворк на Python для быстрого создания моделей ИИ для погоды и климата, предлагая инструменты для вывода и интеграции с новыми контрольными точками моделей.