Prostate Capsule Segmentation from Micro-Ultrasound Images using Adaptive Focal Loss

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

Kaniz Fatema, Vaibhav Thakur, Emad A. Mohammed

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

## Контекст Микро-ультразвук (мк-ультразвук) является продвинутой методологией в области обнаружения рака и компьютерного визуализационного анализа. Одной из сложностей, с которой сталкивается данная область, является точная сегментация простатной капсулы из микро-ультразвуковых изображений. Эта задача становится вызовом из-за неоднозначности границ простатной капсулы, что приводит к неточностям в медицинских выводах. Существующие методы часто сталкиваются с ограничениями в учете этих неоднозначностей, что создает мотивацию для разработки более точных и гибких методов. В данной работе предлагается новая адаптивная фокусная функция потерь, которая динамически адаптируется к трудностям сегментации, учитывая индивидуальные особенности задачи. ## Метод На основе стандартной фокусной функции потерь разработана адаптивная функция потерь, которая включает в себя два основных подхода. Первый подход заключается в использовании стандартной фокусной функции потерь, которая стабилизирует модель и обеспечивает базовый уровень точности. Второй подход включает в себя адаптивные изменения функции потерь, которые акцентуруются на трудных областях, таких как нечеткие границы простатной капсулы. Для этого вводится гибкое динамическое дофильтрование, которое учитывает отклонения между анализами экспертов и неэкспертов. Это позволяет модели адаптироваться к неоднозначным областям и динамически корректировать веса сегментационной модели. ## Результаты Проведенные эксперименты показали, что предлагаемая модель демонстрирует выдающиеся результаты. Метрика Dice Coefficient (DSC) достигла значения 0.940, а метрика Hausdorff Distance (HD) составила 1.949 мм на тестовой выборке. Эти результаты указывают на высокую точность и эффективность использования адаптивных фокусных функций потерь. Такой подход позволяет модели более точно определять нечеткие и трудные области простатной капсулы, что имеет значительное значение для повышения точности в диагностике и планировании лечения рака простаты. ## Значимость Разработанная методика имеет большое значение для как медицинских, так и компьютерных приложений. Она может быть применена в диагностике рака простаты, планировании хирургических операций и мониторинге терапии. Ее основное преимущество заключается в том, что она динамически адаптируется к сложностям сегментации, обеспечивая более точные и надежные результаты. Это может существенно повлиять на прогнозирование и лечение рака, повышая точность и эффективность клинических решений. ## Выводы Результаты проведенных исследований подтверждают эффектив

Abstract

Micro-ultrasound (micro-US) is a promising imaging technique for cancer detection and computer-assisted visualization. This study investigates prostate capsule segmentation using deep learning techniques from micro-US images, addressing the challenges posed by the ambiguous boundaries of the prostate capsule. Existing methods often struggle in such cases, motivating the development of a tailored approach. This study introduces an adaptive focal loss function that dynamically emphasizes both hard and easy regions, taking into account their respective difficulty levels and annotation variability. The proposed methodology has two primary strategies: integrating a standard focal loss function as a baseline to design an adaptive focal loss function for proper prostate capsule segmentation. The focal loss baseline provides a robust foundation, incorporating class balancing and focusing on examples that are difficult to classify. The adaptive focal loss offers additional flexibility, addressing the fuzzy region of the prostate capsule and annotation variability by dilating the hard regions identified through discrepancies between expert and non-expert annotations. The proposed method dynamically adjusts the segmentation model's weights better to identify the fuzzy regions of the prostate capsule. The proposed adaptive focal loss function demonstrates superior performance, achieving a mean dice coefficient (DSC) of 0.940 and a mean Hausdorff distance (HD) of 1.949 mm in the testing dataset. These results highlight the effectiveness of integrating advanced loss functions and adaptive techniques into deep learning models. This enhances the accuracy of prostate capsule segmentation in micro-US images, offering the potential to improve clinical decision-making in prostate cancer diagnosis and treatment planning.

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