arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.11726cs.CV
收藏

一种两阶段多实例学习框架在乳腺X线图像乳腺癌检测中的应用

A Two-Stage Multiple Instance Learning Framework for the Detection of Breast Cancer in Mammograms

Indian Institute of Technology Kharagpur(印度理工学院卡拉格普尔分校);Apollo Gleneagles Hospital(阿波罗格伦伊格尔斯医院);Intel Technology India Pvt. Ltd.(英特尔技术印度私人有限公司)

Sarath Chandra K, Arunava Chakravarty, Nirmalya Ghosh, Tandra Sarkar, Ramanathan Sethuraman, Debdoot Sheet

摘要:

乳房X线照片通常用于大规模筛查乳腺癌,其主要特征在于存在恶性肿块。然而,恶性肿瘤的自动图像级检测是一项具有挑战性的任务,因为肿块区域的尺寸很小,并且难以区分恶性、良性肿块和健康的致密纤维腺组织。为了解决这些问题,我们探索了一个两阶段的多实例学习(MIL)框架。在第一阶段中训练卷积神经网络(CNN)以在乳房X线照片中提取可能包含良性或恶性肿块的局部候选块。第二阶段采用MIL策略进行图像级良性与恶性分类。全局图像级特征被计算为使用CNN学习的块级特征的加权平均。我们的方法在定位肿块的任务上表现良好,平均精度/召回率为0.76/0.80,在INbreast数据集上使用五重交叉验证,在图像级分类任务上实现了0.91的平均AUC。将MIL仅限于在阶段1中提取的候选块,与从整个乳房X线照片中密集提取块相比,分类性能得到了显著改善。

英文摘要:

Mammograms are commonly employed in the large scale screening of breast cancer which is primarily characterized by the presence of malignant masses. However, automated image-level detection of malignancy is a challenging task given the small size of the mass regions and difficulty in discriminating between malignant, benign mass and healthy dense fibro-glandular tissue. To address these issues, we explore a two-stage Multiple Instance Learning (MIL) framework. A Convolutional Neural Network (CNN) is trained in the first stage to extract local candidate patches in the mammograms that may contain either a benign or malignant mass. The second stage employs a MIL strategy for an image level benign vs. malignant classification. A global image-level feature is computed as a weighted average of patch-level features learned using a CNN. Our method performed well on the task of localization of masses with an average Precision/Recall of 0.76/0.80 and acheived an average AUC of 0.91 on the imagelevel classification task using a five-fold cross-validation on the INbreast dataset. Restricting the MIL only to the candidate patches extracted in Stage 1 led to a significant improvement in classification performance in comparison to a dense extraction of patches from the entire mammogram.

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

↑