Interpretable deep learning approach for classification of breast cancer - a comparative analysis of multiple instance learning models - Publication - Bridge of Knowledge

Your account

login

Search

Interpretable deep learning approach for classification of breast cancer - a comparative analysis of multiple instance learning models

Abstract

Breast cancer is the most frequent female cancer. Its early diagnosis increases the chances of a complete cure for the patient. Suitably designed deep learning algorithms can be an excellent tool for quick screening analysis and support radiologists and oncologists in diagnosing breast cancer.The design of a deep learning-based system for automated breast cancer diagnosis is not easy due to the lack of annotated data, especially at pixel level, the large size of the images with relatively small cancer lesion sizes and class imbalance, a wide diversity of cancerous lesions, a variety of breasts, both in size and density, make the training of the neural models challenging. Moreover, clinicians are often concerned about using these black-box models because of the lack of transparency in their inference. To address these issues, we propose an approach taking advantage of Multiple Instance Learning (MIL), supported by attention mechanisms. We researched Attention-based MIL (AMIL), Gated AMIL (GAMIL), Dual Stream MIL (DSMIL) and CLustering-constrained AMIL (CLAM) models trained in a weakly-supervised manner and compared them with a common model in image classification tasks, ResNet18.The approach described in this paper is multimodal and combines two mammographic projections (CC and MLO) in the training process. The developed neural system achieved high classification efficiency. Furthermore, exploiting the generated attentional maps allowed the localisation of cancerous lesions, thus increasing the interpretability of the algorithm. Thanks to this mechanism, we were also able to detect artifacts in the analyzed database, difficult to spot but drastically skewing the algorithm’s performance.

Citations

  • 1

    CrossRef

  • 0

    Web of Science

  • 1

    Scopus

Authors (7)

Cite as

Full text

full text is not available in portal

Keywords

Details

Category:
Conference activity
Type:
publikacja w wydawnictwie zbiorowym recenzowanym (także w materiałach konferencyjnych)
Language:
English
Publication year:
2023
Bibliographic description:
Buler J., Buler R., Bobowicz M., Ferlin M., Rygusik M., Kwasigroch A., Grochowski M.: Interpretable deep learning approach for classification of breast cancer - a comparative analysis of multiple instance learning models// / : , 2023,
DOI:
Digital Object Identifier (open in new tab) 10.1109/mmar58394.2023.10242564
Sources of funding:
  • nie mam dostępu do tej informacji
Verified by:
Gdańsk University of Technology

seen 76 times

Recommended for you

Attention-Based Deep Learning System for Classification of Breast Lesions—Multimodal, Weakly Supervised Approach

2023

Deep CNN based decision support system for detection and assessing the stage of diabetic retinopathy

2018

Self-Supervised Learning to Increase the Performance of Skin Lesion Classification

2020

Breast MRI segmentation by deep learning: key gaps and challenges

2023
Meta Tags