Adversarial Hospital-Invariant Feature Learning for WSI Patch Classification

#### Контекст Патологические исследования стремятся к точным диагнозам и раннему выявлению заболеваний, чтобы снизить риск задержки лечения и улучшить прогноз болезни. Whole-slide image (WSI) анализ, позволяющий анализировать ткани на гранулярном уровне, становится важной частью этого процесса. Однако существуют проблемы, связанные с доменным биасом, который возникает из-за различий в сканировании и предобработке тканей в разных госпиталях. Это может привести к недостоверным выводам и ошибкам в классификации. Поэтому необходимо разработать методы, которые позволят извлечь общие особенности, независимые от конкретных госпиталей. #### Метод Мы предлагаем алгоритм, основанный на адверсарном обучении, для удаления домен-специфических признаков из представлений, полученных из модели, без изменения её основной архитектуры. Метод включает в себя следующие компоненты: 1. **Построение адаптера (adapter)**: задача этого модуля — удалять специфические признаки, связанные с определенным госпиталем, из выходных данных модели. 2. **Введение градиентного обратного классификатора (gradient reversal layer, GRL)**: этот классификатор вносит конфликт между адаптером и основной моделью, принуждая модель выводить более общие признаки, независимые от домена. 3. **Заморозка модели**: представления, полученные из замороженной модели, используются в качестве входных данных для адаптера. Это позволяет избежать переобучения и сохранить хорошую классификационную производительность. #### Результаты Мы проводили эксперименты на множестве патологических данных, в том числе данных из нескольких госпиталей. Наши результаты показали, что алгоритм существенно уменьшает домен-специфичность в представлениях, не ухудшая классификационную точность. Были проведены анализы того, как модель классифицирует разные госпитали, и графические представления признаков, показавшие, что модель более устойчива к разным стилям сканирования. #### Значимость Наш алгоритм может быть применен в различных сценариях, где необходимо обучить модели классификации, владеющие общими признаками, независимыми от госпиталя. Это может помочь улучшить надежность диагноза в разных клинических условиях. Кроме того, наш метод может быть применен для других видов изображений, где домен-специфичность может стать проблемой. #### Выводы Мы предложили новую методологию для обучения моделей классификации, которые извлекают общие признаки в патологической диагностике, независимо от разных госпиталей. Мы показали, что наш алгоритм эффективен в уменьшении домен-специфичности, не ухудшая классификационную точность. Будущие ис