Random Bundle: Brain Metastases Segmentation Ensembling through Annotation Randomization

Abstract

We introduce a novel ensembling method, Random Bundle (RB), that improves performance for brain metastases segmentation. We create our ensemble by training each network on our dataset with 50% of our annotated lesions censored out. We also apply a lopsided bootstrap loss to recover performance after inducing an in silico 50% false negative rate and make our networks more sensitive. We improve our network detection of lesions's mAP value by 39% and more than triple the sensitivity at 80% precision. We also show slight improvements in segmentation quality through DICE score. Further, RB ensembling improves performance over baseline by a larger margin than a variety of popular ensembling strategies. Finally, we show that RB ensembling is computationally efficient by comparing its performance to a single network when both systems are constrained to have the same compute.

References in corpus (8)

• Rethinking Atrous Convolution for Semantic Image Segmentation
• Wide Residual Networks
• Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challenge
• Training Deep Neural Networks on Noisy Labels with Bootstrapping
• Deep Learning Enables Automatic Detection and Segmentation of Brain Metastases on Multi-Sequence MRI
• Fully Convolutional Network Ensembles for White Matter Hyperintensities Segmentation in MR Images
• MRI Pulse Sequence Integration for Deep-Learning Based Brain Metastasis Segmentation
• Brain Metastasis Segmentation Network Trained with Robustness to Annotations with Multiple False Negatives