QMLS: quaternion mutual learning strategy for multi-modal brain tumor segmentation.
Zhengnan DengGuoheng HuangXiaochen YuanGuo ZhongTongxu LinChi-Man PunZhixin HuangZhixin LiangPublished in: Physics in medicine and biology (2023)
Due to non-invasive imaging and the multimodality of Magnetic Resonance Imaging (MRI) images, MRI-based multi-modal brain tumor segmentation (MBTS) studies are attracting more and more attention in recent years. With the great success of convolutional neural networks (CNN) in various computer vision tasks, lots of MBTS models have been proposed to address the technical challenges of MBTS. However, the problem of limited data collection usually exists in MBTS tasks, making existing studies usually have difficulty in fully exploring the multi-modal MRI images to mine complementary information among different modalities.
Approach: We propose a novel Quaternion Mutual Learning Strategy (QMLS), which consists of a voxel-wise lesion knowledge mutual learning mechanism (VLKML mechanism) and a quaternion multimodal feature learning module (QMFL module). Specifically, VLKML mechanism allows the networks to converge to a robust minimum so that aggressive data augmentation techniques can be applied to fully expand the limited data. In particular, quaternion-valued QMFL module treats different modalities as components of quaternions to sufficiently learn complementary information among different modalities on the hypercomplex domain while significantly reduces the number of parameters by about 75\%.
Main results: Extensive experiments on the dataset BraTS 2020 and BraTS 2019 indicate that QMLS achieves superior results to current popular methods with less computational cost.
Significance: We propose a novel algorithm to brain tumor segmentation task and achieves better performance with fewer parameters, which helps the clinical application of automatic brain tumor segmentation.
Keyphrases
- convolutional neural network
- deep learning
- magnetic resonance imaging
- contrast enhanced
- artificial intelligence
- big data
- machine learning
- electronic health record
- working memory
- diffusion weighted imaging
- healthcare
- high resolution
- computed tomography
- magnetic resonance
- pain management
- case control
- photodynamic therapy
- optical coherence tomography
- mass spectrometry