Aller directement au contenu principal

Rédiger un PREreview

High-Performance Classification of Breast Cancer Histopathological Images Using Fine-Tuned Vision Transformers on the BreakHis Dataset

Publié
Serveur de preprints
bioRxiv
DOI
10.1101/2024.08.17.608410

Breast cancer remains one of the most prevalent and deadly cancers worldwide, making accurate diagnosis critical for effective treatment. Histopathological image classification is a key task in medical diagnostics for cancer detection. This paper presents state-of-the-art performance in histopathological image classification of breast cancer using a novel approach with the Vision Transformer (ViT) model fine-tuned using the BreakHis dataset. The BreakHis dataset, comprising of 7,909 breast cancer histopathological images across various magnification levels, serves as a crucial benchmark for evaluating machine learning models in this domain. While previous works have explored the use of ViTs for this task, our approach fine-tunes a ViT pre-trained on ImageNet using the Ranger optimizer, achieving unprecedented performance. The experimental results show that our fine-tuned ViT model achieves an accuracy of 99.99%, precision of 99.98%, recall of 99.99%, F1 score of 99.99%, specificity of 100.00%, false discovery rate (FDR) of 0.00%, false negative rate (FNR) of 0.02%, false positive rate (FPR) of 0.00%, Matthews correlation coefficient (MCC) of 99.97%, and negative predictive value (NPV) of 99.96%. This model and approach represents the highest accuracy ever achieved for BreakHis binary classification in machine learning using any model, underscoring the potential of Vision Transformers to substantially enhance diagnostic accuracy in histopathological image analysis and improve clinical outcomes. Transfer learning was also performed on the BACH and a histopathological image dataset for breast invasive ductal carcinomas (IDC).

Vous pouvez rédiger un PREreview de High-Performance Classification of Breast Cancer Histopathological Images Using Fine-Tuned Vision Transformers on the BreakHis Dataset. Un PREreview est une évaluation d'un preprint et peut varier de quelques phrases à un rapport détaillé, semblable à un rapport d'évaluation par les pairs organisé par une revue.

Avant de commencer

Nous vous demanderons de vous connecter avec votre identifiant ORCID iD. Si vous n'en avez pas, vous pouvez en créer un.

Qu’est-ce qu’un ORCID iD ?

Un ORCID iD est un identifiant unique qui vous distingue de toute personne ayant le même nom ou nom similaire.

Commencer maintenant