Precipitation Prediction Using an Ensemble of Lightweight Learners

Xinzhe Li,Sun Rui,Yiming Niu,Yao Liu
2023-11-30
Abstract:Precipitation prediction plays a crucial role in modern agriculture and industry. However, it poses significant challenges due to the diverse patterns and dynamics in time and space, as well as the scarcity of high precipitation events.
Atmospheric and Oceanic Physics,Artificial Intelligence,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
This paper aims to address the challenges of precipitation prediction, especially in the context of modern agriculture and industry. The current issues include the sparsity of high precipitation events, leading to insufficient training data, as well as the complex and diverse temporal and spatial patterns of precipitation. To tackle these problems, the paper proposes an ensemble learning framework that utilizes multiple lightweight learners to capture the various patterns of precipitation distribution. Specifically, the paper adopts WeatherFusionNet as the base network and re-trains its included U-Net components and additional ConvLSTM modules to handle temporal sequence correlations. Additionally, they introduce an ensemble composed of multiple one-layer convolution learners, which are independently optimized to capture the diversity of precipitation patterns. Similar to the MoE method, the paper also designs a controller model that generates weight maps to focus the learners' attention on potentially rainy areas. Through a three-stage training strategy, the paper ensures that each module is sufficiently optimized at different stages. The experiments demonstrate that this approach achieved first place in the core test and short-term forecast leaderboard of the Weather4Cast 2023 competition, as well as second place in the transfer learning leaderboard, particularly excelling in the prediction of high precipitation events. In conclusion, the proposed ensemble learning framework improves the accuracy of precipitation prediction, particularly for complex and rare precipitation events.