CDCG-UNet: Chaotic Optimization Assisted Brain Tumor Segmentation Based on Dilated Channel Gate Attention U-Net Model.

Abstract

Brain tumours are one of the most deadly and noticeable types of cancer, affecting both children and adults. One of the major drawbacks in brain tumour identification is the late diagnosis and high cost of brain tumour-detecting devices. Most existing approaches use ML algorithms to address problems, but they have drawbacks such as low accuracy, high loss, and high computing cost. To address these challenges, a novel U-Net model for tumour segmentation in magnetic resonance images (MRI) is proposed. Initially, images are claimed from the dataset and pre-processed with the Probabilistic Hybrid Wiener filter (PHWF) to remove unwanted noise and improve image quality. To reduce model complexity, the pre-processed images are submitted to a feature extraction procedure known as 3D Convolutional Vision Transformer (3D-VT). To perform the segmentation approach using chaotic optimization assisted Dilated Channel Gate attention U-Net (CDCG-UNet) model to segment brain tumour regions effectively. The proposed approach segments tumour portions as whole tumour (WT), tumour Core (TC), and Enhancing Tumour (ET) positions. The optimization loss function can be performed using the Chaotic Harris Shrinking Spiral optimization algorithm (CHSOA). The proposed CDCG-UNet model is evaluated with three datasets: BRATS 2021, BRATS 2020, and BRATS 2023. For the BRATS 2021 dataset, the proposed CDCG-UNet model obtained a dice score of 0.972 for ET, 0.987 for CT, and 0.98 for WT. For the BRATS 2020 dataset, the proposed CDCG-UNet model produced a dice score of 98.87% for ET, 98.67% for CT, and 99.1% for WT. The CDCG-UNet model is further evaluated using the BRATS 2023 dataset, which yields 98.42% for ET, 98.08% for CT, and 99.3% for WT.