Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

MitoDetect++: A Domain-Robust Pipeline for Mitosis Detection and Atypical Subtyping

Created by
  • Haebom

Author

Esha Sadia Nasir, Jiaqi Lv, Mostafa Jahanifar, Shan E Ahmed Raza

Outline

MitoDetect++ is an integrated deep learning pipeline for mitotic phase detection and atypical mitotic classification. Detection (Track 1) utilizes a U-Net-based encoder-decoder architecture with EfficientNetV2-L as a backbone and an attention module, trained using a joint segmentation loss. Classification (Track 2) utilizes a Virchow2 vision transformer, fine-tuned efficiently using Low-Rank Adaptation (LoRA), to minimize resource consumption. It integrates powerful augmentation, focal loss, and group-aware hierarchical 5-fold cross-validation to improve generalization performance and mitigate domain shift. Test-Time Augmentation (TTA) is deployed at inference time to enhance robustness. It achieves a balanced accuracy of 0.892 on the validation domain, highlighting its clinical applicability and cross-task scalability.

Takeaways, Limitations

Takeaways:
An efficient mitosis detection and classification pipeline combining U-Net and Vision Transformer is presented.
Efficient resource utilization and improved generalization performance through powerful augmentation techniques using LoRA.
High validation accuracy (0.892) suggests clinical applicability.
Demonstrates scalability for various tasks (detection and classification).
Limitations:
Validation against actual clinical data may be lacking.
There is a possibility of overfitting to certain datasets.
Generalization performance evaluation for various types of atypical mitoses is needed.
Further research is needed on the generalization performance of fine-tuning using LoRA.
👍