An Uncertainty-aware Hierarchical Probabilistic Network for Early Prediction, Quantification and Segmentation of Pulmonary Tumour Growth

Abstract

Early detection and quantification of tumour growth would help clinicians to prescribe more accurate treatments and provide better surgical planning. However, the multifactorial and heterogeneous nature of lung tumour progression hampers identification of growth patterns. In this study, we present a novel method based on a deep hierarchical generative and probabilistic framework that, according to radiological guidelines, predicts tumour growth, quantifies its size and provides a semantic appearance of the future nodule. Unlike previous deterministic solutions, the generative characteristic of our approach also allows us to estimate the uncertainty in the predictions, especially important for complex and doubtful cases. Results of evaluating this method on an independent test set reported a tumour growth balanced accuracy of 74%, a tumour growth size MAE of 1.77 mm and a tumour segmentation Dice score of 78%. These surpassed the performances of equivalent deterministic and alternative generative solutions (i.e. probabilistic U-Net, Bayesian test dropout and Pix2Pix GAN) confirming the suitability of our approach.

arXiv comment24 pages, 9 figures

References in corpus (11)

• Adam: A Method for Stochastic Optimization
• Attention U-Net: Learning Where to Look for the Pancreas
• Road Extraction by Deep Residual U-Net
• Stochastic Backpropagation and Approximate Inference in Deep Generative Models
• Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challenge
• NIPS 2016 Tutorial: Generative Adversarial Networks
• Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inference
• Personalized Radiotherapy Design for Glioblastoma: Integrating Mathematical Tumor Models, Multimodal Scans and Bayesian Inference
• Convolutional Invasion and Expansion Networks for Tumor Growth Prediction
• Uncertainty Estimations by Softplus normalization in Bayesian Convolutional Neural Networks with Variational Inference
• A Hierarchical Probabilistic U-Net for Modeling Multi-Scale Ambiguities