IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i15p2354-d1444610.html
   My bibliography  Save this article

An Interpretable Breast Ultrasound Image Classification Algorithm Based on Convolutional Neural Network and Transformer

Author

Listed:
  • Xiangjia Meng

    (School of Information Engineering, Shandong Youth University of Political Science, Jinan 250103, China
    New Technology Research and Development Center of Intelligent Information Controlling in Universities of Shandong, Shandong Youth University of Political Science, Jinan 250103, China)

  • Jun Ma

    (School of Cyber Science and Engineering, Southeast University, Nanjing 219302, China)

  • Feng Liu

    (School of Information Engineering, Shandong Youth University of Political Science, Jinan 250103, China
    New Technology Research and Development Center of Intelligent Information Controlling in Universities of Shandong, Shandong Youth University of Political Science, Jinan 250103, China)

  • Zhihua Chen

    (School of Information Engineering, Shandong Youth University of Political Science, Jinan 250103, China
    New Technology Research and Development Center of Intelligent Information Controlling in Universities of Shandong, Shandong Youth University of Political Science, Jinan 250103, China)

  • Tingting Zhang

    (School of Information Engineering, Shandong Youth University of Political Science, Jinan 250103, China
    New Technology Research and Development Center of Intelligent Information Controlling in Universities of Shandong, Shandong Youth University of Political Science, Jinan 250103, China)

Abstract

Breast cancer is one of the most common causes of death in women. Early signs of breast cancer can be an abnormality depicted on breast images like breast ultrasonography. Unfortunately, ultrasound images contain a lot of noise, which greatly increases the difficulty for doctors to interpret them. In recent years, computer-aided diagnosis (CAD) has been widely used in medical images, reducing the workload of doctors and the probability of misdiagnosis. However, it still faces the following challenges in clinical practice: one is the lack of interpretability, and another is that the accuracy is not high enough. In this paper, we propose a classification model of breast ultrasound images that leverages tumor boundaries as prior knowledge and strengthens the model to guide classification. Furthermore, we employ the advantages of convolutional neural network (CNN) to extract local features and Transformer to extract global features to achieve information balance and complementarity between the two neural network models which increase the recognition performance of the model. Additionally, an explanation method is used to generate visual results, thereby improving the poor interpretability of deep learning models. Finally, we evaluate the model on the BUSI dataset and compare it with other CNN and Transformer models. Experimental results show that the proposed model obtains an accuracy of 0.9870 and an F1 score of 0.9872, achieving state-of-the-art performance.

Suggested Citation

  • Xiangjia Meng & Jun Ma & Feng Liu & Zhihua Chen & Tingting Zhang, 2024. "An Interpretable Breast Ultrasound Image Classification Algorithm Based on Convolutional Neural Network and Transformer," Mathematics, MDPI, vol. 12(15), pages 1-14, July.
  • Handle: RePEc:gam:jmathe:v:12:y:2024:i:15:p:2354-:d:1444610
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/15/2354/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/12/15/2354/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:15:p:2354-:d:1444610. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.