Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening
Nan Wu,Jason Phang,Jungkyu Park,Yiqiu Shen,Zhe Huang,Masha Zorin,Stanisław Jastrzębski,Thibault Févry,Joe Katsnelson,Eric Kim,Stacey Wolfson,Ujas Parikh,Sushma Gaddam,Leng Leng Young Lin,Kara Ho,Joshua D. Weinstein,Beatriu Reig,Yiming Gao,Hildegard Toth,Kristine Pysarenko,Alana Lewin,Jiyon Lee,Krystal Airola,Eralda Mema,Stephanie Chung,Esther Hwang,Naziya Samreen,S. Gene Kim,Laura Heacock,Linda Moy,Kyunghyun Cho,Krzysztof J. Geras
DOI: https://doi.org/10.48550/arXiv.1903.08297
2019-03-20
Abstract:We present a deep convolutional neural network for breast cancer screening exam classification, trained and evaluated on over 200,000 exams (over 1,000,000 images). Our network achieves an AUC of 0.895 in predicting whether there is a cancer in the breast, when tested on the screening population. We attribute the high accuracy of our model to a two-stage training procedure, which allows us to use a very high-capacity patch-level network to learn from pixel-level labels alongside a network learning from macroscopic breast-level labels. To validate our model, we conducted a reader study with 14 readers, each reading 720 screening mammogram exams, and find our model to be as accurate as experienced radiologists when presented with the same data. Finally, we show that a hybrid model, averaging probability of malignancy predicted by a radiologist with a prediction of our neural network, is more accurate than either of the two separately. To better understand our results, we conduct a thorough analysis of our network's performance on different subpopulations of the screening population, model design, training procedure, errors, and properties of its internal representations.
Machine Learning,Computer Vision and Pattern Recognition