Abstract:Precise breast cancer classification on histopathological images has the potential to greatly improve the diagnosis and patient outcome in oncology. The data imbalance problem largely stems from the inherent imbalance within medical image datasets, where certain tumor subtypes may appear much less frequently. This constitutes a considerable limitation in biased model predictions that can overlook critical but rare classes. In this work, we adopted EfficientNet, a state-of-the-art convolutional neural network (CNN) model that balances high accuracy with computational cost efficiency. To address data imbalance, we introduce an intensive data augmentation pipeline and cost-sensitive learning, improving representation and ensuring that the model does not overly favor majority classes. This approach provides the ability to learn effectively from rare tumor types, improving its robustness. Additionally, we fine-tuned the model using transfer learning, where weights in the beginning trained on a binary classification task were adopted to multi-class classification, improving the capability to detect complex patterns within the BreakHis dataset. Our results underscore significant improvements in the binary classification performance, achieving an exceptional recall increase for benign cases from 0.92 to 0.95, alongside an accuracy enhancement from 97.35 % to 98.23%. Our approach improved the performance of multi-class tasks from 91.27% with regular augmentation to 94.54% with intensive augmentation, reaching 95.04% with transfer learning. This framework demonstrated substantial gains in precision in the minority classes, such as Mucinous carcinoma and Papillary carcinoma, while maintaining high recall consistently across these critical subtypes, as further confirmed by confusion matrix analysis.

Image Classification in Breast Histopathology Using Transfer and Ensemble Learning

Application of Transfer Learning and Ensemble Learning in Image-level Classification for Breast Histopathology

Breast Cancer Histopathology Image Classification Using an Ensemble of Deep Learning Models

Ensemble Deep Learning-Based Image Classification for Breast Cancer Subtype and Invasiveness Diagnosis from Whole Slide Image Histopathology

Ensemble of Deep Features for Breast Cancer Histopathological Image Classification

Breast cancer pathological image classification based on deep learning

Classifying breast cancer using transfer learning models based on histopathological images

Breast Cancer Image Classification Method Based on Deep Transfer Learning

Histopathological image analysis for breast cancer diagnosis by ensembles of convolutional neural networks and genetic algorithms

A New Transfer Learning Based Approach to Magnification Dependent and Independent Classification of Breast Cancer in Histopathological Images

Breast Cancer Diagnosis from Histopathological Image Based on Deep Learning

Equilibrium Optimization-Based Ensemble CNN Framework for Breast Cancer Multiclass Classification Using Histopathological Image

Ensemble approach of deep learning models for binary and multiclass classification of histopathological images for breast cancer

Breast Cancer Multi-classification from Histopathological Images with Structured Deep Learning Model

Breast Cancer Histology Image Classification Based on Deep Neural Networks

Classifying Breast Cancer Histopathological Images Using a Robust Artificial Neural Network Architecture.

Improved Breast Cancer Classification through Combining Transfer Learning and Attention Mechanism

Label-Efficient Breast Cancer Histopathological Image Classification

Hybrid Deep Transfer Network and Rotational Sample Subspace Ensemble Learning for Early Cancer Detection

Breast Tumor Classification Using EfficientNet Deep Learning Model

EMS-Net: Ensemble of Multiscale Convolutional Neural Networks for Classification of Breast Cancer Histology Images.