Attention Mechanism-Enhanced Imaging Techniques for Brain Tumor Classification
Abstract
At present, deep learning has been successfully applied in the field of medical diagnosis, and it can effectively improve the diagnostic accuracy by using deep learning to predict brain tumor images. Based on traditional convolution neural networks tend to ignore the problem of key location in brain tumor image information, this paper proposes a door control channel attention conversion unit, door control channel attention switching unit by using the relationship between a small number of parameters to change the channel, under the condition of without any increase in computational cost more significantly improve classification accuracy. Also, the Sequeeze and Excitation block is introduced for the communication between the channels. Finally, a multipath attentional network model (FSnet) for brain tumor image classification is constructed based on ResNeXt network model. Experimental tests on brain MRI data sets show that FSnet has better classification performance than traditional convolutional neural networks.
Show Figures
Share and Cite
Article Metrics
References
- an M, Pang R, Le Q V. Efficientdet: Scalable and efficient object detection[C].Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2020: 10781-10790.
- Ismael M R, Abdel-Qader I. Brain tumor classification via statistical features and back-propagation neural network[C].2018 IEEE international conference on electro/information technology (EIT). IEEE, 2018: 0252-0257.
- Pashaei A, Sajedi H, Jazayeri N. Brain tumor classification via convolutional neural network and extreme learning machines[C].2018 8th International conference on computer and knowledge engineering (ICCKE). IEEE, 2018: 314-319.
- Abiwinanda N, Hanif M, Hesaputra S T, et al. Brain tumor classification using convolutional neural network[C].World congress on medical physics and biomedical engineering 2018. Springer, Singapore, 2019: 183-189.
- Sultan H H, Salem N M, Al-Atabany W. Multi-classification of brain tumor images using deep neural network[J]. IEEE Access, 2019, 7: 69215-69225.
- Anaraki A K, Ayati M, Kazemi F. Magnetic resonance imaging-based brain tumor grades classification and grading via convolutional neural networks and genetic algorithms[J]. biocybernetics and biomedical engineering, 2019, 39(1): 63-74.
- Díaz-Pernas F J, Martínez-Zarzuela M, Antón-Rodríguez M, et al. Learning and surface boundary feedbacks for colour natural scene perception[J]. Applied Soft Computing, 2017, 61: 30-41.
- ulesz B, Bergen J R. Textons, the fundamental elements in preattentive vision and perception of textures, Bell Systems Tech[J]. J, 1983, 62: 1619-1645.
- Kruger N, Janssen P, Kalkan S, et al. Deep hierarchies in the primate visual cortex: What can we learn for computer vision?[J]. IEEE transactions on pattern analysis and machine intelligence, 2012, 35(8): 1847-1871.
- Afshar P, Plataniotis K N, Mohammadi A. Capsule networks for brain tumor classification based on MRI images and coarse tumor boundaries[C].ICASSP 2019-2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 2019: 1368-1372.
- Yang Z, Zhu L, Wu Y, et al. Gated channel transformation for visual recognition[C].Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2020: 11794-11803.
- Hu J, Shen L, Sun G. Squeeze-and-excitation networks[C].Proceedings of the IEEE conference on computer vision and pattern recognition. 2018: 7132-7141.
- Liu L, Jiang H, He P, et al. On the variance of the adaptive learning rate and beyond[J]. ar Xiv preprint ar Xiv, 2019:1908.03265.
- Kingma D P, Ba J. Adam: a method for stochastic optimization. ar Xiv[J]. ar Xiv preprint ar Xiv:1412.6980, 2014, 22.
- Bottou L. Stochastic gradient descent tricks[M].Neural networks: Tricks of the trade. Springer, Berlin, Heidelberg, 2012: 421-436.
- Mazzia, Vittorio, Francesco Salvetti, and Marcello Chiaberge. "Efficient-Caps Net: Capsule Network with Self-Attention Routing." ar Xiv preprint ar Xiv:2101.12491 (2021).
- Sadad T, Rehman A, Munir A, et al. Brain tumor detection and multi‐classification using advanced deep learning techniques[J]. Microscopy Research and Technique.
- Alaraimi S, Okedu K E, Tianfield H, et al. Transfer learning networks with skip connections for classification of brain tumors[J]. International Journal of Imaging Systems and Technology, 2021.
- Cheng J, Yang W, Huang M, et al. Retrieval of brain tumors by adaptive spatial pooling and fisher vector representation[J]. Plo S one, 2016, 11(6): e0157112.
- Simonyan K , Zisserman A . Very Deep Convolutional Networks for Large-Scale Image Recognition[J]. Computer Science, 2014.
- He K , Zhang X , Ren S , et al. Deep Residual Learning for Image Recognition[J]. IEEE, 2016.