Enhancing Histopathological Image Classification: Optimal Fine-Tuning of Convolutional Neural Networks with Feature Extraction
PDF

Keywords

deep learning
convolutional neural networks
cancerVisionNet
image classification

How to Cite

Mehmood, M. H., Muhammad Hasnain, M. H., Mehmmod, R. ., Usman, . S. ., & Zulfiqar, R. Z. (2025). Enhancing Histopathological Image Classification: Optimal Fine-Tuning of Convolutional Neural Networks with Feature Extraction. KIET Journal of Computing and Information Sciences, 7(2), 35-59. https://doi.org/10.51153/kjcis.v7i2.226

Abstract

The area of medical image analysis has genuine obstacles, such as small data sets, complicated fine-tuning, and choosing the right architecture. This paper proposes a CancerVisionNet model based on the convolutional neural network (CNN) architecture with many layers to extract and classify features from cancer images. To train the CancerVisionNet model and avoid overfitting, data augmentation is carried out using a data set consisting of 220,025 images. The proposed CancerVisionNet model is evaluated on the PatchCamelyon dataset. Its remarkable area under the receiver operating characteristic (ROC) curve (AUC) measure is about 0.9. Compared to the other studies, the CancerVisionNet model stands out with a higher accuracy (95.4%). Moreover, this work demonstrates the potential of CNNs in medical image analysis, providing an effective approach to enhance classification accuracy and paving the way for further advances in the field. Although the results of this study pertain to histopathology and the PatchCamelyon dataset, the potential for a broader application awaits cross-domain validation. Future research works can include exploring alternative architectures and scalability to larger datasets.

https://doi.org/10.51153/kjcis.v7i2.226
PDF

References

Gurcan, M.N., Boucheron, L.E., Can, A., Madabhushi, A., Rajpoot, N.M., and Yener, B.: ‘Histopathological image analysis: A review’, IEEE reviews in biomedical engineering, 2009, 2, pp. 147-171

Metter, D.M., Colgan, T.J., Leung, S.T., Timmons, C.F., and Park, J.Y.: ‘Trends in the US and Canadian pathologist workforces from 2007 to 2017’, JAMA network open, 2019, 2, (5), pp. e194337-e194337

Yaqoob, A., Rehman, F., Sharif, H., Mahmood, M.H., Sharif, S., Ahmad, A., Ali, C.N., Hussain, A., and Khan, M.: ‘Skip Connections' Importance in Biomedical Image Segmentation’, in Editor (Ed.)^(Eds.): ‘Book Skip Connections' Importance in Biomedical Image Segmentation’ (IEEE, 2023, edn.), pp. 1-5

Mohammadian, S., Karsaz, A., and Roshan, Y.M.: ‘Comparative study of fine-tuning of pre-trained convolutional neural networks for diabetic retinopathy screening’, in Editor (Ed.)^(Eds.): ‘Book Comparative study of fine-tuning of pre-trained convolutional neural networks for diabetic retinopathy screening’ (IEEE, 2017, edn.), pp. 1-6

Khan, N.M., Abraham, N., and Hon, M.: ‘Transfer learning with intelligent training data selection for prediction of Alzheimer’s disease’, IEEE Access, 2019, 7, pp. 72726-72735

Hosny, K.M., Kassem, M.A., and Foaud, M.M.: ‘Classification of skin lesions using transfer learning and augmentation with Alex-net’, PLoS One, 2019, 14, (5), pp. e0217293

Harangi, B.: ‘Skin lesion classification with ensembles of deep convolutional neural networks’, Journal of biomedical informatics, 2018, 86, pp. 25-32

Fukushima, K.: ‘Neocognitron: A self-organizing neural network model for a mechanism of pattern recognition unaffected by shift in position’, Biol. Cybern., 1980, 36, (4), pp. 193-202

Krizhevsky, A., Sutskever, I., and Hinton, G.E.: ‘Imagenet classification with deep convolutional neural networks’, Advances in neural information processing systems, 2012, 25

Russakovsky, O., Deng, J., Su, H., Krause, J., Satheesh, S., Ma, S., Huang, Z., Karpathy, A., Khosla, A., and Bernstein, M.: ‘Imagenet large scale visual recognition challenge’, International journal of computer vision, 2015, 115, pp. 211-252

Gupta, R.K., Kaur, M., and Manhas, J.: ‘Tissue level based deep learning framework for early detection of dysplasia in oral squamous epithelium’, Journal of Multimedia Information System, 2019, 6, (2), pp. 81-86

Chollet, F.: ‘Deep learning with Python’ (Simon and Schuster, 2021. 2021)

Yosinski, J., Clune, J., Bengio, Y., and Lipson, H.: ‘How transferable are features in deep neural networks?’, Advances in neural information processing systems, 2014, 27

Kalbhor, M.M., and Shinde, S.V.: ‘Cervical cancer diagnosis using convolution neural network: feature learning and transfer learning approaches’, Soft Computing, 2023, pp. 1-11

Mehra, R.: ‘Breast cancer histology images classification: Training from scratch or transfer learning?’, ICT Express, 2018, 4, (4), pp. 247-254

Spanhol, F.A., Oliveira, L.S., Petitjean, C., and Heutte, L.: ‘A dataset for breast cancer histopathological image classification’, Ieee transactions on biomedical engineering, 2015, 63, (7), pp. 1455-1462

Kassani, S.H., Kassani, P.H., Wesolowski, M.J., Schneider, K.A., and Deters, R.: ‘Classification of histopathological biopsy images using ensemble of deep learning networks’, arXiv preprint arXiv:1909.11870, 2019

Veeling, B.S., Linmans, J., Winkens, J., Cohen, T., and Welling, M.: ‘Rotation equivariant CNNs for digital pathology’, in Editor (Ed.)^(Eds.): ‘Book Rotation equivariant CNNs for digital pathology’ (Springer, 2018, edn.), pp. 210-218

Bejnordi, B.E., Veta, M., Van Diest, P.J., Van Ginneken, B., Karssemeijer, N., Litjens, G., Van Der Laak, J.A., Hermsen, M., Manson, Q.F., and Balkenhol, M.: ‘Diagnostic assessment of deep learning algorithms for detection of lymph node metastases in women with breast cancer’, Jama, 2017, 318, (22), pp. 2199-2210

Moore, J., Linkert, M., Blackburn, C., Carroll, M., Ferguson, R.K., Flynn, H., Gillen, K., Leigh, R., Li, S., and Lindner, D.: ‘OMERO and Bio-Formats 5: flexible access to large bioimaging datasets at scale’, in Editor (Ed.)^(Eds.): ‘Book OMERO and Bio-Formats 5: flexible access to large bioimaging datasets at scale’ (SPIE, 2015, edn.), pp. 37-42

Simonyan, K., and Zisserman, A.: ‘Very deep convolutional networks for large-scale image recognition’, arXiv preprint arXiv:1409.1556, 2014

Howard, A.G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., Andreetto, M., and Adam, H.: ‘Mobilenets: Efficient convolutional neural networks for mobile vision applications’, arXiv preprint arXiv:1704.04861, 2017

Huang, G., Liu, Z., Van Der Maaten, L., and Weinberger, K.Q.: ‘Densely connected convolutional networks’, in Editor (Ed.)^(Eds.): ‘Book Densely connected convolutional networks’ (2017, edn.), pp. 4700-4708

Vesal, S., Ravikumar, N., Davari, A., Ellmann, S., and Maier, A.: ‘Classification of breast cancer histology images using transfer learning’, in Editor (Ed.)^(Eds.): ‘Book Classification of breast cancer histology images using transfer learning’ (Springer, 2018, edn.), pp. 812-819

Szegedy, C., Vanhoucke, V., Ioffe, S., Shlens, J., and Wojna, Z.: ‘Rethinking the inception architecture for computer vision’, in Editor (Ed.)^(Eds.): ‘Book Rethinking the inception architecture for computer vision’ (2016, edn.), pp. 2818-2826

He, K., Zhang, X., Ren, S., and Sun, J.: ‘Deep residual learning for image recognition’, in Editor (Ed.)^(Eds.): ‘Book Deep residual learning for image recognition’ (2016, edn.), pp. 770-778

Deniz, E., ?engür, A., Kadiro?lu, Z., Guo, Y., Bajaj, V., and Budak, Ü.: ‘Transfer learning based histopathologic image classification for breast cancer detection’, Health information science and systems, 2018, 6, pp. 1-7

Ahmad, H.M., Ghuffar, S., and Khurshid, K.: ‘Classification of breast cancer histology images using transfer learning’, in Editor (Ed.)^(Eds.): ‘Book Classification of breast cancer histology images using transfer learning’ (IEEE, 2019, edn.), pp. 328-332

Farhadipour, A.: ‘Lung and colon cancer detection with convolutional neural networks and adaptive histogram equalization’, Iran Journal of Computer Science, 2023, pp. 1-15

Deng, J., Dong, W., Socher, R., Li, L.-J., Li, K., and Fei-Fei, L.: ‘Imagenet: A large-scale hierarchical image database’, in Editor (Ed.)^(Eds.): ‘Book Imagenet: A large-scale hierarchical image database’ (Ieee, 2009, edn.), pp. 248-255

Kassani, S.H., Kassani, P.H., Wesolowski, M.J., Schneider, K.A., and Deters, R.: ‘Deep transfer learning based model for colorectal cancer histopathology segmentation: A comparative study of deep pre-trained models’, International Journal of Medical Informatics, 2022, 159, pp. 104669

Kumar, N., Sharma, M., Singh, V.P., Madan, C., and Mehandia, S.: ‘An empirical study of handcrafted and dense feature extraction techniques for lung and colon cancer classification from histopathological images’, Biomedical Signal Processing and Control, 2022, 75, pp. 103596

?engöz, N., Yi?it, T., Özmen, Ö., and Isik, A.H.: ‘Importance of preprocessing in histopathology image classification using deep convolutional neural network’, Advances in Artificial Intelligence Research, 2022, 2, (1), pp. 1-6

Zhong, Z., Zheng, M., Mai, H., Zhao, J., and Liu, X.: ‘Cancer image classification based on DenseNet model’, in Editor (Ed.)^(Eds.): ‘Book Cancer image classification based on DenseNet model’ (IOP Publishing, 2020, edn.), pp. 012143

Kingma, D.P., and Ba, J.: ‘Adam: A method for stochastic optimization’, arXiv preprint arXiv:1412.6980, 2014

Spanhol, F.A., Oliveira, L.S., Petitjean, C., and Heutte, L.: ‘Breast cancer histopathological image classification using convolutional neural networks’, in Editor (Ed.)^(Eds.): ‘Book Breast cancer histopathological image classification using convolutional neural networks’ (IEEE, 2016, edn.), pp. 2560-2567