Abstract
Intracerebral hemorrhage (ICH) is the second most common and deadliest form of stroke. Despite medical advances, predicting treatment outcomes for ICH remains a challenge. This paper proposes a novel prognostic model that utilizes both imaging and tabular data to predict treatment outcome for ICH. Our model is trained on observational data collected from non-randomized controlled trials, providing reliable predictions of treatment success. Specifically, we propose to employ a variational autoencoder model to generate a low-dimensional prognostic score, which can effectively address the selection bias resulting from the non-randomized controlled trials. Importantly, we develop a variational distributions combination module that combines the information from imaging data, non-imaging clinical data, and treatment assignment to accurately generate the prognostic score. We conducted extensive experiments on a real-world clinical dataset of intracerebral hemorrhage. Our proposed method demonstrates a substantial improvement in treatment outcome prediction compared to existing state-of-the-art approaches. Code is available at https://github.com/med-air/TOP-GPM.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Bica, I., Alaa, A.M., Lambert, C., Van Der Schaar, M.: From real-world patient data to individualized treatment effects using machine learning: current and future methods to address underlying challenges. Clin. Pharmacol. Therap. 109(1), 87–100 (2021)
Cheung, R.T.F., Zou, L.Y.: Use of the original, modified, or new intracerebral hemorrhage score to predict mortality and morbidity after intracerebral hemorrhage. Stroke 34(7), 1717–1722 (2003)
D’Amour, A., Ding, P., Feller, A., Lei, L., Sekhon, J.: Overlap in observational studies with high-dimensional covariates. J. Econometrics 221(2), 644–654 (2021)
Ducruet, A.F., et al.: The complement cascade as a therapeutic target in intracerebral hemorrhage. Exp. Neurol. 219(2), 398–403 (2009)
Feigin, V.L., Lawes, C.M., Bennett, D.A., Barker-Collo, S.L., Parag, V.: Worldwide stroke incidence and early case fatality reported in 56 population-based studies: a systematic review. Lancet Neurol. 8(4), 355–369 (2009)
Flaherty, M., et al.: Long-term mortality after intracerebral hemorrhage. Neurology 66(8), 1182–1186 (2006)
Godoy, D.A., Pinero, G., Di Napoli, M.: Predicting mortality in spontaneous intracerebral hemorrhage: can modification to original score improve the prediction? Stroke 37(4), 1038–1044 (2006)
Gregório, T., et al.: Prognostic models for intracerebral hemorrhage: systematic review and meta-analysis. BMC Med. Res. Methodol. 18, 1–17 (2018)
Gregson, B.A., Mitchell, P., Mendelow, A.D.: Surgical decision making in brain hemorrhage: new analysis of the stich, stich ii, and stitch (trauma) randomized trials. Stroke 50(5), 1108–1115 (2019)
Guo, R., et al.: Machine learning-based approaches for prediction of patients’ functional outcome and mortality after spontaneous intracerebral hemorrhage. J. Personal. Med. 12(1), 112 (2022)
Hall, A.N., et al.: Identifying modifiable predictors of patient outcomes after intracerebral hemorrhage with machine learning. Neurocrit. Care 34, 73–84 (2021)
Hansen, B.B.: The prognostic analogue of the propensity score. Biometrika 95(2), 481–488 (2008)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: CVPR, pp. 770–778 (2016)
Hemphill, J.C., III, Bonovich, D.C., Besmertis, L., Manley, G.T., Johnston, S.C.: The ICH score: a simple, reliable grading scale for intracerebral hemorrhage. Stroke 32(4), 891–897 (2001)
Hemphill, J.C., III, et al.: Guidelines for the management of spontaneous intracerebral hemorrhage: a guideline for healthcare professionals from the American Heart Association/American Stroke Association. Stroke 46(7), 2032–2060 (2015)
Hinton, G.E.: Training products of experts by minimizing contrastive divergence. Neural Comput. 14(8), 1771–1800 (2002)
Huang, S.C., Pareek, A., Seyyedi, S., Banerjee, I., Lungren, M.P.: Fusion of medical imaging and electronic health records using deep learning: a systematic review and implementation guidelines. NPJ Digit. Med. 3(1), 1–9 (2020)
Ji, R., et al.: A novel risk score to predict 1-year functional outcome after intracerebral hemorrhage and comparison with existing scores. Critical Care 17, 1–10 (2013)
Johansson, F., Shalit, U., Sontag, D.: Learning representations for counterfactual inference. In: ICML, pp. 3020–3029. PMLR (2016)
Kim, J.Y., Bae, H.J.: Spontaneous intracerebral hemorrhage: management. J. Stroke 19(1), 28 (2017)
Kingma, D.P., Welling, M.: Auto-encoding variational Bayes. In: ICLR (2014)
Lee, C., van der Schaar, M.: A variational information bottleneck approach to multi-omics data integration. In: International Conference on Artificial Intelligence and Statistics, pp. 1513–1521. PMLR (2021)
Louizos, C., Shalit, U., Mooij, J.M., Sontag, D., Zemel, R., Welling, M.: Causal effect inference with deep latent-variable models. NeurIPS 30 (2017)
Peters, J., Janzing, D., Schölkopf, B.: Elements of Causal Inference: Foundations and Learning Algorithms. The MIT Press (2017)
Pölsterl, S., Wolf, T.N., Wachinger, C.: Combining 3D image and tabular data via the dynamic affine feature map transform. In: de Bruijne, M., et al. (eds.) MICCAI 2021. LNCS, vol. 12905, pp. 688–698. Springer, Cham (2021). https://doi.org/10.1007/978-3-030-87240-3_66
Shalit, U., Johansson, F.D., Sontag, D.: Estimating individual treatment effect: generalization bounds and algorithms. In: ICML, pp. 3076–3085. PMLR (2017)
Shi, C., Blei, D., Veitch, V.: Adapting neural networks for the estimation of treatment effects. NeurIPS 32 (2019)
Shi, Y., Paige, B., Torr, P., et al.: Variational mixture-of-experts autoencoders for multi-modal deep generative models. NeurIPS 32 (2019)
Stein, M., Luecke, M., Preuss, M., Boeker, D.K., Joedicke, A., Oertel, M.F.: Spontaneous intracerebral hemorrhage with ventricular extension and the grading of obstructive hydrocephalus: the prediction of outcome of a special life-threatening entity. Neurosurgery 67(5), 1243–1252 (2010)
Steiner, T., et al.: European Stroke Organisation (ESO) guidelines for the management of spontaneous intracerebral hemorrhage. Int. J. Stroke 9(7), 840–855 (2014)
Teasdale, G., Jennett, B.: Assessment of coma and impaired consciousness: a practical scale. The Lancet 304(7872), 81–84 (1974)
Van Swieten, J., Koudstaal, P., Visser, M., Schouten, H., Van Gijn, J.: Interobserver agreement for the assessment of handicap in stroke patients. Stroke 19(5), 604–607 (1988)
Wu, M., Goodman, N.: Multimodal generative models for scalable weakly-supervised learning. NeurIPS 31 (2018)
Wu, P., Fukumizu, K.: \(\backslash \)beta-intact-VAE: identifying and estimating causal effects under limited overlap. In: ICLR (2022)
Yao, L., Li, S., Li, Y., Huai, M., Gao, J., Zhang, A.: Representation learning for treatment effect estimation from observational data. NeurIPS 31 (2018)
Yoon, J., Jordon, J., Van Der Schaar, M.: Ganite: estimation of individualized treatment effects using generative adversarial nets. In: ICLR (2018)
Acknowledgement
This work was supported in part by Shenzhen Portion of Shenzhen-Hong Kong Science and Technology Innovation Cooperation Zone under HZQB-KCZYB-20200089, in part by Hong Kong Innovation and Technology Commission Project No. ITS/238/21, in part by Hong Kong Research Grants Council Project No. T45-401/22-N, in part by Science, Technology and Innovation Commission of Shenzhen Municipality Project No. SGDX20220530111201008. We also thank the Hong Kong Hospital Authority Data Collaboration Laboratory (HADCL) for their support of this study.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
1 Electronic supplementary material
Below is the link to the electronic supplementary material.
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Ma, W. et al. (2023). Treatment Outcome Prediction for Intracerebral Hemorrhage via Generative Prognostic Model with Imaging and Tabular Data. In: Greenspan, H., et al. Medical Image Computing and Computer Assisted Intervention – MICCAI 2023. MICCAI 2023. Lecture Notes in Computer Science, vol 14224. Springer, Cham. https://doi.org/10.1007/978-3-031-43904-9_69
Download citation
DOI: https://doi.org/10.1007/978-3-031-43904-9_69
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-43903-2
Online ISBN: 978-3-031-43904-9
eBook Packages: Computer ScienceComputer Science (R0)