Multitask Learning with Multiscale Residual Attention for Brain Tumor Segmentation and Classification
文献类型:期刊论文
作者 | Gaoxiang Li, Xiao Hui, Wenjing Li, Yanlin Luo |
刊名 | Machine Intelligence Research |
出版日期 | 2023 |
卷号 | 20期号:6页码:897-908 |
ISSN号 | 2731-538X |
关键词 | Brain tumor segmentation and classification, multitask learning, multiscale residual attention module (MRAM), dynamic weight training, prior knowledge |
DOI | 10.1007/s11633-022-1392-6 |
英文摘要 | Automatic segmentation and classification of brain tumors are of great importance to clinical treatment. However, they are challenging due to the varied and small morphology of the tumors. In this paper, we propose a multitask multiscale residual attention network (MMRAN) to simultaneously solve the problem of accurately segmenting and classifying brain tumors. The proposed MMRAN is based on U-Net, and a parallel branch is added at the end of the encoder as the classification network. First, we propose a novel multiscale residual attention module (MRAM) that can aggregate contextual features and combine channel attention and spatial attention better and add it to the shared parameter layer of MMRAN. Second, we propose a method of dynamic weight training that can improve model performance while minimizing the need for multiple experiments to determine the optimal weights for each task. Finally, prior knowledge of brain tumors is added to the postprocessing of segmented images to further improve the segmentation accuracy. We evaluated MMRAN on a brain tumor data set containing meningioma, glioma, and pituitary tumors. In terms of segmentation performance, our method achieves Dice, Hausdorff distance (HD), mean intersection over union (MIoU), and mean pixel accuracy (MPA) values of 80.03%, 6.649 mm, 84.38%, and 89.41%, respectively. In terms of classification performance, our method achieves accuracy, recall, precision, and F1-score of 89.87%, 90.44%, 88.56%, and 89.49%, respectively. Compared with other networks, MMRAN performs better in segmentation and classification, which significantly aids medical professionals in brain tumor management. The code and dataset are available at https://github.com/linkenfaqiu/MMRAN. |
源URL | [http://ir.ia.ac.cn/handle/173211/54174] |
专题 | 自动化研究所_学术期刊_International Journal of Automation and Computing |
作者单位 | School of Artificial Intelligence, Beijing Normal University, Beijing 100875, China |
推荐引用方式 GB/T 7714 | Gaoxiang Li, Xiao Hui, Wenjing Li, Yanlin Luo. Multitask Learning with Multiscale Residual Attention for Brain Tumor Segmentation and Classification[J]. Machine Intelligence Research,2023,20(6):897-908. |
APA | Gaoxiang Li, Xiao Hui, Wenjing Li, Yanlin Luo.(2023).Multitask Learning with Multiscale Residual Attention for Brain Tumor Segmentation and Classification.Machine Intelligence Research,20(6),897-908. |
MLA | Gaoxiang Li, Xiao Hui, Wenjing Li, Yanlin Luo."Multitask Learning with Multiscale Residual Attention for Brain Tumor Segmentation and Classification".Machine Intelligence Research 20.6(2023):897-908. |
入库方式: OAI收割
来源:自动化研究所
浏览0
下载0
收藏0
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。