A BERT encoding with Recurrent Neural Network and Long-Short Term Memory for breast cancer image classification
Sushovan Chaudhury, Kartik Sau
Computer-aided diagnostic systems developed with Artificial Intelligence (AI) are a major advancement in healthcare analytics, assisting radiologists with a second opinion on cancer diagnosis. In this study, we explore two modalities of breast images, ultrasound, and histology, for their classification into cancerous and non- cancerous categories. Traditionally Convolution Neural Networks (CNN) have done a commendable job of extracting features using convolution kernels from images with great accuracy. Also, the Bidirectional Encoder Representations from Transformers (BERT) has been widely used in Natural Language Processing (NLP) for feature encoding and downstream tasks like segmentation and classification. We extract the power of Vision Transformers (ViT) and implement transfer learning using BERT pre-training of Image Transformers (BEiT) as a feature encoding technique. We use the encoded features for classification with Recurrent Neural Network - Long short Term Memory (RNN-LSTM). The classification is performed on two modalities of breast image datasets: BUSI1311 and breast histopathological images. Both modalities yielded competitive accuracies. The BUSI1311 dataset produced 99 percent accuracy compared to 91 percent accuracy for breast histology images.
Year of publication: |
2023
|
---|---|
Authors: | Chaudhury, Sushovan ; Sau, Kartik |
Published in: |
Decision analytics journal. - Amsterdam : Elsevier, ISSN 2772-6622, ZDB-ID 3106160-6. - Vol. 6.2023, Art.-No. 100177, p. 1-15
|
Subject: | Neuronale Netze | Neural networks | Krebskrankheit | Cancer | Theorie | Theory |
Saved in:
freely available
Type of publication: | Article |
---|---|
Type of publication (narrower categories): | Aufsatz in Zeitschrift ; Article in journal |
Language: | English |
Notes: | Aufsatz zurückgezogen |
Other identifiers: | 10.1016/j.dajour.2023.100177 [DOI] |
Source: | ECONIS - Online Catalogue of the ZBW |
Persistent link: https://www.econbiz.de/10014506422
Saved in favorites
Similar items by subject
-
Supriya, M., (2020)
-
Hierarchical Bayesian neural networks : an application to a prostate cancer study
Ghosh, Malay, (2004)
-
A Review on Data Mining Techniques for Prediction of Breast Cancer Recurrence
PadmaPriya, R.S., (2020)
- More ...