| Literature DB >> 35991584 |
Abstract
Medical Resonance Imaging (MRI) is one of the preferred imaging methods for brain tumor diagnosis and getting detailed information on tumor type, location, size, identification, and detection. Segmentation divides an image into multiple segments and describes the separation of the suspicious region from pre-processed MRI images to make the simpler image that is more meaningful and easier to examine. There are many segmentation methods, embedded with detection devices, and the response of each method is different. The study article focuses on comparing the performance of several image segmentation algorithms for brain tumor diagnosis, such as Otsu's, watershed, level set, K-means, HAAR Discrete Wavelet Transform (DWT), and Convolutional Neural Network (CNN). All of the techniques are simulated in MATLAB using online images from the Brain Tumor Image Segmentation Benchmark (BRATS) dataset-2018. The performance of these methods is analyzed based on response time and measures such as recall, precision, F-measures, and accuracy. The measured accuracy of Otsu's, watershed, level set, K-means, DWT, and CNN methods is 71.42%, 78.26%, 80.45%, 84.34%, 86.95%, and 91.39 respectively. The response time of CNN is 2.519 s in the MATLAB simulation environment for the designed algorithm. The novelty of the work is that CNN has been proven the best algorithm in comparison to all other methods for brain tumor image segmentation. The simulated and estimated parameters provide the direction to researchers to choose the specific algorithm for embedded hardware solutions and develop the optimal machine-learning models, as the industries are looking for the optimal solutions of CNN and deep learning-based hardware models for the brain tumor.Entities:
Keywords: Convolutional neural network (CNN); Discrete Wavelet Transform (DWT); Image segmentation; MATLAB image processing; Magnetic resonance imaging (MRI)
Year: 2022 PMID: 35991584 PMCID: PMC9379244 DOI: 10.1007/s11042-022-13636-y
Source DB: PubMed Journal: Multimed Tools Appl ISSN: 1380-7501 Impact factor: 2.577
Fig. 1a Sub-bands in DWT processing [12] b Filter banks in DWT processing [33]
Fig. 2Example of HAAR DWT [12]
Fig. 3CNN architecture
Image size with different layers in CNN
| Image | Layer | Filter Size | Image Size |
|---|---|---|---|
| Image-1 | Input image | 2 × 2 | 64 × 64 × 1 |
| Convolution layer-1 | 2 × 2 | 32 × 32 × 1 | |
| Pooling layer-1 | 2 × 2 | 2 × 2 × 1 | |
| Convolution layer-2 | 2 × 2 | 16 × 16 × 1 | |
| Pooling layer-2 | 2 × 2 | 2 × 2 × 1 | |
| Fully-Connected Layer | 2 × 2 | 8 × 8 × 1 | |
| Image-2 | Input image | 2 × 2 | 64 × 64 × 1 |
| Convolution layer-1 | 2 × 2 | 32 × 32 × 1 | |
| Pooling layer-1 | 2 × 2 | 2 × 2 × 1 | |
| Convolution layer-2 | 2 × 2 | 16 × 16 × 1 | |
| Pooling layer-2 | 2 × 2 | 2 × 2 × 1 | |
| Fully-Connected Layer | 2 × 2 | 8 × 8 × 1 | |
| : | : | : | : |
| Image − 10 | Input image | 2 × 2 | 64 × 64 × 1 |
| Convolution layer-1 | 2 × 2 | 32 × 32 × 1 | |
| Pooling layer-1 | 2 × 2 | 2 × 2 × 1 | |
| Convolution layer-2 | 2 × 2 | 16 × 16 × 1 | |
| Pooling layer-2 | 2 × 2 | 2 × 2 × 1 | |
| Fully-Connected Layer | 2 × 2 | 8 × 8 × 1 |
Fig. 4Methodology for tumor detection
MATLAB response time of different methods for brain tumor segmentation
| Method | Response time (Seconds) |
|---|---|
| CNN | 2.519 |
| DWT | 2.675 |
| K-Means | 4.571 |
| Level Set Method | 7.290 |
| Watershed Algorithm | 9.219 |
| Otsu’s Method | 12.500 |
Fig. 5Response time comparison graph
Comparative performance measures
| Method | TP | TN | FP | FN | Recall | Precision | F-measure | Accuracy |
|---|---|---|---|---|---|---|---|---|
| CNN | 80 | 90 | 4 | 12 | 0.869565217 | 0.952380952 | 0.909090909 | 0.913978495 |
| DWT | 85 | 75 | 13 | 11 | 0.885416667 | 0.867346939 | 0.876288660 | 0.869565217 |
| K-Means | 81 | 86 | 25 | 6 | 0.931034483 | 0.764150943 | 0.839378238 | 0.843434343 |
| Level Set | 76 | 64 | 12 | 22 | 0.775510204 | 0.863636364 | 0.817204301 | 0.804597701 |
| Watershed | 72 | 54 | 15 | 20 | 0.782608696 | 0.827586207 | 0.804469274 | 0.782608696 |
| Otsu’s | 75 | 35 | 9 | 35 | 0.681818182 | 0.892857143 | 0.773195876 | 0.714285714 |
Fig. 6Brain tumor segmentation using the watershed algorithm and level set methods
Fig. 7Brain tumor segmentation using CNN and other algorithms