Uncertainty-Gated Deformable Network for Breast Tumor Segmentation in MR Images

2509.15758v1 eess.IV, cs.CV 2025-09-23
Авторы:

Yue Zhang, Jiahua Dong, Chengtao Peng, Qiuli Wang, Dan Song, Guiduo Duan

Резюме на русском

## Контекст Сегментация маститных опухолей на магнитных резонансных изображениях (MRI) является критически важной задачей в диагностике рака груди. Однако, существующие методы сталкиваются с рядом проблем, в том числе неэффективностью в адаптивной моделировании неоднородных форм опухолей и недостаточностью интеграции локальных и глобальных признаков. Эти ограничения приводят к неточностям в области сегментации, что может повлиять на качество диагностики. Наша цель заключается в разработке метода, который мог бы эффективно обрабатывать эти проблемы, обеспечивая точную сегментацию. ## Метод Мы предлагаем **Uncertainty-Gated Deformable Network (UGDN)**, который объединяет преимущества методов CNN и Transformer. Наш подход включает моделирование деформации в обеих сетях, что позволяет адаптивно регулировать режимы приёма при помощи деформативных модулей. Мы также представляем **Uncertainty-Gated Enhancing Module (U-GEM)**, которое адаптивно обменивается признаками между CNN и Transformer на основе пиксельной неопределённости. Для улучшения определения краев опухоли мы вводим Boundary-Sensitive Deep Supervision Loss. Это позволяет улучшить точность разметки и обеспечить более точное детектирование краев опухоли. ## Результаты Мы проводили эксперименты на двух клинических датасетах MRI, полученных у пациентов с раком груди. Наши результаты показали, что UGDN превосходит современные методы сегментации по метрикам Dice Similarity Coefficient (DSC) и Intersection over Union (IoU). Эксперименты подтвердили эффективность метода в моделировании неоднородных форм опухолей и интеграции локальных и глобальных признаков. Мы также проанализировали неопределённость в сегментации, показав, что U-GEM позволяет выбирать оптимальные компоненты из CNN и Transformer для каждого пикселя. ## Значимость Метод UGDN может быть применён в различных областях медицины, где требуется точная сегментация медицинских изображений. Основные преимущества включают в себя улучшенную точность сегментации, универсальность и гибкость в моделировании неоднородных форм опухолей. Мы предвидим, что этот подход может иметь потенциал для улучшения качества диагностики и лечения рака груди, повышая роль машинного обучения в этой сфере. ## Выводы Мы представили новую сеть, которая эффективно обрабатывает проблемы сегментации маститных опухолей на MRI. Наши результаты показали, что UGDN превосходит существующие методы, обеспечивая точную сегментацию и имея потенциал для клинического применения. Мы планируем расширить нашу работу, включив более широкий спектр вариантов заболеваний и других медицинских изображений.

Abstract

Accurate segmentation of breast tumors in magnetic resonance images (MRI) is essential for breast cancer diagnosis, yet existing methods face challenges in capturing irregular tumor shapes and effectively integrating local and global features. To address these limitations, we propose an uncertainty-gated deformable network to leverage the complementary information from CNN and Transformers. Specifically, we incorporates deformable feature modeling into both convolution and attention modules, enabling adaptive receptive fields for irregular tumor contours. We also design an Uncertainty-Gated Enhancing Module (U-GEM) to selectively exchange complementary features between CNN and Transformer based on pixel-wise uncertainty, enhancing both local and global representations. Additionally, a Boundary-sensitive Deep Supervision Loss is introduced to further improve tumor boundary delineation. Comprehensive experiments on two clinical breast MRI datasets demonstrate that our method achieves superior segmentation performance compared with state-of-the-art methods, highlighting its clinical potential for accurate breast tumor delineation.

Ссылки и действия