Df-Segdiff: Adiffusion Segmentation Model Using a New Distributed Parallel Computing Algorithm

Abstract

Brain tumours are among the most life-threatening diseases, and automatic segmentation of brain tumours from medical images is crucial for clinicians to identify and quantify tumour regions with high precision. While traditional segmentation models have laid the groundwork, diffusion models have since been developed to better manage complex medical data. However, diffusion models often face challenges related to insufficient parallel computing power and inefficient GPU utilization. To address these issues, we propose the DF-SegDiff model, which includes diffusion segmentation, parallel data processing, a distributed training model, a dynamic balancing parameter and model fusion. This approach significantly reduces training time while achieving an average Dice score of 0.87, with several samples reaching Dice values close to 0.94. By combining BRATS2020 with the Medical Segmentation Decathlon dataset, we also integrated a comprehensive dataset containing 800 training samples and 53 test samples. Evaluation of the model using Dice, IoU, and other relevant metrics demonstrates that our method outperforms current state-of-the-art techniques.