arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.07613cs.CVcs.LGeess.IVstat.ML

An interpretable classifier for high-resolution breast cancer screening images utilizing weakly supervised localization

Center for Data Science, New York University(纽约大学数据科学中心);Department of Computer Science, Courant Institute, New York University(纽约大学库朗研究所计算机科学系);Department of Radiology, NYU School of Medicine(纽约大学医学院放射科);Perlmutter Cancer Center, NYU Langone Health(纽约大学朗格尼健康系统珀尔马特癌症中心);Center for Advanced Imaging Innovation and Research, NYU Langone Health(纽约大学朗格尼健康系统先进成像创新与研究中心);CIFAR Associate Fellow(CIFAR研究员(注:CIFAR是加拿大英属哥伦比亚大学的一个研究机构,但根据提取规则,此处仅提取“CIFAR”相关部分,严格按英文保留))

Yiqiu Shen, Nan Wu, Jason Phang, Jungkyu Park, Kangning Liu, Sudarshini Tyagi, Laura Heacock, S. Gene Kim, Linda Moy, Kyunghyun Cho, Krzysztof J. Geras

英文摘要:

Medical images differ from natural images in significantly higher resolutions and smaller regions of interest. Because of these differences, neural network architectures that work well for natural images might not be applicable to medical image analysis. In this work, we extend the globally-aware multiple instance classifier, a framework we proposed to address these unique properties of medical images. This model first uses a low-capacity, yet memory-efficient, network on the whole image to identify the most informative regions. It then applies another higher-capacity network to collect details from chosen regions. Finally, it employs a fusion module that aggregates global and local information to make a final prediction. While existing methods often require lesion segmentation during training, our model is trained with only image-level labels and can generate pixel-level saliency maps indicating possible malignant findings. We apply the model to screening mammography interpretation: predicting the presence or absence of benign and malignant lesions. On the NYU Breast Cancer Screening Dataset, consisting of more than one million images, our model achieves an AUC of 0.93 in classifying breasts with malignant findings, outperforming ResNet-34 and Faster R-CNN. Compared to ResNet-34, our model is 4.1x faster for inference while using 78.4% less GPU memory. Furthermore, we demonstrate, in a reader study, that our model surpasses radiologist-level AUC by a margin of 0.11. The proposed model is available online: https://github.com/nyukat/GMIC.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑