Deep Active Learning for Lung Disease Severity Classification from Chest X-rays: Learning with Less Data in the Presence of Class Imbalance

#### Контекст Ранний диагноз и ранняя стадия лечения заболеваний, в том числе легочных, является ключевым фактором успешного лечения и предотвращения серьезных осложнений. Одним из наиболее распространенных средств диагностики являются рентгеновые исследования легких (Chest X-rays, CXRs). Однако традиционный метод диагностики на основе ручного анализа CXRs требует опытных специалистов и значительного времени. Более того, многие больные не имеют доступа к такому диагностическому обслуживанию из-за нехватки квалифицированных специалистов. Дополнительно, классификация заболеваний по тяжести часто сталкивается с проблемой несбалансированных классов в данных, что создает дополнительные затруднения для обучения эффективных моделей. #### Метод В данном исследовании рассматривается deep active learning (ДАЛ) как метод для эффективного обучения моделей с меньшим количеством меток. Использовалась аппроксимация Bayesian Neural Network (BNN) для уменьшения вероятности overfitting и повышения уверенности в прогнозировании. Для обработки несбалансированных классов применялась весовая функция потерь. Модель обучалась с использованием Monte Carlo Dropout, чтобы уменьшить вероятность переобучения и повысить устойчивость модели к шуму. Для выбора самых информативных образцов из необученного корпуса использовались различные acquisition functions (например, Entropy Sampling и Mean STD Sampling). Экспериментальное исследование проводилось на 2319 CXRs от 963 пациентов с COVID-19, подвергшимися классификации по тяжести заболевания (нормальное состояние, умеренная и серьезная тяжесть). #### Результаты С помощью Entropy Sampling, модель с BNN и весовой функцией потерь показала 93.7% точности в бинарной классификации (нормальное состояние vs. заболевание), используя всего 15.4% от общего количества данных. В многоклассовой классификации (три класса тяжести), Mean STD Sampling достигла 70.3% точности, используя лишь 23.1% меток. Эти результаты были более эффективными по сравнению с более сложными и вычислительно трудозатратными методами, такими как BALD (Batch-Active Learning by Diverse Gradient Estimation). Был также отмечен значительный экономия во времени обучения и выборки, что открывает возможность применения этого подхода в реальном времени. #### Значимость Результаты исследования демонстрируют, что ДАЛ с BNN и весовой функцией потерь может эффективно использоваться для классификации тяжести легочных заболеваний, даже при несбалансированных классах. Это позволяет снизить требования к количеству меток, необходимых для обучения моделей, и улучшить скорость реагирования в критических ситуациях, таких как лечение COVID-19. Модель может