Adaptive Learning Strategies for Mitotic Figure Classification in MIDOG2025 Challenge

2509.02640v1 eess.IV, cs.AI, cs.CV 2025-09-05
Авторы:

Biwen Meng, Xi Long, Jingxin Liu

Резюме на русском

## Контекст Атипичные митотические фигуры (AMFs) являются клинически значимыми признаками аномального разделения клеток, но их надежное выявление остается сложной задачей. Это связано с морфологической амбигуитей и различиями в источниках сканирования. Несмотря на развитие методологий машинного обучения, обеспечивающих высокую точность при работе с типичными митозами, стандартные модели часто сталкиваются с проблемами при работе с AMFs из-за их нарушенной структуры и мелких деталей. Данная работа сосредоточена на изучении и применении адаптивных стратегий обучения для улучшения классификации AMFs в рамках челленджа MIDOG2025 Track 2, направленного на решение этой проблемы. ## Метод Мы рассмотрели три варианта адаптации патологической базисной модели UNI2-h для решения задачи классификации AMFs. Начальный подход основывался на LoRA (Low-Rank Adaptation), позволяющем адаптировать модель к новым данным. Далее, мы исследовали визуальный тюнинг промптов (Visual Prompt Tuning, VPT), который позволяет изменять представления входных данных во время обучения, чтобы повысить общий потенциал модели. Наконец, мы оптимизировали тест-тайм аугментацию (TTA), используя методы нормализации цвета стемлов Vahadane и Macenko. Эти техники были интегрированы вместе, чтобы улучшить устойчивость модели к различным условиям сканирования и морфологическим особенностям AMFs. ## Результаты Наши эксперименты проводились на данных MIDOG2025 Track 2. Мы сравнили результаты базовой модели LoRA с адаптивными модификациями, включая VPT и TTA с нормализацией цвета. Модель с VPT показала значительное улучшение в общей точности и подавлении ложноположительных выявлений. Добавление TTA с нормализацией цвета далее повысило устойчивость модели к различным условиям. В результате, наша модель достигла балансированной точности 0.8837 и ROC-AUC 0.9513 на предварительном лидерборде, что расположило нашу команду в пределах топ-10. ## Значимость Наше исследование имеет значительное значение для клинической практики, где точное выявление AMFs критично для оценки риска рака. Методы, предложенные в этой работе, могут быть применены для классификации других нестандартных клетки в патологических изображениях. Интеграция промптов и нормализации цвета может быть решающей для решения проблем с производительностью моделей в условиях сильной и нестандартной дисперсии данных. Эти находки могут быть распространены на другие области медицинского машинного обучения, где обработка и классификация изображений требуют высокой точности и устойчивости. ## Выводы Мы проанализировали адаптивные стратегии обучения

Abstract

Atypical mitotic figures (AMFs) are clinically relevant indicators of abnormal cell division, yet their reliable detection remains challenging due to morphological ambiguity and scanner variability. In this work, we investigated three variants of adapting the pathology foundation model UNI2-h for the MIDOG2025 Track 2 challenge. Starting from a LoRA-based baseline, we found that visual prompt tuning (VPT) substantially improved generalization, and that further integrating test-time augmentation (TTA) with Vahadane and Macenko stain normalization provided the best robustness. Our final submission achieved a balanced accuracy of 0.8837 and an ROC-AUC of 0.9513 on the preliminary leaderboard, ranking within the top 10 teams. These results demonstrate that prompt-based adaptation combined with stain-normalization TTA offers an effective strategy for atypical mitosis classification under diverse imaging conditions.

Ссылки и действия