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

Коротко

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

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

Розробники можуть використовувати відкриті програми NVIDIA, такі як Earth2Studio для побудови інференційних конвеєрів і Physics Nemo для тренування моделей, щоб швидко створювати моделі погоди та клімату.

Набір Earth-2 об’єднує ряд прискорених інструментів і моделей, об’єднуючи можливості, які раніше були розподілені по всьому ландшафту AI для погоди та клімату. Його повністю відкритий каркас дозволяє розробникам налаштовувати та вдосконалювати моделі відповідно до своїх потреб, використовуючи власні дані та інфраструктуру для створення самостійних, суверенних прогнозів погоди та клімату.

З цим підходом Earth-2 пропонує провідний набір відкритих моделей погоди та клімату, підтримуваний доступною екосистемою програмного забезпечення, що дозволяє створювати індивідуальні, повністю контрольовані можливості прогнозування.

NVIDIA розширює набір Earth-2 Weather AI на Hugging Face моделями StormScope, Atlas і HealDA

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

Моделюючи динаміку бур безпосередньо та прогнозуючи виходи супутників і радарів, 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 це формує повністю AI-управляний конвеєр, здатний забезпечувати високоточні прогнози у масштабі.

Ці доповнення розширюють існуючий набір відкритих моделей погоди та клімату NVIDIA, до якого входять FourcastNet3, CorrDiff, cBottle і DLESym. У підтримку цієї екосистеми NVIDIA Earth2Studio надає відкритий Python-фреймворк для швидкого створення AI-моделей погоди та клімату, пропонуючи інструменти для інференції та інтеграції з новими контрольними точками моделей.