Enhancing Disaster Prediction with Bayesian Deep Learning: A Robust Approach for Uncertainty Estimation

Hao Peng^1*Sen Shen²Haichao Zhang¹Fei Wang¹Fawang Guo¹Ruige Zhang³

• ¹Power China Guiyang Engineering Corporation Limited, Guiyang, China
• ²The Chinese University of Hong Kong Department of Mechanical and Automation Engineering, Hong Kong, Hong Kong, SAR China
• ³Henan Zhuqueyun Network Technology Co., Ltd., Zhengzhou, China

Accurate disaster prediction combined with reliable uncertainty quantification is crucial for timely and effective decision-making in emergency management. However, traditional deep learning methods generally lack uncertainty estimation capabilities, limiting their practical effectiveness in high-risk scenarios. To overcome these limitations, this study proposes an enhanced Bayesian Deep Neural Network (BDNN) tailored for flood forecasting, effectively integrating Variational Inference (VI), Monte Carlo (MC) Dropout, and a Hierarchical Attention Mechanism. By leveraging hydrological and meteorological data from the Yellow River basin (2001-2023), the BDNN model not only achieves superior prediction accuracy (94.6%) but also significantly enhances reliability through robust uncertainty quantification. Comparative analyses demonstrate that the proposed approach markedly outperforms conventional models such as Random Forest, XGBoost, and Multi-layer Perceptron. Ablation studies further confirm the critical role of the hierarchical attention mechanism in capturing essential features, while VI and MC Dropout substantially improve prediction reliability. These advancements highlight the potential of BDNNs to significantly enhance disaster preparedness and support more informed emergency response decisions in complex, uncertain environments.