fbpx
  • Нейросеть MetNet предсказывает осадки на ближайшие 8 часов

    Google AI опубликовали нейросеть, которая предсказывает атмосферные осадки на ближайшие 8 часов. MetNet обходит систему, которая основана на физических законах и используются в NOAA для предсказания осадков. Предложенная модель способна выдавать предсказания в радиусе 1 километра для двухминутных интервалов за несколько секунд, а не за час. MetNet принимает на вход неразмеченные данные, которые автоматически собираются с радаров и спутников. На выходе модель выдает распределение вероятностей выпадения осадков для каждого региона.

    Архитектура модели

    Модель работает с квадратами 64 км x 64 км, которые покрывают Соединенные Штаты Америки с разрешением в 1 км. Фактически модель учитывает данные вне квадратов, чтобы принимать во внимание возможные передвижения облаков и осадков. Чтобы учесть контекст вокруг текущего квадрата, модель обрабатывает данные в округе 1024 км x 1024 км. Модель состоит из экодера (сверточная LSTM) и пространственного агрегатора. Агрегатор использует self-attention для каждой оси, чтобы улавливать долгосрочные взаимосвязи в данных. Итоговое предсказание модель выдает для квадрата 64 км x 64 км.

    Составные части архитектуры модели

    Результаты работы модели

    Исследователи оценили MetNet на базовом датасете с уровнями атмосферных осадков. В качестве базовых моделей для сравнения они выбрали NOAA High Resolution Rapid Refresh (HRRR) и модель, которая предсказывает передвижение оптического потока. Ниже видно, что MetNet обходит базовые модели.

    Сравнение MetNet с бейзлайн-моделями