TY - JOUR
T1 - QMLS
T2 - quaternion mutual learning strategy for multi-modal brain tumor segmentation
AU - Deng, Zhengnan
AU - Huang, Guoheng
AU - Yuan, Xiaochen
AU - Zhong, Guo
AU - Lin, Tongxu
AU - Pun, Chi Man
AU - Huang, Zhixin
AU - Liang, Zhixin
N1 - Publisher Copyright:
© 2023 Institute of Physics and Engineering in Medicine.
PY - 2024/1/7
Y1 - 2024/1/7
N2 - Objective. Due to non-invasive imaging and the multimodality of magnetic resonance imaging (MRI) images, MRI-based multi-modal brain tumor segmentation (MBTS) studies have attracted more and more attention in recent years. With the great success of convolutional neural networks 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 typically 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 multi-modal feature learning module (QMFL module). Specifically, the VLKML mechanism allows the networks to converge to a robust minimum so that aggressive data augmentation techniques can be applied to expand the limited data fully. In particular, the 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 reducing 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 for brain tumor segmentation task that achieves better performance with fewer parameters, which helps the clinical application of automatic brain tumor segmentation.
AB - Objective. Due to non-invasive imaging and the multimodality of magnetic resonance imaging (MRI) images, MRI-based multi-modal brain tumor segmentation (MBTS) studies have attracted more and more attention in recent years. With the great success of convolutional neural networks 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 typically 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 multi-modal feature learning module (QMFL module). Specifically, the VLKML mechanism allows the networks to converge to a robust minimum so that aggressive data augmentation techniques can be applied to expand the limited data fully. In particular, the 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 reducing 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 for brain tumor segmentation task that achieves better performance with fewer parameters, which helps the clinical application of automatic brain tumor segmentation.
KW - Brain tumor segmentation
KW - Lightweight
KW - Mutual learning
KW - Quaternion neural networks
UR - http://www.scopus.com/inward/record.url?scp=85181148639&partnerID=8YFLogxK
U2 - 10.1088/1361-6560/ad135e
DO - 10.1088/1361-6560/ad135e
M3 - Article
C2 - 38061066
AN - SCOPUS:85181148639
SN - 0031-9155
VL - 69
JO - Physics in Medicine and Biology
JF - Physics in Medicine and Biology
IS - 1
M1 - 015014
ER -