Author:
Bacanin Nebojsa,Zivkovic Miodrag,Al-Turjman Fadi,Venkatachalam K.,Trojovský Pavel,Strumberger Ivana,Bezdan Timea
Abstract
AbstractDeep learning has recently been utilized with great success in a large number of diverse application domains, such as visual and face recognition, natural language processing, speech recognition, and handwriting identification. Convolutional neural networks, that belong to the deep learning models, are a subtype of artificial neural networks, which are inspired by the complex structure of the human brain and are often used for image classification tasks. One of the biggest challenges in all deep neural networks is the overfitting issue, which happens when the model performs well on the training data, but fails to make accurate predictions for the new data that is fed into the model. Several regularization methods have been introduced to prevent the overfitting problem. In the research presented in this manuscript, the overfitting challenge was tackled by selecting a proper value for the regularization parameter dropout by utilizing a swarm intelligence approach. Notwithstanding that the swarm algorithms have already been successfully applied to this domain, according to the available literature survey, their potential is still not fully investigated. Finding the optimal value of dropout is a challenging and time-consuming task if it is performed manually. Therefore, this research proposes an automated framework based on the hybridized sine cosine algorithm for tackling this major deep learning issue. The first experiment was conducted over four benchmark datasets: MNIST, CIFAR10, Semeion, and UPS, while the second experiment was performed on the brain tumor magnetic resonance imaging classification task. The obtained experimental results are compared to those generated by several similar approaches. The overall experimental results indicate that the proposed method outperforms other state-of-the-art methods included in the comparative analysis in terms of classification error and accuracy.
Funder
Ministarstvo Prosvete, Nauke i Tehnološkog Razvoja
Excellence Project University of Hradec Králové
Publisher
Springer Science and Business Media LLC
Reference71 articles.
1. McCarthy, J. From here to human-level AI. Artif. Intell. 171(18), 1174–1182 (2007).
2. Arel, I., Rose, D. C. & Karnowski, T. P. Deep machine learning-a new frontier in artificial intelligence research [research frontier]. IEEE Comput. Intell. Mag. 5(4), 13–18 (2010).
3. Hongtao, L. & Qinchuan, Z. Applications of deep convolutional neural network in computer vision. J. Data Acquis. Process. 31(1), 1–17 (2016).
4. Xiao, T., Xu, Y., Yang, K., Zhang, J., Peng, Y. & Zhang, Z. The application of two-level attention models in deep convolutional neural network for fine-grained image classification, in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2015, 842–850.
5. Zhang, Y., Zhao, D., Sun, J., Zou, G. & Li, W. Adaptive convolutional neural network and its application in face recognition. Neural Process. Lett. 43(2), 389–399 (2016).
Cited by
85 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献