MSI-UNet: A Flexible UNet-Based Multi-Scale Interactive Framework for 3D Gastric Tumor Segmentation on CT Scans

Heyun Chen,Zifan Chen,Jie Zhao,Haoshen Li,Jiazheng Li,Yiting Liu,Mingze Yuan,Peng Bao,Xinyu Nan,Bin Dong,Lei Tang,Li Zhang
DOI: https://doi.org/10.1109/isbi56570.2024.10635129
2024-01-01
Abstract:Accurate segmentation of gastric tumors is critical yet presents a formidable challenge in medical imaging, where conventional UNet-based frameworks, despite their prevalence, falter on intricate tumor samples due to their limited interactive capacities. The SAM-based segmentation methods address this shortcoming yet with insufficient accuracy. By ingeniously blending images with mask inputs, our MSI-UNet leverages a U-shaped design to deliver pixel-perfect segmentation accuracy, while a novel multi-scale attention module adeptly harnesses interaction points for refined information extraction. When benchmarked on gastric tumor segmentation tasks, MSI-UNet surpasses existing state-of-the-art methods, elevating the Dice Similarity Coefficient (DSC) from 74.82% to 79.3% and minimizing Average Surface Distance (ASD) from 6.46 to 1.98, achieving a comparable accuracy with inter-radiologist consistency of 79.7% DSC. Furthermore, our framework demonstrates superior predictive prowess in survival analysis, enhancing the C-index from 61.7% to 68.68%. Ample experimental comparisons have substantiated that MSI-UNet holds the potential to offer considerable assistance to healthcare professionals in managing and decoding subsequent medical procedures.
What problem does this paper attempt to address?