Journal
IEEE SIGNAL PROCESSING LETTERS
Volume 27, Issue -, Pages 2024-2028Publisher
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/LSP.2020.3034858
Keywords
Brain Tumor Classification; Bayesian Capsule Networks; Uncertainty Measure; Radiomics
Categories
Funding
- Natural Sciences and Engineering Research Council (NSERC) of Canada under the NSERC [RGPIN-2016-04988]
Ask authors/readers for more resources
Convolutional neural networks (CNNs), which have been the state-of-the-art in many image-related applications, are prone to losing important spatial information between image instances. Capsule networks (CapsNets), on the other hand, are capable of leveraging such information through their routing by agreement process, making them powerful architectures for small datasets, such asmedical imaging ones. Within the domain of medical imaging problems, brain tumor classification is of paramount importance, due to the deadly nature of this cancer and the consequences of the tumor misclassification. In our recent works, we showed potentials of developing CapsNet architecture for the task of brain tumor type classification. Similar to other deep learning models, however, CapsNets do not capture prediction uncertainty (coming from the uncertainty in the model weights, which is significantly important in keeping the human experts in the loop, by returning the uncertain samples. In this paper, we propose a Bayesian CapsNet framework, referred to as the BayesCap, that can provide not only the mean predictions, but also entropy as a measure of prediction uncertainty. Results show that filtering out the uncertain predictions can improve the accuracy, confirming that returning the uncertain predictions is an appropriate strategy for improving interpretability of the network.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available