Proceedings

Context-Aware Concept Distillation for Trustworthy Flood Prediction

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Social Good. Pages 7283-7291.

https://doi.org/10.24963/ijcai.2026/810

Abstract

Effective flood risk management relies on accurate forecasting, yet the ”black box” nature of state-of-the-art Deep Learning models creates a barrier

to trust and accountability in high-stakes public safety decisions. While existing Explainable AI (XAI) methods offer local attributions, they fail to provide the verifiable, operationally meaningful causal narratives required by disaster response authorities. To address this societal challenge, we propose Context-Aware Concept Distillation (CACD), a framework developed in collaboration with domain experts to distill opaque LSTMs into interpretable, hydrology-aware surrogate models. We introduce an unsupervised pipeline to discover a ”Hydrological Language” and a Residual Hypernetwork that dynamically modulates these concepts based on static basin characteristics. Evaluated on 5,203 basins globally, our model achieves high fidelity (Median NSE 0.70), significantly outperforming black-box baselines (e.g., Multi Layer Perceptrons) on unseen future data. By demonstrating that human-interpretable concepts are sufficient to reconstruct flood dynamics, this work balances AI accuracy with the transparency required for responsible environmental decision-making.

  1. [Bramm et al., 2025] Andrei M. Bramm, Pavel V. Matrenin, and Alexandra I. Khalyasmaa. A review of xai methods applications in forecasting runoff and water level hydro- logical tasks. Mathematics, 13(17):2830, 2025.
  2. [Dikshit et al., 2024] Abhirup Dikshit, Biswajeet Pradhan, Sahar S. Matin, Ghassan Beydoun, M. Santosh, Hyuck-Jin Park, and Khairul Nizam Abdul Maulud. Artificial intel- ligence: A new era for spatial modelling and interpreting climate-induced hazard assessment. Geoscience Frontiers, 15(4):101815, 2024.
  3. [Frame et al., 2022] Jonathan M. Frame, Frederik Kratzert, Daniel Klotz, Martin Gauch, Guy Shalev, Oren Gilon, Lorne M. Qualls, Hoshin V. Gupta, and Grey S. Near- ing. Deep learning rainfall-runoff predictions of ex- treme events. Hydrology and Earth System Sciences, 26(13):3377-3392, 2022.
  4. [Jiang et al., 2022] Shijie Jiang, Emanuele Bevacqua, and Jakob Zscheischler. River flooding mechanisms and their changes in europe revealed by explainable machine learn- ing. Hydrology and Earth System Sciences, 26(24):6339- 6359, 2022.
  5. [Kim et al., 2018] Been Kim, Martin Wattenberg, Justin Gilmer, Carrie Cai, James Wexler, Fernanda Viegas, and Rory Sayres. Interpretability beyond feature attribution: Quantitative testing with concept activation vectors (tcav). In Proceedings of the 35th International Conference on Machine Learning (ICML), pages 2668-2677. PMLR, 2018.
  6. [Koh et al., 2020] Pang Wei Koh, Thao Nguyen, Yew Siang Tang, Stephen Mussmann, Emma Pierson, Been Kim, and Percy Liang. Concept bottleneck models. In Proceedings of the 37th International Conference on Machine Learning (ICML), pages 5338-5348. PMLR, 2020.
  7. [Kratzert et al., 2019] Frederik Kratzert, Daniel Klotz, Guy Shalev, Günter Klambauer, Sepp Hochreiter, and Grey Nearing. Towards learning universal, regional, and lo- cal hydrological behaviors via machine learning applied to large-sample datasets. Hydrology and Earth System Sci- ences, 23(12):5089-5110, 2019.
  8. [Kratzert et al., 2023] Frederik Kratzert, Grey Nearing, Nans Addor, Tyler Erickson, Martin Gauch, Oren Gilon, Lukas Gudmundsson, Avinatan Hassidim, Daniel Klotz, Sella Nevo, Guy Shalev, and Yossi Matias. Caravan: A global community dataset for large-sample hydrology. Sci- entific Data, 10(1):61, 2023.
  9. [Liu et al., 2024] Jiangtao Liu, Yuchen Bian, Kathryn Law- son, and Chaopeng Shen. Probing the limit of hydrologic predictability with the transformer network. Journal of Hydrology, 637:131389, 2024.
  10. [Lundberg and Lee, 2017] Scott M. Lundberg and Su-In Lee. A unified approach to interpreting model predic- tions. In Advances in Neural Information Processing Sys- tems (NeurIPS), volume 30, pages 4765-4774, 2017.
  11. [Mamalakis et al., 2022] Antonios Mamalakis, Imme Ebert- Uphoff, and Elizabeth A. Barnes. Explainable artificial in- telligence in meteorology and climate science: Model fine- tuning, calibrating trust and learning new science. In An- dreas Holzinger, Randy Goebel, Ruth Fong, Tero Moon, Klaus-Robert Müller, and Wojciech Samek, editors, xxAI - Beyond Explainable AI, volume 13200 of Lecture Notes in Computer Science, pages 315-339. Springer, Cham, 2022.
  12. [Moriasi et al., 2007] Daniel N Moriasi, Jeffrey G Arnold, Michael W Van Liew, Ronald L Bingner, Robert Daren Harmel, and Tamie L. Veith. Model evaluation guide- lines for systematic quantification of accuracy in water- shed simulations. Transactions of the ASABE, 50(3):885- 900, 2007.
  13. [Morin et al., 2009] Efrat Morin, Yael Jacoby, Shilo Navon, and Erez Bet-Halachmi. Towards flash-flood prediction in the dry dead sea region utilizing radar rainfall information. Advances in Water Resources, 32(7):1066-1076, 2009.
  14. [Nearing et al., 2024] Grey Nearing, Deborah Cohen, Vusumuzi Dube, Martin Gauch, Oren Gilon, Shaun Harri- gan, Avinatan Hassidim, Daniel Klotz, Frederik Kratzert, Asher Metzger, Sella Nevo, Florian Pappenberger, Christel Prudhomme, Guy Shalev, Shlomo Shenzis, Tadele Yednkachw Tekalign, Dana Weitzner, and Yossi Matias. Global prediction of extreme floods in ungauged watersheds. Nature, 627:559-563, 2024.
  15. [Nevo et al., 2022] Sella Nevo, Efrat Morin, Adi Gerzi Rosenthal, Asher Metzger, Chen Barshai, Dana Weitzner, Dafi Voloshin, Frederik Kratzert, Gal Elidan, Gideon Dror, Gregory Begelman, Grey Nearing, Guy Shalev, Hila Noga, Ira Shavitt, Liora Yuklea, Moriah Royz, Niv Giladi, Nofar Peled Levi, Ofir Reich, Oren Gilon, Ronnie Maor, Shahar Timnat, Tal Shechter, Vladimir Anisimov, Yotam Gigi, Yuval Levin, Zach Moshe, Zvika Ben-Haim, Avinatan Hassidim, and Yossi Matias. Flood forecasting with machine learning models in an operational framework. Hydrology and Earth System Sciences, 26(15):4013-4032, 2022.
  16. [Ribeiro et al., 2016] Marco Tulio Ribeiro, Sameer Singh, and Carlos Guestrin. "why should i trust you?": Explain- ing the predictions of any classifier. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowl- edge Discovery and Data Mining (KDD), pages 1135- 1144, 2016.
  17. [Rinat et al., 2018] Yair Rinat, Francesco Marra, Davide Zoccatelli, and Efrat Morin. Controls of flash flood peak discharge in mediterranean basins and the special role of runoff-contributing areas. Journal of Hydrology, 565:846- 860, 2018.
  18. [Rojat et al., 2021] Thomas Rojat, Raphaël Puget, David Filliat, Javier Del Ser, Rodolphe Gelin, and Na- talia Díaz-Rodríguez. Explainable artificial intelligence (xai) on timeseries data: A survey. arXiv preprint arXiv:2104.00950, 2021.
  19. [Selvaraju et al., 2017] Ramprasaath R. Selvaraju, Michael Cogswell, Abhishek Das, Ramakrishna Vedantam, Devi Parikh, and Dhruv Batra. Grad-cam: Visual explana- tions from deep networks via gradient-based localization. In Proceedings of the IEEE International Conference on Computer Vision (ICCV), pages 618-626, 2017.
  20. [Simonyan et al., 2014] Karen Simonyan, Andrea Vedaldi, and Andrew Zisserman. Deep inside convolutional networks: Visualising image classification models and saliency maps. In Proceedings of the International Con- ference on Learning Representations (ICLR) Workshop, 2014.
  21. [Visave, 2025] Jaideep Visave. Transparency in ai for emer- gency management: building trust and accountability. AI and Ethics, 5:3967-3980, Mar 2025.
  22. [Vu et al., 2025] Kiana Vu, İsmet Selc ¸uk Özer, Phung Lai, Zheng Wu, Thilanka Munasinghe, and Jennifer Wei. From black box to insight: Explainable ai for extreme event pre- paredness. arXiv preprint arXiv:2511.13712, 2025.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards