Skip to main content

Advertisement

Springer Nature Link
Log in

Improved machine learning performances with transfer learning to predicting need for hospitalization in arboviral infections against the small dataset

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

Abstract

The prediction of hospital patients and outpatients with suspected arboviral infection individuals in research-limited settings of the urban areas is defined as a challenging process for clinicians. Dengue, Chikungunya, and Zika arboviruses have gained attention in recent years because of the high prevalence in the society and financial burden of major global health systems. In this study, we proposed a machine learning algorithm based prediction model over retrospective medical records, which are named as SISA (the Severity Index for Suspected Arbovirus) and SISAL (the Severity Index for Suspected Arbovirus with Laboratory) datasets. Therefore, we aim to inform the clinicians about the use of machine learning with transfer learning success for diagnosis and comprehensive comparison of the classification performances over the SISA/SISAL datasets in the resource-limited settings that may cause to the small datasets of arboviral infection. In this study, Convolutional Neural Network and Long Short-Term Memory have achieved 100% accuracy and 1 of area under the curve (AUC) score, Fully Connected Deep Network has provided 92.86% accuracy and 0.969 AUC score in the SISAL dataset with transfer learning. Moreover, 98.73% accuracy and 0.988 AUC score were obtained by Convolutional Neural Network and Long Short-Term Memory for the SISA dataset. Furthermore, Linear Discriminant Analysis (shallow algorithm) has provided reaching up to 96.43% accuracy. Notably, deep learning based models have achieved improved performances compared to the previously reported study.

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

Similar content being viewed by others

Explore related subjects

Discover the latest articles and news from researchers in related subjects, suggested using machine learning.

References

  1. Sippyid R et al (2020) Severity index for suspected arbovirus (SISA): machine learning for accurate prediction of hospitalization in subjects suspected of arboviral infection. PLoS Negl Trop Dis 14(2):1–20

    Google Scholar 

  2. Kuniya T (2020) Prediction of the epidemic peak of coronavirus disease in Japan, 2020. J Clin Med 9(3):789

    Article  Google Scholar 

  3. Walczak S, Scorpio RJ, Pofahl WE (1998) Predicting hospital length of stay with neural networks. Proc Elev Int FLAIRS Conf 333–337

  4. Chauhan S, Vig L, Grazia DFDM, Corbetta M, Ahmad S, Zorzi M (2019) A comparison of shallow and deep learning methods for predicting cognitive performance of stroke patients from MRI lesion images. Front Neuroinform 13:1–12

    Article  Google Scholar 

  5. Zhong G, Ling X, Wang LN (2019) From shallow feature learning to deep learning: Benefits from the width and depth of deep architectures. Wiley Interdiscip Rev Data Min Knowl Discov 9(1):1–14

    Article  Google Scholar 

  6. Jagannatha AN, Yu H (2016) Bidirectional RNN for medical event detection in electronic health records. Proceedings of the 2016 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies 176(1):473–482.

  7. Alpaydın E (2010) Introduction to machine learning, Second Edition.

  8. Zhang Z, Wang C, Gan C, Sun S, Wang M (2019) Automatic Modulation classification using convolutional neural network with features fusion of SPWVD and BJD. IEEE Trans Signal Inf Process Over Netw 5(3):469–478

    Article  MathSciNet  Google Scholar 

  9. Ozer I, Ozer Z, Findik O (2018) Noise robust sound event classification with convolutional neural network. Neurocomputing 272:505–512

    Article  Google Scholar 

  10. Gorur K, Bozkurt M, Bascil M, Temurtas F (2019) GKP signal processing using deep CNN and SVM for tongue-machine interface. Trait du Signal 36(4):319–329

    Article  Google Scholar 

  11. Zhang L, Wang S, Liu B (2018) Deep learning for sentiment analysis: a survey. Wiley Interdiscip Rev Data Min Knowl Discov 8(4):1–25

    Article  Google Scholar 

  12. Zhang J, Zhu Y, Zhang X, Ye M, Yang J (2018) Developing a long short-term memory (LSTM) based model for predicting water table depth in agricultural areas. J Hydrol 561:918–929

    Article  Google Scholar 

  13. Bengio Y, Simard P, Frasconi P (1994) Learning long-term dependencies with gradient descent is difficult. IEEE Trans Neural Netw 5(2):157–166

    Article  Google Scholar 

  14. Xia J, Pan S, Yan M, Cai G, Yan J, Ning G (2020) Prognostic model of small sample critical diseases based on transfer learning. Sheng wu yi xue Gong Cheng xue za zhi= Journal of Biomedical Engineering= Shengwu Yixue Gongchengxue Zazhi 37(1):1–9

    Google Scholar 

  15. Gorur K, Bozkurt MR, Bascil MS, Temurtas F (2020) Comparative evaluation for PCA and ICA on tongue-machine interface using glossokinetic potential responses. Celal Bayar Univ J Sci 16(1):35–46

    Google Scholar 

  16. Delen D, Walker G, Kadam A (2005) Predicting breast cancer survivability: a comparison of three data mining methods. Artif Intell Med 34(2):113–127

    Article  Google Scholar 

  17. Temurtas H, Yumusak N, Temurtas F (2009) A comparative study on diabetes disease diagnosis using neural networks. Expert Syst Appl 36(4):8610–8615

    Article  Google Scholar 

  18. Cetin O, Temurtas F (2019) Classification of magnetoencephalography Signals regarding visual stimuli by generalized regression neural network. Dicle Tıp Derg 45(3):19–25

    Google Scholar 

  19. Yang S, Berdine G (2017) The receiver operating characteristic (ROC) curve. Southwest Respir Crit Care Chronicles 5(19):34

    Article  Google Scholar 

  20. Kwon JM, Lee Y, Lee Y, Lee S, Park J (2018) An algorithm based on deep learning for predicting in-hospital cardiac arrest. J Am Heart Assoc 7(13):1–11

    Article  Google Scholar 

  21. Krizhevsky A, Sutskever I, Hinton GE (2017) ImageNet classification with deep convolutional neural networks. Commun ACM 60(6):84–90

    Article  Google Scholar 

  22. Szegedy C et al (2015) Going deeper with convolutions. IEEE Conf Comput Vis Pattern Recogn (CVPR) 1–9

  23. Chopra C, Sinha S, Jaroli S, Shukla A, Maheshwari S (2017) Recurrent neural networks with non-sequential data to predict hospital readmission of diabetic patients. In: Proceedings of the 2017 International Conference on Computational Biology and Bioinformatics (pp. 18–23).

  24. Mao J, Xu W, Yang Y, Wang J, Huang Z, Yuille A (2014) Deep captioning with multimodal recurrent neural networks (m-rnn)

  25. Feng S, Zhou H, Dong H (2019) Using deep neural network with small dataset to predict material defects. Mater Des 162:300–310

    Article  Google Scholar 

  26. La L, Guo Q, Cao Q, Wang Y (2014) Transfer learning with reasonable boosting strategy. Neural Comput Appl 24:807–816

    Article  Google Scholar 

  27. Zhang Y, Ling C (2018) A strategy to apply machine learning to small datasets in materials science. Nat ComputMater 4:25

    Google Scholar 

  28. Li H, Barnaghi P et al. (2020) Continual learning using multi-view task conditional neural networks. Cornell University Computer Science (Machine Learning). Accessed 24 Nov 2020

  29. Scheck T, Grassi AP et al. (2020) A CNN-based feature space for semi-supervised incremental learning in assisted living applications. Cornell University Computer Science (Computer Vision and Pattern Recognition). Accessed 24 Nov 2020

  30. Baldwin T, Arora G et al. (2019) Does an LSTM forget more than a CNN? An empirical study of catastrophic forgetting in NLP. In: Proceedings of the 17th Workshop of the Australasian Language Technology Association

  31. Parisi IG, Kemker R et al (2019) Continual life long learning with neural networks: a review. Neural Netw 113:54–71

    Article  Google Scholar 

  32. Liang W, Yao J, Chen A et al (2020) Early triage of critically ill COVID-19 patients using deep learning. Nat Commun 11:3543

    Article  Google Scholar 

  33. Hilton CB, Milinovich A et al (2020) Personalized predictions of patient outcomes during and after hospitalization using artificial intelligence. Nat Dig Med 3:51

    Google Scholar 

  34. Fathima SA et al. (2012) Comparitive analysis of machine learning techniques for classification of arbovirus. IEEE Int Conf Biomed Health Inform 376–379

  35. Cle M et al (2020) Neurocognitive impacts of arbovirus infections. J Neuroinflam 17(1):233

    Article  Google Scholar 

Download references

Funding

There is no funding for this research study.

Author information

Authors and Affiliations

  1. Computer Engineering, Bandirma Onyedi Eylul University, Balikesir, 10200, Turkey

    Ilyas Ozer

  2. Electrical and Electronics Engineering, Bandirma Onyedi Eylul University, Balikesir, 10200, Turkey

    Onursal Cetin, Kutlucan Gorur & Feyzullah Temurtas

Authors
  1. Ilyas Ozer
    View author publications

    You can also search for this author inPubMed Google Scholar

  2. Onursal Cetin
    View author publications

    You can also search for this author inPubMed Google Scholar

  3. Kutlucan Gorur
    View author publications

    You can also search for this author inPubMed Google Scholar

  4. Feyzullah Temurtas
    View author publications

    You can also search for this author inPubMed Google Scholar

Corresponding author

Correspondence to Kutlucan Gorur.

Ethics declarations

Conflicts of interest

The authors declare that they have 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

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Ozer, I., Cetin, O., Gorur, K. et al. Improved machine learning performances with transfer learning to predicting need for hospitalization in arboviral infections against the small dataset. Neural Comput & Applic 33, 14975–14989 (2021). https://doi.org/10.1007/s00521-021-06133-0

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s00521-021-06133-0

Keywords