基于扩张和注意力引导残差学习的多视图乳腺密度分类。
Multi-View Mammographic Density Classification by Dilated and Attention-Guided Residual Learning.
出版信息
IEEE/ACM Trans Comput Biol Bioinform. 2021 May-Jun;18(3):1003-1013. doi: 10.1109/TCBB.2020.2970713. Epub 2021 Jun 3.
Breast density is widely adopted to reflect the likelihood of early breast cancer development. Existing methods of mammographic density classification either require steps of manual operations or achieve only moderate classification accuracy due to the limited model capacity. In this study, we present a radiomics approach based on dilated and attention-guided residual learning for the task of mammographic density classification. The proposed method was instantiated with two datasets, one clinical dataset and one publicly available dataset, and classification accuracies of 88.7 and 70.0 percent were obtained, respectively. Although the classification accuracy of the public dataset was lower than the clinical dataset, which was very likely related to the dataset size, our proposed model still achieved a better performance than the naive residual networks and several recently published deep learning-based approaches. Furthermore, we designed a multi-stream network architecture specifically targeting at analyzing the multi-view mammograms. Utilizing the clinical dataset, we validated that multi-view inputs were beneficial to the breast density classification task with an increase of at least 2.0 percent in accuracy and the different views lead to different model classification capacities. Our method has a great potential to be further developed and applied in computer-aided diagnosis systems. Our code is available at https://github.com/lich0031/Mammographic_Density_Classification.
乳房密度广泛用于反映早期乳腺癌发展的可能性。现有的乳腺密度分类方法要么需要手动操作步骤,要么由于模型容量有限,只能达到中等的分类准确性。在这项研究中,我们提出了一种基于扩张和注意引导残差学习的乳腺密度分类的放射组学方法。该方法使用两个数据集进行实例化,一个是临床数据集,一个是公开可用的数据集,分别获得了 88.7%和 70.0%的分类准确率。虽然公共数据集的分类准确率低于临床数据集,这很可能与数据集的大小有关,但我们提出的模型仍然比朴素残差网络和几个最近发表的基于深度学习的方法表现更好。此外,我们设计了一种多流网络架构,专门用于分析多视图乳房 X 光片。利用临床数据集,我们验证了多视图输入有助于提高乳腺密度分类任务的准确性,至少提高了 2.0%,并且不同的视图导致不同的模型分类能力。我们的方法具有很大的潜力,可以进一步开发和应用于计算机辅助诊断系统。我们的代码可在 https://github.com/lich0031/Mammographic_Density_Classification 上获得。