Prompt-Guided Patch UNet-VAE with Adversarial Supervision for Adrenal Gland Segmentation in Computed Tomography Medical Images

#### Контекст Определение изображений небольших и неоднородных органов, таких как адренальные железы, в рамках томографических медицинских изображений, представляет собой сложную задачу в силу нескольких факторов. Эти факторы включают классическую несбалансированность классов, недостаточную детализацию пространственного контекста и недостаточную информативность аннотированных данных. Эти ограничения приводят к проблемам в получении точных сегментационных моделей. В этой работе предлагается инновационный подход, который объединяет в себе составляющие вариационной реконструкции, направленной на повышение качества изображений, супервизированную сегментацию и адверсарные техники с использованием патчей для улучшения обобщаемости и точности изображений. Модель включает в себя сочетание технологий вариационного аппарата с техникой UNet, чтобы обеспечить тщательную реконструкцию изображений и получение точных сегментационных масок. #### Метод Ключевым элементом этой модели является использование VAE-UNet структуры, которая занимается объединением двух задач: реконструкции входных патчей и построения сегментационных масок на уровне возвратных воздействий. Данный подход позволяет модели учиться различать анатомические структуры и их аппаратурные особенности. В дополнение, проводится оптимизация тренировочного процесса с помощью инновационной тренировочной схемы, в которой вводятся синтетические патчи, генерируемые из внутреннего латентного пространства модели. Эта схема позволяет модели более устойчиво реагировать на трудности классификации в классических зонах сильной неоднородности. Более того, применяется слой адверсарной супервизии с использованием PatchGAN-стиля, чтобы оценивать реалистичность изображений на уровне патчей. Для повышения точности восприятия изображений используется особый вид реконструкционной функции потерь, основанный на оценках VGG-сети. #### Результаты Эксперименты проводились на данных BTCV, богатых вариациями вида томографических снимков. Модель показала существенное повышение точности сегментации, особенно в зонах, требующих высокой точности, таких как края органов. За счет интеграции генеративно-дискриминативного подхода удалось добиться более высокого качества реконструкции изображений и увеличить выносливость модели в отношении данных, ограниченных аннотациями. Особое внимание уделено изучению влияния разного соотношения синтетических и настоящих патчей в процессе обучения. Это позволило оптимизировать баланс между реалистичностью, разнообразием и анатомической консистентность