ConvNeXt Model for Breast Cancer Image Classification
Devin Setiawan, Andrea Stevens Karnyoto, Indo Intan, Bens Pardamean · 2024
Breast cancer is an occurrence of cancer that attacks breast tissue and is the most common cancer among women worldwide, affecting one in eight women. In this modern world, breast cancer image classification simplifies the process of analyzing, providing objective and accurate results. By leveraging machine learning algorithms and computer vision techniques, we developed breast cancer detection. The dataset is histopathology dataset from BreakHis and UNHAS Hospital. We chose the ConvNeXt-Tiny model then modified the classifier head as the proposed method. Before the dataset is processed by the model, we augment the images by applying random horizontal and vertical flips, adjustments to brightness, contrast, saturation, and hue using color jitter. The augmentation process simulates real-world variance and enhances the model's ability to generalize to unseen data. Our proposed model gained better performance (accuracy, F1-Score) results compared two other techniques to VGG16 and SVM. According to our experiments, the F1-Score for the ConvNeXt-Tiny model with classifier head modification is higher at 0.9516, than the gain for VGG16 at 0.9292, and the gain for the SVM at 0.83.