Advancing Multi-Organ and Pan-Cancer Segmentation in Abdominal CT Scans through Scale-Aware and Self-Attentive Modulation

09 Sept 2023 (modified: 22 Dec 2023)Submitted to FLARE 2023EveryoneRevisionsBibTeX
Keywords: Multi-organ and pan-cancer segmentation, Hybrid CNN-Transformer model, Scale-aware and self-attention modulation
Abstract: Accurately segmenting abdominal organs and tumors within computed tomography (CT) scans holds paramount significance for facilitating computer-aided diagnosis and devising treatment plans. However, inherent challenges such as lesion heterogeneity and the scarcity of adequately annotated data hamper model development. In this study, we present a two-phase cascaded framework to address the complexities of multi-organ and pan-cancer segmentation. A lightweight CNN first generates candidate regions of interest (ROIs) followed by a hybrid CNNTransformer model culminating in refined segmentation by synergizing scale-aware modulation for local features and self-attention for global context. Our proposed method secured the 5th position in the MICCAI FLARE23 final test set, showcasing its competitive edge in achieving precise target segmentation, with mean Dice Similarity Coefficients of 90.51% for multi-organ and 53.04% for pan-cancer respectively. Additionally, efficient inference is exhibited with an average runtime of 18 seconds per 512 × 512 × 215 3D volume with less than 2G GPU memory consumption.
Supplementary Material: zip
Submission Number: 8
Loading