Modeling Uncertainty in Multi-Modal Fusion for Lung Cancer Survival Analysis

Fusion of multimodal data is important for disease understanding. In this paper, we propose a new method of fusion exploiting the uncertainty in prediction produced by the individual modality learners. Specifically, we extend the joint label fusion method by taking model uncertainty into account when estimating correlations among predictions produced by different modalities. Through experimental study in survival prediction for non-small cell lung cancer patients who received surgical resection, we demonstrated promising performance produced by the proposed method.

[1]  Minh N. Do,et al.  Multimodal Fusion of Imaging and Genomics for Lung Cancer Recurrence Prediction , 2020, 2020 IEEE 17th International Symposium on Biomedical Imaging (ISBI).

[2]  Louis-Philippe Morency,et al.  Multimodal Machine Learning: A Survey and Taxonomy , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[3]  Andriy Fedorov,et al.  Computational Radiomics System to Decode the Radiographic Phenotype. , 2017, Cancer research.

[4]  J. Kongerud,et al.  Long-term survival after surgical resection for non-small cell lung cancer , 2017 .

[5]  Charles Blundell,et al.  Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles , 2016, NIPS.

[6]  Adrian E. Raftery,et al.  Bayesian Model Averaging: A Tutorial , 2016 .

[7]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[8]  Paul A. Yushkevich,et al.  Multi-Atlas Segmentation with Joint Label Fusion , 2013, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[9]  Nitish Srivastava,et al.  Multimodal learning with deep Boltzmann machines , 2012, J. Mach. Learn. Res..

[10]  Juhan Nam,et al.  Multimodal Deep Learning , 2011, ICML.

[11]  A. Fagan,et al.  Multimodal techniques for diagnosis and prognosis of Alzheimer's disease , 2009, Nature.

[12]  Samy Bengio,et al.  How do correlation and variance of base-experts affect fusion in biometric authentication tasks? , 2005, IEEE Transactions on Signal Processing.

[13]  Sargur N. Srihari,et al.  Decision Combination in Multiple Classifier Systems , 1994, IEEE Trans. Pattern Anal. Mach. Intell..