Normal and Atypical Mitosis Image Classifier using Efficient Vision Transformer

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qi, Xuan, Labella, Dominic, Sanford, Thomas, Lee, Maxwell
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914017458520064
author Qi, Xuan
Labella, Dominic
Sanford, Thomas
Lee, Maxwell
author_facet Qi, Xuan
Labella, Dominic
Sanford, Thomas
Lee, Maxwell
contents We tackle atypical versus normal mitosis classification in the MIDOG 2025 challenge using EfficientViT-L2, a hybrid CNN--ViT architecture optimized for accuracy and efficiency. A unified dataset of 13,938 nuclei from seven cancer types (MIDOG++ and AMi-Br) was used, with atypical mitoses comprising ~15. To assess domain generalization, we applied leave-one-cancer-type-out cross-validation with 5-fold ensembles, using stain-deconvolution for image augmentation. For challenge submissions, we trained an ensemble with the same 5-fold split but on all cancer types. In the preliminary evaluation phase, this model achieved balanced accuracy of 0.859, ROC AUC of 0.942, and raw accuracy of 0.85, demonstrating competitive and well-balanced performance across metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2509_02589
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Normal and Atypical Mitosis Image Classifier using Efficient Vision Transformer
Qi, Xuan
Labella, Dominic
Sanford, Thomas
Lee, Maxwell
Image and Video Processing
Artificial Intelligence
Computer Vision and Pattern Recognition
We tackle atypical versus normal mitosis classification in the MIDOG 2025 challenge using EfficientViT-L2, a hybrid CNN--ViT architecture optimized for accuracy and efficiency. A unified dataset of 13,938 nuclei from seven cancer types (MIDOG++ and AMi-Br) was used, with atypical mitoses comprising ~15. To assess domain generalization, we applied leave-one-cancer-type-out cross-validation with 5-fold ensembles, using stain-deconvolution for image augmentation. For challenge submissions, we trained an ensemble with the same 5-fold split but on all cancer types. In the preliminary evaluation phase, this model achieved balanced accuracy of 0.859, ROC AUC of 0.942, and raw accuracy of 0.85, demonstrating competitive and well-balanced performance across metrics.
title Normal and Atypical Mitosis Image Classifier using Efficient Vision Transformer
topic Image and Video Processing
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2509.02589