Deep Multi-scale Multi-instance Networks with Regional Scoring for Mammogram Classification

Wenjie Liu,Xin Shu,Lei Zhang,Dong Li,Qing Lv
DOI: https://doi.org/10.1109/tai.2021.3136146
2021-01-01
IEEE Transactions on Artificial Intelligence
Abstract:Mammography is one of the most commonly used methods for breast cancer screening. Most mammogram classification or segmentation models are trained with additional manual segmentation annotation data, which is difficult to obtain. Meanwhile, these models rarely take into account the fact that lesion features may vary at different scales. Moreover, most models have difficulty in quantifying the effect of single lesion regions on prediction results. To solve the above problems, we propose deep multiscale multi-instance networks with regional scoring. This model classifies mammograms automatically by image labels without using any other annotation information during the training procedure. The proposed multiscale structures include large scale, medium scale, and small scale, and they can extract features of mammograms from global to local. Because there may be more than one breast lesion region in a mammogram, the proposed multi-instance structure divides the image into multiple regions and selects few regions of interest to represent the whole image together. The regional scoring mechanism quantifies the contribution of each region to the prediction results, and thus, makes the results interpretable. In addition, the proposed method can roughly localize the lesion area by the score of each region. The experimental results on a publicly available dataset and a private dataset have shown that the proposed method achieved the highest AUC on both datasets compared with the state-of-the-art models.
What problem does this paper attempt to address?