3D Deep Attention Network for Survival Prediction from Magnetic Resonance Images in Glioblastoma

Existing deep convolutional neural network-based survival analysis neither consider the modern attention mechanism nor use 3D tomographic medical images such as magnetic resonance images (MRI). This paper for the first time presents a 3D deep convolutional neural network using attention mechanism for survival prediction from multiparametric MRI in glioblastoma (GBM) patients. The attention module is incorporated into the residual network to enhance the representation power of meaningful features while suppress unimportant ones. The proposed model achieves an C-index of 0.71 in the training dataset and 0.68 in an independent test dataset, which outperforms both the traditional Cox model (0.60,0.54) and the non-attentive model (0.63,0.61). It indicates that the proposed 3D attention network has the potential of offering better performance in predicting survival using MRI than traditional survival analysis.