CorrDiff: Corrective Diffusion Model for Accurate MRI Brain Tumor Segmentation

Wenqing Li,Wenhui Huang,Yuanjie Zheng

DOI: https://doi.org/10.1109/jbhi.2024.3353272

IF: 7.7

2024-01-01

IEEE Journal of Biomedical and Health Informatics

Abstract:Accurate segmentation of brain tumors in MRI images is imperative for precise clinical diagnosis and treatment. However, existing medical image segmentation methods exhibit errors, which can be categorized into two types: random errors and systematic errors. Random errors, arising from various unpredictable effects, pose challenges in terms of detection and correction. Conversely, systematic errors, attributable to systematic effects, can be effectively addressed through machine learning techniques. In this paper, we propose a corrective diffusion model for accurate MRI brain tumor segmentation by correcting systematic errors. This marks the first application of the diffusion model for correcting systematic segmentation errors. Additionally, we introduce the Vector Quantized Variational Autoencoder (VQ-VAE) to compress the original data into a discrete coding codebook. This not only reduces the dimensionality of the training data but also enhances the stability of the correction diffusion model. Furthermore, we propose the Multi-Fusion Attention Mechanism, which can effectively enhances the segmentation performance of brain tumor images, and enhance the flexibility and reliability of the corrective diffusion model. Our model is evaluated on the BRATS2019, BRATS2020, and Jun Cheng datasets. Experimental results demonstrate the effectiveness of our model over state-of-the-art methods in brain tumor segmentation.

computer science, interdisciplinary applications,mathematical & computational biology,medical informatics, information systems

What problem does this paper attempt to address?

The paper aims to address the issue of accurate segmentation in MRI brain tumor images, particularly in the context of clinical diagnosis and treatment planning. The authors point out that although significant progress has been made in medical image segmentation technology in recent years, existing methods still suffer from two types of errors: random errors and systematic errors. To improve segmentation accuracy, the paper proposes a corrective diffusion model called Corrective Diffusion (CorrDiff).

The main contributions include:

1. **Proposing the Corrective Diffusion Model**: This is a novel model designed for accurately segmenting MRI brain tumor images. Unlike traditional segmentation methods based on the diffusion process, the corrective diffusion model is inspired by corrective learning and specifically targets the systematic errors in the initial segmentation results, thereby improving the accuracy of the segmentation results.

2. **Introducing the Vector Quantized Variational Autoencoder (VQ-VAE)**: This model is used to compress the original data into a discrete codebook, which not only reduces the dimensionality of the training data but also enhances the stability of the corrective diffusion model.

3. **Proposing a Multi-Fusion Attention Mechanism**: This mechanism enhances the flexibility and reliability of the model by effectively combining MRI brain tumor image information and systematic error maps, further improving the overall performance of the model.

In summary, the main goal of the paper is to improve the accuracy of MRI brain tumor segmentation through the aforementioned innovations, thereby supporting more effective clinical diagnosis and treatment planning. Experimental results show that the proposed model outperforms the current state-of-the-art methods on the BRATS2019, BRATS2020, and Jun Cheng datasets.