В Google AI использовали сверточную нейросеть для предсказания атмосферных осадков на ближайшие 6 часов. Модель основывается на архитектуре U-Net. Предложенная нейросеть обходит традиционные модели для предсказания осадков по точности предсказаний.
Метеорологические агентства используют различные инструменты для мониторинга погоды. Примером являются радар Doppler, который измеряет осадки в реальном времени, погодные спутники, которые собирают мультиспектральные изображения, и наземные станции, которые измеряют силу ветра и количество осадков напрямую. На текущий момент предсказание осадков требует сбора данных одновременно из нескольких источников.
В Google AI для предсказания осадков использовали сверточную нейросеть, которая принимает на вход снимок с радара и выдает предсказание на ближайшее будущее. При этом особенностью модели является то, что она делает локализованные предсказания на короткий промежуток времени (до 6 часов). Исследователи называют такую задачу “nowcasting”.
Преимущества использования машинного обучения заключаются в:
- Низкой задержке предсказаний (5-10 минут);
- Низкой вычислительной затратности обученной модели на инференсе;
- Более высокой точности предсказаний, которая превосходит традиционные модели
Как это работает
Примером того, какие предсказания делает модель, является задача radar-to-radar forecasting. На вход модели поступает последовательность снимков с радара. На выходе модель предсказывает, как снимки с радаров изменятся через N часов с текущего момента. N ранжируется от 0 до 6. Из-за того, что данные радаров представлены в виде изображений, исследователи рассматривают проблему как задачу компьютерного зрения.
В предложенном подходе модель не получает никакие метаданные о том, как работает атмосфера. Одной из задач модели является их выучить из входных изображений. Так, задача предсказания осадков превращается в задачу image-to-image translation. В качестве архитектуры исследователи выбрали U-Net.
Сравнение модели с традиционными подходами
Исследователи сравнили нейросетевой подход с стандартными моделями для предсказания осадков. Среди стандартных моделей были High Resolution Rapid Refresh (HRRR), optical flow (OF) алгоритм и persistence model. Ниже видно, качество предсказаний HRRR и сверточной нейросети. Нейросеть обходит стандартный подход.
