Uncertainty Quantification and Explainable AI for Aneurysm Rupture Risk Prediction: A Bayesian Deep Learning Approach to Hemodynamic Analysis
PDF

Keywords

Uncertainty Quantification
Aneurysm Rupture Prediction
Bayesian Deep Learning
Explainable AI

Abstract

Aneurysm rupture is a life-threatening medical emergency with high mortality rates, making accurate rupture risk prediction essential for clinical decision-making. While computational hemodynamic modeling and deep learning approaches have shown promise in aneurysm assessment, existing methods often lack robust uncertainty quantification and interpretable predictions that clinicians can trust. This paper proposes a Bayesian deep learning framework for aneurysm rupture risk prediction that combines hemodynamic feature analysis with uncertainty-aware neural networks and explainable AI techniques. Our approach leverages Monte Carlo dropout for epistemic uncertainty estimation and integrates SHAP-based explanations to provide clinicians with transparent, calibrated risk predictions. Through extensive experiments on clinical hemodynamic datasets, we demonstrate that the proposed framework produces well-calibrated probability estimates and provides interpretable explanations that highlight the hemodynamic features most critical for rupture risk. Our work contributes to the growing field of trustworthy AI in healthcare, providing a principled approach to uncertainty-aware, explainable medical decision support for aneurysm management.

PDF

References

1. Ruder, S. (2017). An overview of multi-task learning in deep neural networks. arXiv preprint arXiv:1706.05098.

2. Zang, J., & Liu, H. (2024, June). Explanation based bias decoupling regularization for natural language inference. In 2024 International Joint Conference on Neural Networks (IJCNN) (pp. 1-8). IEEE.

3. Liu D, Wang X, Zhao D, Sun Z, Biekan J, Wen Z, Xu L and Liu J (2022) Influence of MRI-based boundary conditions on type B aortic dissection simulations in false lumen with or without abdominal aorta involvement. Front. Physiol. 13:977275. doi: 10.3389/fphys.2022.977275

4. Gal, Y., & Ghahramani, Z. (2016). Dropout as a Bayesian approximation: Representing model uncertainty in deep learning. In International Conference on Machine Learning (ICML) (pp. 1050-1059). PMLR.

5. Lu, Y., Hu, K., & Zhang, L. (2026, May). S3G: Stock State Space Graph for Enhanced Stock Trend Prediction. In ICASSP 2026-2026 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) (pp. 4081-4085). IEEE.

6. Fan T, Wang J, Wang X, Chen X, Zhao D, Xie F and Chen G (2025) Integrated multidisciplinary approach to aneurysm hemodynamic analysis: numerical simulation, in Vitro experiment, and deep learning. Front. Bioeng. Biotechnol. 13:1602190. doi: 10.3389/fbioe.2025.1602190

7. Lundberg, S. M., & Lee, S. I. (2017). A unified approach to interpreting model predictions. In Advances in Neural Information Processing Systems 30 (NeurIPS) (pp. 4765-4774). Curran Associates.

8. Selvaraju, R. R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., & Batra, D. (2017). Grad-CAM: Visual explanations from deep networks via gradient-based localization. In Proceedings of the IEEE International Conference on Computer Vision (ICCV) (pp. 618-626). IEEE.

9. Ye X, Zhang J, Cheng Z, Lu F, Wen Z, Yu G, Chen G, Xie F, Qiao D, Xing J, Tan W, Zhao D and Ren M (2025) Hemodynamic modeling of aortic arch aneurysm treatment using the Castor™ branched stent graft: a virtual coil embolization simulation framework. Front. Physiol. 16:1629346. doi: 10.3389/fphys.2025.1629346

10. Kendall, A., & Gal, Y. (2017). What uncertainties do we need in Bayesian deep learning for computer vision? In Advances in Neural Information Processing Systems 30 (NeurIPS) (pp. 5574-5584). Curran Associates.

11. Ronneberger, O., Fischer, P., & Brox, T. (2015). U-Net: Convolutional networks for biomedical image segmentation. In International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI) (pp. 234-241). Springer.

12. Guo, C., Pleiss, G., Sun, Y., & Weinberger, K. Q. (2017). On calibration of modern neural networks. In International Conference on Machine Learning (ICML) (pp. 1321-1330). PMLR.

13. Klein, A., Falkner, S., Mansur, N., & Hutter, F. (2017). RoBO: A flexible and robust Bayesian optimization framework in Python. In NIPS Bayesian Optimization Workshop.

14. Charbonnier, C., et al. (2020). Aneurysm rupture risk assessment using biomechanical and hemodynamic indices: A systematic review. Journal of Biomechanics, 112, 110045.

15. Zhang, Y., et al. (2023). Uncertainty-aware deep learning for medical image analysis. Medical Image Analysis, 83, 102652.