DoSReMC: Domain Shift Resilient Mammography Classification using Batch Normalization Adaptation

## Контекст Анализ и диагностика рака молочной железы чрезвычайно важны для раннего выявления и улучшения прогноза болезни. Искусственный интеллект (AI), особенно с помощью глубоких нейросетей, доминирует в автоматизированных системах раннего диагностирования. Однако эти системы часто сталкиваются с проблемой переобучения и плохой универсальности в разных клинических средах. Одна из причин этому является **domain shift**, то есть различия в распределениях данных между исходными и целевыми наборами данных. Данная проблема ограничивает внедрение AI в реальные клинические условия, где данные часто отличаются по распределению, калибровке и даже полу-отражению X-X-X-X. Наша цель заключается в разработке модели, которая была бы воспроизводима и устойчива к таким различиям. ## Метод Мы предлагаем **DoSReMC** (Domain Shift Resilient Mammography Classification), метод, который адаптирует **Batch Normalization (BN)**-слои без необходимости полной переучивания сети. Безусловно, BN-слои являются основным источником **domain dependence**, так как они развиваются во время обучения на определенной доменной подвыборке. Мы разработали алгоритм, который заново обучает только эти слои, задавая направление, призванное снизить различия между доменами. Для улучшения результатов, мы интегрировали **adversarial training**, чтобы дублировать выводы объектов из разных доменов. Мы применяем нашу модель к трем большим датасетам, включая новый клинический набор данных HCTP, полученный с патологически известными исходами. ## Результаты Мы проводили систематические эксперименты, сравнивая нашу модель с исходными сетями и другими методами адаптации. На HCTP-датасете, наша модель показала **8% увеличение точности** и **12% увеличение F1-меры** в сравнении с оригинальной сетью, которая страдала от domain shift. Также мы применили нашу модель к другим датасетам, получив улучшения в объёме от 5% до 10%. Основной сила нашего подхода заключается в том, что он позволяет **использовать предобученные модели** без трудоемкого переобучения в новых клинических условиях. ## Значимость Наш подход имеет большой потенциал в медицинских приложениях, где доступность и качество диагностики могут сильно зависеть от специфики данных. **DoSReMC** делает модели более универсальными, что позволяет их применять в различных клинических условиях. Этот подход может быть использован не только для классификации рака молочной железы но и для других задач раннего диагностирования в медицине. Мы считаем, что наша работа открывает путь к более **стабильным и эффективным системам** раннего диагностирования в медицине. ## Выводы Мы доказали, что **BN-адаптация** и **adversarial training**