Learning multi-frequency features in convolutional network for mammography classification

Abstract

Breast cancer is a common life-threatening disease among women. Computer-aided methods can provide second opinion or decision support for early diagnosis in mammography images. However, the whole images classification is highly challenging due to small sizes of lesion and slow contrast between lesions and fibro-glandular tissue. In this paper, inspired by conventional machine learning methods, we present a Multi Frequency Attention Network (MFA-Net) to highlight the salient features. The network decomposes the features into low spatial frequency components and high spatial frequency components, and then recalibrates discriminating features based on two-dimensional Discrete Cosine Transform in two different frequency parts separately. Low spatial frequency features help determine if there is a tumor while high spatial frequency features help focus more on the margin of the tumor. Our studies empirically show that compared to traditional convolutional neural network (CNN), the proposed method mitigates the influence of the margin of pectoral muscle and breast in mammography, which brings significant improvement. For malignant and benign classification, by using transfer learning, the proposed MFA-Net achieves the AUC index 91.71% on the INbreast dataset.

Keywords: Breast cancer diagnosis; Classification; Deep learning; Mammography.