Two-view attention-guided convolutional neural network for mammographic image classificationShow others and affiliations
2023 (English)In: CAAI Transactions on Intelligence Technology, ISSN 2468-6557, Vol. 8, no 2, p. 453-467Article in journal (Refereed) Published
Abstract [en]
Deep learning has been widely used in the field of mammographic image classification owing to its superiority in automatic feature extraction. However, general deep learning models cannot achieve very satisfactory classification results on mammographic images because these models are not specifically designed for mammographic images and do not take the specific traits of these images into account. To exploit the essential discriminant information of mammographic images, we propose a novel classification method based on a convolutional neural network. Specifically, the proposed method designs two branches to extract the discriminative features from mammographic images from the mediolateral oblique and craniocaudal (CC) mammographic views. The features extracted from the two-view mammographic images contain complementary information that enables breast cancer to be more easily distinguished. Moreover, the attention block is introduced to capture the channel-wise information by adjusting the weight of each feature map, which is beneficial to emphasising the important features of mammographic images. Furthermore, we add a penalty term based on the fuzzy cluster algorithm to the cross-entropy function, which improves the generalisation ability of the classification model by maximising the interclass distance and minimising the intraclass distance of the samples. The experimental results on The Digital database for Screening Mammography INbreast and MIAS mammography databases illustrate that the proposed method achieves the best classification performance and is more robust than the compared state-of-the-art classification methods.
Place, publisher, year, edition, pages
Institution of Engineering and Technology (IET) , 2023. Vol. 8, no 2, p. 453-467
Keywords [en]
convolutional neural network, deep learning, mammographic image, medical image processing, Classification (of information), Computer aided diagnosis, Convolution, Convolutional neural networks, Image classification, X ray screens, Automatic feature extraction, Classification methods, Classification results, Images classification, Learning models, Mammographic images, Medical images processing, Two views, Mammography
National Category
Medical Imaging Computer graphics and computer vision
Identifiers
URN: urn:nbn:se:kth:diva-323278DOI: 10.1049/cit2.12096ISI: 000784624100001Scopus ID: 2-s2.0-85128516179OAI: oai:DiVA.org:kth-323278DiVA, id: diva2:1730274
Note
QC 20250512
2023-01-242023-01-242025-05-12Bibliographically approved