Title
Improving Uncertainty Estimations for Mammogram Classification using Semi-Supervised Learning
Abstract
Computer aided diagnosis for mammogram images have seen positive results through the usage of deep learning architectures. However, limited sample sizes for the target datasets might prevent the usage of a deep learning model under real world scenarios. The usage of unlabeled data to improve the accuracy of the model can be an approach to tackle the lack of target data. Moreover, important model attributes for the medical domain as model uncertainty might be improved through the usage of unlabeled data. Therefore, in this work we explore the impact of using unlabeled data through the implementation of a recent approach known as MixMatch, for mammogram images. We evaluate the improvement on accuracy and uncertainty of the model using popular and simple approaches to estimate uncertainty. For this aim, we propose the usage of the uncertainty balanced accuracy metric.
Year
DOI
Venue
2021
10.1109/IJCNN52387.2021.9533719
2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN)
Keywords
DocType
ISSN
Uncertainty Estimation, Breast Cancer, Mammogram, Semi-Supervised Deep Learning, MixMatch
Conference
2161-4393
Citations 
PageRank 
References 
0
0.34
0
Authors
9