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 обходит базовые модели.
