CED-Diff: a clinically-experience-driven diffusion model with differentiated feature extraction for head and neck tumour PET/CT segmentation.
Authors
Affiliations (3)
Affiliations (3)
- School of Software, Shenyang University of Technology, No. 111, Shenliao Road, Shenyang Economic and Technological Development Zone, Shenyang, 110870, China.
- Shenyang University of Technology, ., Shenyang, 110870, China.
- Ritsumeikan University, Ritsumeikan University, Kyoto, 603-8577, Japan.
Abstract
Automatic segmentation of head and neck tumours from PET/CT images is important for radiotherapy planning and related image-guided workflows, but remains challenging because PET and CT provide complementary yet heterogeneous information. Existing fusion strategies often do not explicitly model metabolic hotspots in PET and anatomical boundary information in CT, which can impair the delineation of small lesions and lesions with indistinct boundaries. We propose CED-Diff, a diffusion-based tumour segmentation framework that integrates Differentiated Feature Extraction (DFE) and Task-Oriented Auxiliary Supervision (TAS). DFE explicitly models metabolic activity in PET and anatomical boundary information in CT to derive more discriminative features for segmentation, while TAS provides auxiliary constraints to improve feature learning. On the HeadNeck dataset, CED-Diff achieved the highest Dice score among the evaluated methods, with a Dice score of 66.28%, along with a Sensitivity of 78.70% and a Precision of 69.08%. These results suggest that explicit modelling of complementary PET and CT information can improve the accuracy and robustness of automatic head and neck tumour segmentation.