Sun, Yanming ORCID: https://orcid.org/0000-0003-0471-8215 and Wang, Chunyan (2022) A computation-efficient CNN system for high-quality brain tumor segmentation. Biomedical Signal Processing and Control, 74 (103475).
Preview |
Text (application/pdf)
688kBYanmingSun_manuscript.pdf - Accepted Version Available under License Spectrum Terms of Access. |
Official URL: https://doi.org/10.1016/j.bspc.2021.103475
Abstract
In this paper, a reliable computation-efficient system of Convolutional Neural Network (CNN) is proposed for brain tumor segmentation. It consists of a segmentation-CNN, a pre-CNN block for data reduction and a refinement block. The unique CNN is custom-designed, following the proposed paradigm of ASCNN (Application Specific CNN), to perform mono-modality and cross-modality feature extractions, tumor localization and pixel classification. It features modality-wise normalization to improve the input data quality, depthwise convolution, combined with instance normalization, for the mono-modality feature extraction, bilinear upsampling for dimension expansion without introducing randomness, and weighted data addition for signal modulation. The proposed activation function Full-ReLU helps to halve the number of kernels in convolution layers of high-pass filtering without degrading processing quality. In this specific design context, the CNN is structured to have 7 convolution layers, requiring only 108 kernels and 20308 trainable parameters in total. The number of kernels in each layer is made just-sufficient for its task, instead of exponentially growing over the layers, with a view to a higher information density in data channels and lower randomness in network training. Extensive experiments with BRATS2018 dataset have been conducted to confirm the high-level processing quality and reproducibility of the system. The mean-dice-scores for enhancing-tumor, whole-tumor and tumor-core are 77.2%, 89.2% and 76.3%, respectively. Testing each patient case requires only 29.07G Flops, a tiny fraction of what found in literature. The simple structure and reliable high processing quality of the proposed system will facilitate its implementation and medical applications.
Divisions: | Concordia University > Gina Cody School of Engineering and Computer Science > Electrical and Computer Engineering |
---|---|
Item Type: | Article |
Refereed: | Yes |
Authors: | Sun, Yanming and Wang, Chunyan |
Journal or Publication: | Biomedical Signal Processing and Control |
Date: | April 2022 |
Digital Object Identifier (DOI): | 10.1016/j.bspc.2021.103475 |
ID Code: | 994652 |
Deposited By: | Chunyan Wang |
Deposited On: | 08 Oct 2024 16:16 |
Last Modified: | 08 Oct 2024 16:16 |
Repository Staff Only: item control page