Enhancing the Breast Histopathology Image Analysis for Cancer Detection Using Variational Autoencoder

Int J Environ Res Public Health. 2023 Feb 27;20(5):4244. doi: 10.3390/ijerph20054244.

Abstract

A breast tissue biopsy is performed to identify the nature of a tumour, as it can be either cancerous or benign. The first implementations involved the use of machine learning algorithms. Random Forest and Support Vector Machine (SVM) were used to classify the input histopathological images into whether they were cancerous or non-cancerous. The implementations continued to provide promising results, and then Artificial Neural Networks (ANNs) were applied for this purpose. We propose an approach for reconstructing the images using a Variational Autoencoder (VAE) and the Denoising Variational Autoencoder (DVAE) and then use a Convolutional Neural Network (CNN) model. Afterwards, we predicted whether the input image was cancerous or non-cancerous. Our implementation provides predictions with 73% accuracy, which is greater than the results produced by our custom-built CNN on our dataset. The proposed architecture will prove to be a new field of research and a new area to be explored in the field of computer vision using CNN and Generative Modelling since it incorporates reconstructions of the original input images and provides predictions on them thereafter.

Keywords: deep learning; histopathology image; variational autoencoder.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Image Processing, Computer-Assisted / methods
  • Machine Learning
  • Neoplasms*
  • Neural Networks, Computer*

Grants and funding

This research was funded by Nourah bint Abdulrahman University Researchers Supporting Project number (PNURSP2023R178), Princess Nourah bint Abdulrahman University, Riyadh, Saudi Arabia.