Skip to main content

Advertisement

Springer Nature Link
Log in

A novel genetic algorithm-based approach for compression and acceleration of deep learning convolution neural network: an application in computer tomography lung cancer data

  • Original Article
  • Published:
Neural Computing and Applications Aims and scope Submit manuscript

Abstract

Deep learning (DL) models are computationally expensive in space and time, which makes it difficult to deploy DL models in edge computing devices, such as Raspberry-Pi or Jetson Nano. The current strategy uses genetic algorithm (GA), which compresses the deep convolution neural network models without compromising performance. GA was applied by converting the CNN layers into binary vectors. Further, the fitness function in GA was computed based on (i) the minimization of hidden units and (ii) test accuracy. The GA-based strategy was applied on different pre-trained architectures, namely AlexNet, VGG16, SqueezeNet, and ResNet50, respectively, by using three kinds of datasets, namely MNIST, CIFAR-10, and CIFAR-100. The proposed approach demonstrated the reduction in the storage space of AlexNet by 87.62%, 80.97%, and 86.20% corresponding to the datasets MNIST, CIFAR-10, and CIFAR-100, respectively. Further, for the same three datasets, namely VGG16, ResNet50, and SqueezeNet, the system average compression was 91.15%, 78.42%, and 38.40%, respectively. In addition to that, the inference time of the models using proposed strategy was significantly improved with an average of the four datasets of ~ 35.61%, 9.23%, 73.76%, and 79.93% corresponding to AlexNet, SqueezeNet, ResNet50, and VGG16 models. Further, our method when applied to the proposed CNN using the LIDC-IDRI dataset showed a 90.3% reduction in the storage space and inference time. DL system when optimized using GA shows improved performance in both storage and execution time.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11
Fig. 12
Fig. 13
Fig. 14
Fig. 15
Fig. 16
Fig. 17

Similar content being viewed by others

Explore related subjects

Discover the latest articles, news and stories from top researchers in related subjects.

References

  1. Agarwal M, Gupta SK, Biswas K (2020) Development of efficient cnn model for tomato crop disease identification. Sustain Comput: Inform Syst 28:100407

    Google Scholar 

  2. Agarwal M, Gupta S, Biswas K (2020) A new Conv2D model with modified ReLU activation function for identification of disease type and severity in cucumber plant. Sustain Comput: Inform Syst, 100473

  3. Agarwal M, et al (2019) FCNN-LDA: a faster convolution neural network model for leaf disease identification on apple's leaf dataset. In: 2019 12th International conference on information and communication technology and system (ICTS). 2019. IEEE

  4. Agarwal M, et al (2020) Potato crop disease classification using convolutional neural network. In: Smart systems and IoT: innovations in computing, Springer. pp 391–400

  5. Agarwal M, Gupta SK, Biswas K. Grape disease identification using convolution neural network. In: 2019 23rd International computer science and engineering conference (ICSEC). IEEE

  6. Agarwal M, Gupta SK, Biswas K (2021) Plant leaf disease segmentation using compressed UNet architecture. In: Pacific-Asia conference on knowledge discovery and data mining. Springer

  7. Saba L, et al (2021) Multimodality carotid plaque tissue characterization and classification in the artificial intelligence paradigm: a narrative review for stroke application. Ann Transl Med 9(14)

  8. Agarwal M, et al (2021) Wilson disease tissue classification and characterization using seven artificial intelligence models embedded with 3D optimization paradigm on a weak training brain magnetic resonance imaging datasets: a supercomputer application. Med Biol Eng Comput, p 1–23

  9. Agarwal M et al (2021) A novel block imaging technique using nine artificial intelligence models for COVID-19 disease classification, characterization and severity measurement in lung computed tomography scans on an italian cohort. J Med Syst 45(3):1–30

    Article  Google Scholar 

  10. Saba L et al (2021) Six artificial intelligence paradigms for tissue characterisation and classification of non-COVID-19 pneumonia against COVID-19 pneumonia in computed tomography lungs. Int J Comput Assist Radiol Surg 16(3):423–434

    Article  Google Scholar 

  11. Saba L, et al (2021) A multicenter study on carotid ultrasound plaque tissue characterization and classification using six deep artificial intelligence models: a stroke application. IEEE Trans Instrum Meas

  12. Nguyen H et al (2018) Deep learning methods in transportation domain: a review. IET Intel Transp Syst 12(9):998–1004

    Article  Google Scholar 

  13. Wang Y et al (2019) Enhancing transportation systems via deep learning: a survey. Transp Res Part C: Emerg Technol 99:144–163

    Article  Google Scholar 

  14. Veres M, Moussa M (2019) Deep learning for intelligent transportation systems: a survey of emerging trends. IEEE Trans Intell Transp Syst 21(8):3152–3168

    Article  Google Scholar 

  15. Sreenu G, Durai MS (2019) Intelligent video surveillance: a review through deep learning techniques for crowd analysis. J Big Data 6(1):1–27

    Article  Google Scholar 

  16. Malik J et al (2020) Hybrid deep learning: an efficient reconnaissance and surveillance detection mechanism in SDN. IEEE Access 8:134695–134706

    Article  Google Scholar 

  17. Kamilaris A, Prenafeta-Boldú FX (2018) Deep learning in agriculture: a survey. Comput Electron Agric 147:70–90

    Article  Google Scholar 

  18. Goodfellow I, Bengio Y, Courville A (2016) Deep learning. MIT Press

  19. Iandola FN, et al (2016) SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and< 0.5 MB model size. arXiv preprint arXiv:1602.07360

  20. Alom MZ, et al (2018) The history began from alexnet: a comprehensive survey on deep learning approaches. arXiv preprint arXiv:1803.01164

  21. Mateen M et al (2019) Fundus image classification using VGG-19 architecture with PCA and SVD. Symmetry 11(1):1

    Article  Google Scholar 

  22. Li Y et al (2018) Research on a surface defect detection algorithm based on MobileNet-SSD. Appl Sci 8(9):1678

    Article  Google Scholar 

  23. Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. In: Advances in neural information processing systems

  24. Acharya UR et al (2014) Evolutionary algorithm-based classifier parameter tuning for automatic ovarian cancer tissue characterization and classification. Ultraschall in der Medizin-Eur J Ultrasound 35(03):237–245

    Google Scholar 

  25. Shen F, Narayanan R, Suri JS (2008) Rapid motion compensation for prostate biopsy using GPU. In: 2008 30th Annual international conference of the IEEE Engineering in Medicine and Biology Society. IEEE

  26. Narayanan R et al (2008) Adaptation of a 3D prostate cancer atlas for transrectal ultrasound guided target-specific biopsy. Phys Med Biol 53(20):N397

    Article  Google Scholar 

  27. Sudeep P et al (2016) Speckle reduction in medical ultrasound images using an unbiased non-local means method. Biomed Signal Process Control 28:1–8

    Article  Google Scholar 

  28. Skandha SS, et al (2020) 3-D optimized classification and characterization artificial intelligence paradigm for cardiovascular/stroke risk stratification using carotid ultrasound-based delineated plaque: Atheromatic™ 2.0. Comput Biol Med 125: 103958

  29. Saba L, et al, Ultrasound-based internal carotid artery plaque characterization using deep learning paradigm on a supercomputer: a cardiovascular disease/stroke risk assessment system. Int J Cardiovasc Imaging, p 1–18

  30. Sanagala SS et al (2021) Ten fast transfer learning models for carotid ultrasound plaque tissue characterization in augmentation framework embedded with Heatmaps for stroke risk stratification. Diagnostics 11(11):2109

    Article  Google Scholar 

  31. Suri JS, et al (2021) COVLIAS 1.0 vs. MedSeg: artificial intelligence-based comparative study for automated COVID-19 computed tomography lung segmentation in Italian and Croatian Cohorts. Diagnostics 11(12): 2367

  32. Jain PK, et al (2021) Automated deep learning-based paradigm for high-risk plaque detection in B-mode common carotid ultrasound scans: an asymptomatic Japanese cohort study. Int Angiol: J Int Union Angiol 2021

  33. Saba L, et al (2021) Six artificial intelligence paradigms for tissue characterisation and classification of non-COVID-19 pneumonia against COVID-19 pneumonia in computed tomography lungs. Int J Comput Assist Radiol Surg, pp 1–12

  34. Jain PK et al (2021) Unseen artificial intelligence: deep learning paradigm for segmentation of low atherosclerotic plaque in carotid ultrasound: a multicenter cardiovascular study. Diagnostics 11(12):2257

    Article  Google Scholar 

  35. Skandha SS et al (2022) A hybrid deep learning paradigm for carotid plaque tissue characterization and its validation in multicenter cohorts using a supercomputer framework. Comput Biol Med 141:105131

    Article  Google Scholar 

  36. Suri J, et al (2021) Systematic review of artificial intelligence in acute respiratory distress syndrome for COVID-19 lung patients: a biomedical imaging perspective. IEEE J Biomed Health Inform

  37. Paul S et al (2022) Bias investigation in artificial intelligence systems for early detection of Parkinson’s disease: a narrative review. Diagnostics 12(1):166

    Article  Google Scholar 

  38. Das S, et al (2022) An artificial intelligence framework and its bias for brain tumor segmentation: a narrative review. Comput Biol Med, 105273

  39. Gong Y, et al (2014) Compressing deep convolutional networks using vector quantization. arXiv preprint arXiv:1412.6115

  40. Cheng Y, et al (2017) A survey of model compression and acceleration for deep neural networks. arXiv preprint arXiv:1710.09282

  41. Han S, Mao H, Dally WJ (2015) Deep compression: compressing deep neural networks with pruning, trained quantization and Huffman coding. arXiv preprint arXiv:1510.00149

  42. Luo J-H, Wu J, Lin W (2017) Thinet: a filter level pruning method for deep neural network compression. In: Proceedings of the IEEE international conference on computer vision

  43. Anwar S, Hwang K, Sung W (2017) Structured pruning of deep convolutional neural networks. ACM J Emerg Technol Comput Syst 13(3):1–18

    Article  Google Scholar 

  44. He Y, Zhang X, Sun J (2017) Channel pruning for accelerating very deep neural networks. In: Proceedings of the IEEE international conference on computer vision

  45. Han S, et al (2015) Learning both weights and connections for efficient neural networks. arXiv preprint arXiv:1506.02626

  46. Li H, et al (2016) Pruning filters for efficient convnets. arXiv preprint arXiv:1608.08710

  47. Liu Z, et al (2017) Learning efficient convolutional networks through network slimming. In: Proceedings of the IEEE international conference on computer vision

  48. Choudhary T et al (2020) A comprehensive survey on model compression and acceleration. Artif Intell Rev 53(7):5113–5155

    Article  Google Scholar 

  49. Zhang Q et al (2019) Recent advances in convolutional neural network acceleration. Neurocomputing 323:37–51

    Article  Google Scholar 

  50. Chen C-J, Chen K-C, Martin-Kuo M-C (2018) Acceleration of neural network model execution on embedded systems. In: 2018 International symposium on VLSI design, automation and test (VLSI-DAT). 2018. IEEE

  51. Cheng Y et al (2018) Model compression and acceleration for deep neural networks: the principles, progress, and challenges. IEEE Signal Process Mag 35(1):126–136

    Article  Google Scholar 

  52. Yang C, et al (2019) Multi-objective pruning for cnns using genetic algorithm. In: International conference on artificial neural networks. Springer

  53. Samala RK et al (2018) Evolutionary pruning of transfer learned deep convolutional neural network for breast cancer diagnosis in digital breast tomosynthesis. Phys Med Biol 63(9):095005

    Article  Google Scholar 

  54. Agarwal M, Gupta SK, Biswas K (2021) A compressed and accelerated SegNet for plant leaf disease segmentation: a differential evolution based approach. In: Pacific-Asia conference on knowledge discovery and data mining. Springer

  55. Goldberg DE, Holland JH (1988) Genetic algorithms and machine learning

  56. Sastry K, Goldberg D, Kendall G (2005) Genetic algorithms. In: Search methodologies. Springer, pp 97–125

  57. Goldberg DE (1989) Genetic algorithms in search, optimization, and machine learning. Addison. Reading

  58. Razali NM, Geraghty J (2011) Genetic algorithm performance with different selection strategies in solving TSP. In: Proceedings of the world congress on engineering. 2011. International Association of Engineers Hong Kong

  59. Vasconcelos J et al (2001) Improvements in genetic algorithms. IEEE Trans Magn 37(5):3414–3417

    Article  Google Scholar 

  60. Wright AH (1991) Genetic algorithms for real parameter optimization. In: Foundations of genetic algorithms, Elsevier, pp 205–218

  61. Schmitt LM (2001) Theory of genetic algorithms. Theoret Comput Sci 259(1–2):1–61

    Article  MathSciNet  MATH  Google Scholar 

  62. Srinivas M, Patnaik LM (1994) Genetic algorithms: a survey. Computer 27(6):17–26

    Article  Google Scholar 

  63. Safe M, et al (2004) On stopping criteria for genetic algorithms. In: Brazilian symposium on artificial intelligence. Springer

  64. Agarwal M, et al (2019) A convolution neural network based approach to detect the disease in corn crop. In: 2019 IEEE 9th international conference on advanced computing (IACC). IEEE

  65. Tandel GS, et al (2020) Multiclass magnetic resonance imaging brain tumor classification using artificial intelligence paradigm. Comput Biol Med, p 103804

  66. Rogers A, Prugel-Bennett A (1999) Genetic drift in genetic algorithm selection schemes. IEEE Trans Evol Comput 3(4):298–303

    Article  Google Scholar 

  67. LeCun Y et al (1998) Gradient-based learning applied to document recognition. Proc IEEE 86(11):2278–2324

    Article  Google Scholar 

  68. Krizhevsky A, Hinton G (2009) Learning multiple layers of features from tiny images

  69. Han F, et al, A texture feature analysis for diagnosis of pulmonary nodules using LIDC-IDRI database. In: 2013 IEEE international conference on medical imaging physics and engineering. IEEE

  70. Ullah I, Petrosino A (2016) About pyramid structure in convolutional neural networks. In: 2016 International joint conference on neural networks (IJCNN). IEEE

  71. Hu Y, et al (2018) A novel channel pruning method for deep neural network compression. arXiv preprint arXiv:1805.11394

  72. Yu R, et al (2018) Nisp: Pruning networks using neuron importance score propagation. In: Proceedings of the IEEE conference on computer vision and pattern recognition

  73. Li T, et al (2019) Compressing convolutional neural networks via factorized convolutional filters. In: Proceedings of the IEEE/CVF conference on computer vision and pattern recognition

  74. Mohan A, et al (2017) Internet of video things in 2030: a world with many cameras. In: 2017 IEEE international symposium on circuits and systems (ISCAS). IEEE

  75. Jain PK et al (2021) Hybrid deep learning segmentation models for atherosclerotic plaque in internal carotid artery B-mode ultrasound. Comput Biol Med 136:104721

    Article  Google Scholar 

  76. Suri JS, et al (2021) COVLIAS 1.0: lung segmentation in COVID-19 computed tomography scans using hybrid deep learning artificial intelligence models. Diagnostics 11(8):1405

  77. Jena B, et al (2021) Artificial intelligence-based hybrid deep learning models for image classification: the first narrative review. Comput Biol Med, p 104803

  78. Laine A, Sanches JM, Suri JS (2012) Ultrasound imaging: advances and applications. Springer

  79. Kuppili V et al (2017) Extreme learning machine framework for risk stratification of fatty liver disease using ultrasound tissue characterization. J Med Syst 41(10):152

    Article  Google Scholar 

  80. El-Baz AS, et al (2011) Multi modality state-of-the-art medical image segmentation and registration methodologies, vol. 1. Springer

  81. El-Baz A, Jiang X, Suri JS (20156) Biomedical image segmentation: advances and trends. CRC Press

  82. Acharya UR et al (2013) Understanding symptomatology of atherosclerotic plaque by image-based tissue characterization. Comput Methods Programs Biomed 110(1):66–75

    Article  Google Scholar 

  83. Shrivastava VK et al (2015) Reliable and accurate psoriasis disease classification in dermatology images using comprehensive feature space in machine learning paradigm. Expert Syst Appl 42(15–16):6184–6195

    Article  Google Scholar 

  84. Khalifa F, et al (2011) State-of-the-art medical image registration methodologies: a survey. In: Multi modality state-of-the-art medical image segmentation and registration methodologies, Springer, pp 235–280

  85. El-Baz A, Gimel’farb G, Suri JS (2015) Stochastic modeling for medical image analysis. CRC Press

  86. Acharya UR et al (2006) Heart rate variability: a review. Med Biol Eng Comput 44(12):1031–1051

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. CSE Department, CMR College of Engineering & Technology, Hyderabad, India

    Sanagala S. Skandha & Vijaya K. Koppula

  2. Department of CSE, Bennett University, Greater Noida, UP, India

    Sanagala S. Skandha, Mohit Agarwal, Kumar Utkarsh & Suneet K. Gupta

  3. Global Biomedical Technologies, Inc., Roseville, CA, USA

    Jasjit S. Suri

  4. Stroke Diagnostic and Monitoring Division, AtheroPoint™ LLC, Roseville, CA, 95661, USA

    Jasjit S. Suri

Authors
  1. Sanagala S. Skandha
    View author publications

    You can also search for this author inPubMed Google Scholar

  2. Mohit Agarwal
    View author publications

    You can also search for this author inPubMed Google Scholar

  3. Kumar Utkarsh
    View author publications

    You can also search for this author inPubMed Google Scholar

  4. Suneet K. Gupta
    View author publications

    You can also search for this author inPubMed Google Scholar

  5. Vijaya K. Koppula
    View author publications

    You can also search for this author inPubMed Google Scholar

  6. Jasjit S. Suri
    View author publications

    You can also search for this author inPubMed Google Scholar

Corresponding authors

Correspondence to Suneet K. Gupta or Jasjit S. Suri.

Ethics declarations

Conflict of interest

There is no conflict of interest.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor 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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Skandha, S.S., Agarwal, M., Utkarsh, K. et al. A novel genetic algorithm-based approach for compression and acceleration of deep learning convolution neural network: an application in computer tomography lung cancer data. Neural Comput & Applic 34, 20915–20937 (2022). https://doi.org/10.1007/s00521-022-07567-w

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s00521-022-07567-w

Keywords