Back close

Unleashing the Power of Hierarchical Variational Autoencoder for Predicting Breast Cancer

Publication Type : Journal

Publisher : Institute of Electrical and Electronics Engineers (IEEE)

Source : IEEE Access

Url : https://doi.org/10.1109/access.2024.3518612

Keywords : Keywords assist with retrieval of results and provide a means to discovering other relevant content.

Campus : Kochi

School : School of Medicine

Department : Medical Oncology

Year : 2024

Abstract : Breast cancer continues to be a major health concern worldwide. Early and accurate prediction is crucial for effective treatment and improving survival rates. Computer Aided Diagnosis system serves as an invaluable tool for radiologists, aiming to reduce diagnostic errors and enhance the accuracy of diagnosis. These systems incorporate various processing techniques, including pre-processing, segmentation, feature extraction, and classification. Moreover, deep learning methods frequently suffer from sub optimal performance and demand substantial computational resources. This study focuses on developing an automated classification model for mammography images to aid in breast cancer diagnosis. Our proposed model initiates with noise removal using median filters, followed by the removal of the pectoral muscle in images through the Canny-edge detection method. On these preprocessed images, we applied data augmentation using a two-point crossover technique, addressing issues of small datasets and class imbalances common in medical image analysis. The images then undergo multi-scale representation via the fourth-order complex diffusion algorithm. Feature extraction is conducted on these multi-scaled images using a Hierarchical Variational Auto-encoders and then classified using a Support Vector Machine. Employing fourth-order complex diffusion for initial multi-scale representation significantly enhances the accuracy of feature extraction resulting in robust classification performance. The training process involves two different datasets like MIAS and the KAU-BCMD. Test results for the KAU-BCMD dataset include: accuracy of 99.80%, Area Under the Curve of 99.30%, F1-score of 99.20%, balanced accuracy of 99.80%, and Matthews correlation coefficient of 99.20%. For the MIAS dataset, test results show accuracy of 99.30%, Area Under the Curve of 99.10%, F1-score of 98.30%, balanced accuracy of 99.00%, and Matthews correlation coefficient of 99.00%. Our validation results clearl...

Cite this Research Publication : V. Sreelekshmi, K. Pavithran, Jyothisha J. Nair, Unleashing the Power of Hierarchical Variational Autoencoder for Predicting Breast Cancer, IEEE Access, Institute of Electrical and Electronics Engineers (IEEE), 2024, https://doi.org/10.1109/access.2024.3518612

Admissions Apply Now