Abstract
In recent years, numerous deep learning models for medical image classification have emerged, with varying accuracies influenced by factors like image quality, content, and the convoluted low-level features. In this article, we aim to identify key features that enhance the precision of convolutional neural networks (CNNs), which are widely employed in this domain. We empirically demonstrate that parameters like activation functions, image augmentation, convolutional layers, filter sizes, and max pooling significantly impact precision, while others like hidden layers, epochs, and batch sizes affect quality and runtime negatively. To validate our proposal, we use a publicly-available dataset of colored Breast Cancer images which comprises 27,751 images distributed over three sub-datasets as follows: 4162 testing images (2980 no cancer and 1182 with cancer), 19,426 training images (13,911 no cancer and 5515 with cancer), and 4163 validation images (2981 no cancer and 1182 with cancer). Multiple experimental runs were conducted, and the outcomes of each run were recorded, taking into account various evaluation metrics and components of the CNN architecture. Results demonstrate that the highest CNN classifier’s accuracy was achieved when coupling RelU and Sigmoid activation functions using a 3 × 3 filter with a batch of size 32 and 20 epochs and 2 hidden layers. A Similar accuracy rate (83.55%) has also been achieved when coupling ReLU and Sigmoid activation functions using the DensNet-201 pre-trained CNN model.
Similar content being viewed by others
Data availability
Not applicable.
References
Chaurasia V, Pal S, Tiwari B (2018) Prediction of benign and malignant breast cancer using data mining techniques. J Algorithms Comput Technol 12(2):119–126
Yurttakal AH, Erbay H, İkizceli T, Karaçavuş S (2020) Detection of breast cancer via deep convolution neural networks using MRI images. Multimed Tools Appl 79(21):15555–15573
Vaka AR, Soni B, Reddy S (2020) Breast cancer detection by leveraging machine learning. ICT Express 6(4):320–324
Ting FF, Tan YJ, Sim KS (2019) Convolutional neural network improvement for breast cancer classification. Expert Syst Appl 120:103–115
Cai L, Gao J, Zhao D (2020) A review of the application of deep learning in medical image classification and segmentation. Ann Transl Med 8(11):713
Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. Adv Neural Inf Process Syst 25: 84-90
Singh R, Agarwal BB (2023) An automated brain tumor classification in MR images using an enhanced convolutional neural network. Int J Inf Technol 15(2):665–674
He K, Zhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. In: Proceedings of the IEEE conference on computer vision and pattern recognition, Las Vegas, NV, USA.
Agrawal S, Sahu SP (2023) Image-based Parkinson disease detection using deep transfer learning and optimization algorithm. Int J Inf Technol 16:871–879
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ (2017) Densely connected convolutional networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition, Honolulu, HI, USA.
Singh O, Singh KK (2023) An approach to classify lung and colon cancer of histopathology images using deep feature extraction and an ensemble method. Int J Inf Technol 15(8):4149–4160
Bhairnallykar ST, Narawade V (2023) Segmentation of MR images using DN convolutional neural network. Int J Inf Technol 15(8):4565–4576
Hu J, Shen L, Sun G (2018) Squeeze-and-excitation networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition, Salt Lake City, UT, USA.
Mishra AK, Roy P, Bandyopadhyay S, Das SK (2022) Feature fusion based machine learning pipeline to improve breast cancer prediction. Multimed Tools Appl 81(26):37627–37655
Toğaçar M, Özkurt KB, Ergen B, Cömert Z (2020) BreastNet: a novel convolutional neural network model through histopathological images for the diagnosis of breast cancer. Phys A 545:123592
Lin C-J, Jeng S-Y (2020) Optimization of deep learning network parameters using uniform experimental design for breast cancer histopathological image classification. Diagnostics 10(9):662
Kaymak S, Helwan A, Uzun D (2017) Breast cancer image classification using artificial neural networks. Procedia Comput Sci 120:126–131
Nazeri K, Aminpour A, Ebrahimi M (2018) Two-stage convolutional neural network for breast cancer histology image classification. In: International conference image analysis and recognition. Springer
Wahab N, Khan A (2020) Multifaceted fused-CNN based scoring of breast cancer whole-slide histopathology images. Appl Soft Comput 97:106808
Nejad EM, Affendey LS, Latip RB, Bin Ishak I. Classification of histopathology images of breast into benign and malignant using a single-layer convolutional neural network. In: Proceedings of the International conference on imaging, signal processing and communication
Gour M, Jain S, Sunil Kumar T (2020) Residual learning based CNN for breast cancer histopathological image classification. Int J Imaging Syst Technol 30(3):621–635
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
All authors certify that they have no affiliations with or involvement in any organization or entity with any financial interest or non-financial interest in the subject matter or materials discussed in this manuscript. The authors did not receive support from any organization for this work. The study was conducted on a publicly available dataset as cited within.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Maree, M., Zanoon, T., Dababat, A. et al. Constructing a hybrid activation and parameter-fusion based CNN medical image classifier. Int. j. inf. tecnol. (2024). https://doi.org/10.1007/s41870-024-01798-x
Received:
Accepted:
Published:
DOI: https://doi.org/10.1007/s41870-024-01798-x