LightNet: A Novel Lightweight Convolutional Network for Brain Tumor Segmentation in Healthcare

被引:7
作者
Wu, Dongyuan [1 ]
Tao, Junyi [2 ]
Qin, Zhen [1 ]
Mumtaz, Rao Asad [3 ]
Qin, Jing [1 ]
Yu, Linfang [4 ,5 ]
Courtney, Jane [6 ]
机构
[1] Univ Elect Sci & Technol China, Network & Data Secur Key Lab Sichuan Prov, Chengdu 610054, Peoples R China
[2] Amazon Web Serv, Seattle, WA 98101 USA
[3] Khyber Med Univ, Peshawar 23301, Pakistan
[4] Univ Elect Sci & Technol China, Chengdu 610054, Peoples R China
[5] Univ Elect Sci & Technol China, China Mobile Chengdu Ind Res Inst, Chengdu 610054, Peoples R China
[6] Technol Univ Dublin, Sch Elect & Elect Engn, Dublin D07EWV4, Ireland
基金
中国国家自然科学基金;
关键词
Segmentation; Complex convolution; Light operations; Attention; Efficiency;
D O I
10.1109/JBHI.2023.3297227
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Diagnosis, treatment planning, surveillance, and the monitoring of clinical trials for brain diseases all benefit greatly from neuroimaging-based tumor segmentation. Recently, Convolutional Neural Networks (CNNs) have demonstrated promising results in enhancing the efficiency of image-based brain tumor segmentation. Most current work on CNNs, however, is devoted to creating increasingly complicated convolution modules to improve performance, which in turn raises the computing cost of the model. This work proposes a simple and effective feed-forward CNN, LightNet (Light Network). Based on multi-path and multi-level, it replaces traditional convolutional methods with light operations, which reduces network parameters and redundant feature maps. In the up-sampling stage, a light channel attention module is added to achieve richer multi-scale and spatial semantic feature information extraction of brain tumor. The performance of the network is evaluated in the Multimodal Brain Tumor Segmentation Challenge (BraTS 2015) dataset, and results are presented here alongside other high-performing CNNs. Results show comparable accuracy with other methods but with increased efficiency, segmentation performance, and reduced redundancy and computational complexity. The result is a high-performing network with a balance between efficiency and accuracy, allowing, for example, better energy performance on mobile devices.
引用
收藏
页码:4471 / 4482
页数:12
相关论文
共 47 条
[1]   SCA-CNN: Spatial and Channel-wise Attention in Convolutional Networks for Image Captioning [J].
Chen, Long ;
Zhang, Hanwang ;
Xiao, Jun ;
Nie, Liqiang ;
Shao, Jian ;
Liu, Wei ;
Chua, Tat-Seng .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6298-6306
[2]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[3]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861
[4]   Automated Brain Tumor Segmentation Using Multimodal Brain Scans: A Survey Based on Models Submitted to the BraTS 2012-2018 Challenges [J].
Ghaffari, Mina ;
Sowmya, Arcot ;
Oliver, Ruth .
IEEE REVIEWS IN BIOMEDICAL ENGINEERING, 2020, 13 :156-168
[5]   Deep Pyramidal Residual Networks [J].
Han, Dongyoon ;
Kim, Jiwhan ;
Kim, Junmo .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6307-6315
[6]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[7]  
Han S, 2016, Arxiv, DOI arXiv:1510.00149
[8]   Brain tumor segmentation with Deep Neural Networks [J].
Havaei, Mohammad ;
Davy, Axel ;
Warde-Farley, David ;
Biard, Antoine ;
Courville, Aaron ;
Bengio, Yoshua ;
Pal, Chris ;
Jodoin, Pierre-Marc ;
Larochelle, Hugo .
MEDICAL IMAGE ANALYSIS, 2017, 35 :18-31
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]   AMC: AutoML for Model Compression and Acceleration on Mobile Devices [J].
He, Yihui ;
Lin, Ji ;
Liu, Zhijian ;
Wang, Hanrui ;
Li, Li-Jia ;
Han, Song .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :815-832