Diffusion-Based Data Augmentation for Medical Image Segmentation

## Контекст В области медицинской информатики становится все более актуальной задача автоматизации сегментации медицинских изображений для обнаружения и исследования различных заболеваний. Одним из основных вызовов в этой области является нехватка больших количеств качественно подготовленных данных, особенно для редких и характеризующихся необычными образованиями. Эта проблема снижает эффективность существующих медицинских моделей сегментации и приводит к ухудшению результатов диагностики. Основная мотивация заключается в развитии методов, позволяющих сгенерировать реалистичные данные для обучения моделей, особенно для редких аномалий, которые требуют точного распознавания и диагностики. ## Метод Предлагаемый подход, DiffAug, заключается в сочетании текстово-управляемого диффузионного подхода с автоматической валидацией сегментации. Основной компонентой этого метода является условное диффузионное моделирование, которое используется для генерации аномалий в медицинских изображениях. Используя текстовые справочные сигналы, полученные из медицинской литературы, а также пространственные маски, DiffAug генерирует реалистичные образцы аномалий, добавляя их на нормальные изображения через процесс inpainting. Для обеспечения высокого качества результатов вводится валидационный механизм, основанный на латентном пространстве, который динамически проверяет сгенерированные образцы на точность и смысловую соответствие. Данный метод оптимизирует процесс генерации и валидации сегментации в одном проходе, что позволяет эффективно использовать ресурсы и повысить точность. ## Результаты На трех медицинских бенчмарках (CVC-ClinicDB, Kvasir-SEG, REFUGE2) проведены эксперименты, демонстрирующие высокую эффективность DiffAug по сравнению с традиционными методами. Модель показала увеличение Dice-метрики на 8-10%, что значительно превосходит существующие алгоритмы. Более того, DiffAug существенно сократил false negative rate на 28%, что является критически важной особенностью для диагностики таких заболеваний, как малые полипы или незначительные поверхностные опухоли. Эти результаты указывают на то, что DiffAug не только улучшает качество сегментации, но и повышает способность моделей распознавать редкие и сложные аномалии. ## Значимость DiffAug может применяться в различных областях, включая раннюю диагностику, медицинские исследования и обучение моделей сегментации. Одним из основных преимуществ является увеличение точности распознавания редких и необычных заболеваний, что имеет практическое значение для раннего выявления и профилактики. Этот подход также обесп