GRASPing Anatomy to Improve Pathology Segmentation

2508.03374v1 cs.CV 2025-08-09
Авторы:

Keyi Li, Alexander Jaus, Jens Kleesiek, Rainer Stiefelhagen

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

Текущие методы глубокого обучения для сегментации заболеваний часто ориентируются только на паттерны, игнорируя анатомический контекст, который критичен для точного определения патологий. Для решения этой проблемы предложена модель GRASP (Guided Representation Alignment for the Segmentation of Pathologies), гибкий модульный подход, который улучшает модели сегментации заболеваний, используя существующие модели сегментации анатомии. Методом GRASP включается интеграция псевдометок анатомии и алгоритмов переноса признаков, что позволяет модели без дополнительного тренирования анатомических моделей получать знания об анатомии. Это улучшает точность сегментации. Результаты исследований показывают, что GRASP показывает высокую эффективность на нескольких датасетах, а его двухуровневая стратегия внедрения анатомического контекста с помощью псевдометок и трансформеров демонстрирует эффективность в интеграции анатомического контекста в модели.

Abstract

Radiologists rely on anatomical understanding to accurately delineate pathologies, yet most current deep learning approaches use pure pattern recognition and ignore the anatomical context in which pathologies develop. To narrow this gap, we introduce GRASP (Guided Representation Alignment for the Segmentation of Pathologies), a modular plug-and-play framework that enhances pathology segmentation models by leveraging existing anatomy segmentation models through pseudolabel integration and feature alignment. Unlike previous approaches that obtain anatomical knowledge via auxiliary training, GRASP integrates into standard pathology optimization regimes without retraining anatomical components. We evaluate GRASP on two PET/CT datasets, conduct systematic ablation studies, and investigate the framework's inner workings. We find that GRASP consistently achieves top rankings across multiple evaluation metrics and diverse architectures. The framework's dual anatomy injection strategy, combining anatomical pseudo-labels as input channels with transformer-guided anatomical feature fusion, effectively incorporates anatomical context.

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