.png){kind=icon}
返回
Mutual ensemble learning for brain tumor segmentation
.png){kind=icon}
PRE
AI
DOI：10.1016/j.neucom.2022.06.058.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
It is challenging to reduce the generalization errors of brain tumor segmentation models on test data, as the nature of the high diversity of tumors. The model ensemble combining multiple models to make the final prediction is a reliable strategy to reduce generalization errors. Conventionally, these models that acted as member networks of the ensemble are trained separately. Each member network is trained inde-pendently on a data subset or a single perspective of 3D brain volumes. To enable each member network to be trained on the complete training dataset as well as 3D images, we propose a mutual learning method to mutually train multiple 3D segmentation networks of the ensemble, called mutual ensemble learning (MEL). Mutual learning can enable knowledge exchange between networks and let them teach each other during the training process so that each member can converge to a better local minimum com-pared with when trained separately. Meanwhile, mutual learning also keeps the partially independent errors made by different member networks, and combining these member networks can reduce overall errors on the test set. In a word, combining these stronger members gives better ensemble results and it would not bring more cost in the test stage. To realize the mutual learning of multiple 3D brain tumor segmentation networks, we introduce a novel loss called Consensus Dice loss to explicitly exchange infor-mation among different networks during training. This loss considers the overlap of prediction probabil-ities by multiple networks and ground truth to enhance the overlapping pixels' probabilities. Extensive experiments are conducted on the brain tumor segmentation, i.e., BRATS 2015 and 2018 datasets. Results on the databases indicate that the proposed approach consistently improves the segmentation performance of the baseline network. Furthermore, our proposed method achieves state-of-the-art per-formance on the BRATS 2018 online validation set, and it is flexible to be extended by diverse homoge-neous and heterogeneous state-of-the-art networks.(c) 2022 Elsevier B.V. All rights reserved.
Keyword:
Mutual ensemble learning
Model ensemble
Convolutional network
Brain tumor
Segmentation
期刊
IF:
6.5
论文数:
2.5W
被引数:
6.5W
机构
暂无机构信息
引用论文
Analysis of the formation of monodisperse populations by homogeneous nucleation通过均相成核分析单分散种群的形成
PREAI
分享收藏分享收藏Segmentation of the multimodal brain tumor image used the multi-pathway architecture method based on 3D FCN.png)
NEUROCOMPUTING
NEUROCOMPUTINGIF6.5
PREAI
分享收藏AdaEn-Net: An ensemble of adaptive 2D-3D Fully Convolutional Networks for medical image segmentationNEURAL NETWORKS
NEURAL NETWORKSIF6.3
PREAI
分享收藏Dual-force convolutional neural networks for accurate brain tumor segmentationPATTERN RECOGNITION
PATTERN RECOGNITIONIF7.6
PREAI
分享收藏
查看更多内容