An Intelligent Stroke Detection Framework Utilizing Genetic Algorithm-Based Feature Selection and Bidirectional LSTM on CT Imagery
iacs CAI

Computing and Algorithm Insight

Computing and Algorithm Insight is a peer-reviewed journal publishing research in artificial intelligence, soft...

Publishing Model

Open Access
This journal published by Integra Academic Press

Abstract

Cerebrovascular pathologies, particularly strokes, are a leading cause of global mortality and morbidity, where timely diagnosis is critical for effective clinical intervention. This research introduces an advanced diagnostic framework for early stroke identification by integrating Computed Tomography (CT) imaging with a hybrid computational approach. The system utilizes a Genetic Algorithm (GA) to optimize feature selection from neural network outputs, identifying the most discriminative biomarkers within the image data. These refined features are subsequently classified using a Bidirectional Long Short-Term Memory (BiLSTM) network. Rigorous validation through cross-validation showed that the proposed model achieved a 96.5% accuracy, significantly outperforming traditional classifiers like Random Forests and Support Vector Machines. By utilizing metrics such as AUC-ROC and F1-score, the study demonstrates that this architecture serves as a robust decision-support tool, enabling medical practitioners to implement more precise and rapid stroke management strategies.

Keywords: Stroke Detection Genetic Algorithm Bidirectional LSTM CT Image Classification Feature Selection


References

World Stroke Organization. (2023). Vision Impairment and Blindness. Accessed: Feb. 3, 2023. [Online]. Available: world-stroke.org.

A. Khosla, et al., “An integrated machine learning approach to stroke prediction,” in Proc. 16th ACM SIGKDD Int. Conf. Knowl. Discovery Data Mining, Jul. 2010, pp. 183–192.

Centers for Disease Control Prevention. (2023). Stroke Facts. Accessed: Mar. 25, 2023. [Online]. Available: cdc.gov.

Y. Xu, et al., “Role of ferroptosis in stroke,” Cellular Mol. Neurobiol., vol. 43, no. 1, pp. 205–222, Jan. 2023.

X. Li and G. Chen, “CNS-peripheral immune interactions in hemorrhagic stroke,” J. Cerebral Blood Flow Metabolism, vol. 43, no. 2, pp. 185–197, Feb. 2023.

Y. Yang, et al., “Accurate prediction of stroke for hypertensive patients based on medical big data and machine learning algorithms: Retrospective study,” JMIR Med. Informat., vol. 9, no. 11, Nov. 2021, Art. no. e30277.

M. Chun, et al., “Stroke risk prediction using machine learning: A prospective cohort study of 0.5 million Chinese adults,” J. Amer. Med. Inform. Assoc., vol. 28, no. 8, pp. 1719–1727, Jul. 2021.

G. Ntaios, et al., “Machine-learning-derived model for the stratification of cardiovascular risk in patients with ischemic stroke,” J. Stroke Cerebrovascular Diseases, vol. 30, no. 10, Oct. 2021, Art. no. 106018.

K. Uchida, et al., “Development of machine learning models to predict probabilities and types of stroke at prehospital stage: The Japan urgent stroke triage score using machine learning (JUST-ML),” Transl. Stroke Res., vol. 13, no. 3, pp. 370–381, Jun. 2022.

H. Z. U. Rehman, et al., “Conventional and deep learning methods for skull stripping in brain MRI,” Appl. Sci., vol. 10, no. 5, p. 1773, Mar. 2020.

M. Lim, et al., “Link prediction in time-evolving criminal network with deep reinforcement learning technique,” IEEE Access, vol. 7, pp. 184797–184807, 2019.

M. Lim, et al., “Hidden link prediction in criminal networks using the deep reinforcement learning technique,” Computers, vol. 8, no. 1, p. 8, Jan. 2019.

S. Bacchi, et al., “Deep learning in the prediction of ischaemic stroke thrombolysis functional outcomes,” Academic Radiol., vol. 27, no. 2, pp. e19–e23, Feb. 2020.

M. Saritha, et al., “Classification of MRI brain images using combined wavelet entropy based spider web plots and probabilistic neural network,” Pattern Recognit. Lett., vol. 34, no. 16, pp. 2151–2156, Dec. 2013.

S. Dev, et al., “A predictive analytics approach for stroke prediction using machine learning and neural networks,” Healthcare Anal., vol. 2, Nov. 2022, Art. no. 100032.

L. Ali, et al., “An automated diagnostic system for heart disease prediction based on chi square statistical model and optimally configured deep neural network,” IEEE Access, vol. 7, pp. 34938–34945, 2019.

A. Phaphuangwittayakul, et al., “An optimal deep learning framework for multi-type hemorrhagic lesions detection and quantification in head CT images for traumatic brain injury,” Int. J. Speech Technol., vol. 52, no. 7, pp. 7320–7338, May 2022.

S. Gudadhe and A. Thakare, “Classification of intracranial hemorrhage CT images for stroke analysis with transformed and image-based GLCM features,” Tech. Rep., 2022.

S. Gudadhe, et al., “A novel machine learning-based feature extraction method for classifying intracranial hemorrhage computed tomography images,” Healthcare Anal., vol. 3, Nov. 2023, Art. no. 100196.

M. Yeo, et al., “Evaluation of techniques to improve a deep learning algorithm for the automatic detection of intracranial haemorrhage on CT head imaging,” Eur. Radiol. Experim., vol. 7, no. 1, p. 17, Apr. 2023.

K. Khosravi, et al., “Soil water erosion susceptibility assessment using deep learning algorithms,” J. Hydrol., vol. 618, Mar. 2023, Art. no. 129229.

J. Donnelly, et al., “Gaussian process emulation of spatio-temporal outputs of a 2D inland flood model,” Water Res., vol. 225, Oct. 2022, Art. no. 119100.

E. Rasool, et al., “Breast microcalcification detection in digital mammograms using deep transfer learning approaches,” in Proc. 9th Int. Conf. Comput. Data Eng., Jan. 2023, pp. 58–65.

J. Donnelly, et al., “Forecasting global climate drivers using Gaussian processes and convolutional autoencoders,” Eng. Appl. Artif. Intell., vol. 128, Feb. 2024, Art. no. 107536.

M. A. Saleem, et al., “Sooty tern optimization algorithm-based deep learning model for diagnosing NSCLC tumours,” Sensors, vol. 23, no. 4, p. 2147, Feb. 2023.

B. Ghiasi, et al., “Uncertainty quantification of granular computing-neural network model for prediction of pollutant longitudinal dispersion coefficient in aquatic streams,” Sci. Rep., vol. 12, no. 1, p. 4610, Mar. 2022.

S. Sindhiya and S. Gunasundari, “A survey on genetic algorithm based feature selection for disease diagnosis system,” in Proc. IEEE Int. Conf. Comput. Commun. Syst., Feb. 2014, pp. 164–169.

A. Javeed, et al., “Predicting dementia risk factors based on feature selection and neural networks,” Comput., Mater. Continua, vol. 75, no. 2, pp. 2491–2508, 2023.

P. Alirezazadeh, et al., “A genetic algorithm-based feature selection for kinship verification,” IEEE Signal Process. Lett., vol. 22, no. 12, pp. 2459–2463, Dec. 2015.

R. Noori, et al., “An efficient data driven-based model for prediction of the total sediment load in rivers,” Hydrology, vol. 9, no. 2, p. 36, Feb. 2022.

A. Javeed, et al., “Predictive power of XGBoost_BiLSTM model: A machine-learning approach for accurate sleep apnea detection using electronic health data,” Int. J. Comput. Intell. Syst., vol. 16, no. 1, p. 188, Nov. 2023.

A. Mousa and B. Schuller, “Contextual bidirectional long short-term memory recurrent neural network language models: A generative approach to sentiment analysis,” Tech. Rep., 2017.

A. Javeed, et al., “Early prediction of dementia using feature extraction battery (FEB) and optimized support vector machine (SVM) for classification,” Biomedicines, vol. 11, no. 2, p. 439, Feb. 2023.

S. Yadav and S. Shukla, “Analysis of k-Fold cross-validation over hold-out validation on colossal datasets for quality classification,” in Proc. IEEE 6th Int. Conf. Adv. Comput. (IACC), Feb. 2016, pp. 78–83.

A. Javeed, et al., “Breaking barriers: A statistical and machine learning-based hybrid system for predicting dementia,” Frontiers Bioeng. Biotechnol., vol. 11, Jan. 2024, Art. no. 1336255.

A. Javeed, et al., “An intelligent learning system for unbiased prediction of dementia based on autoencoder and AdaBoost ensemble learning,” Life, vol. 12, no. 7, p. 1097, Jul. 2022.

K. Pal and B. V. Patel, “Data classification with k-fold cross validation and holdout accuracy estimation methods with 5 different machine learning techniques,” in Proc. 4th Int. Conf. Comput. Methodologies Commun. (ICCMC), Mar. 2020, pp. 83–87.

A. Javeed, et al., “A clinical decision support system (CDSS) for unbiased prediction of caesarean section based on features extraction and optimized classification,” Comput. Intell. Neurosci., vol. 2022, pp. 1–13, Jun. 2022.

(2021). Brain Stroke CT Image Dataset. Accessed: Jul. 3, 2023. [Online]. Available: kaggle.com.

J. O. Westgard and J. A. Lott, “Precision and accuracy: Concepts and assessment by method evaluation testing,” CRC Crit. Rev. Clin. Lab. Sci., vol. 13, no. 4, pp. 283–330, Jan. 1981.

H. Dalianis, “Evaluation metrics and evaluation,” in Clinical Text Mining: Secondary Use of Electronic Patient Records, 2018, pp. 45–53.

L. Ali, et al., “Parkinson's disease detection based on features refinement through L1 regularized SVM and deep neural network,” Sci. Rep., vol. 14, no. 1, p. 1333, Jan. 2024.

A. Javeed, et al., “Decision support system for predicting mortality in cardiac patients based on machine learning,” Appl. Sci., vol. 13, no. 8, p. 5188, Apr. 2023.

X. Zhang, et al., “The use of ROC and AUC in the validation of objective image fusion evaluation metrics,” Signal Process., vol. 115, pp. 38–48, Oct. 2015.

L. Ali, et al., “Reliable Parkinson's disease detection by analyzing handwritten drawings: Construction of an unbiased cascaded learning system based on feature selection and adaptive boosting model,” IEEE Access, vol. 7, pp. 116480–116489, 2019.

T. Fawcett, “An introduction to ROC analysis,” Pattern Recognit. Lett., vol. 27, no. 8, pp. 861–874, Jun. 2006.

G. Sailasya and G. L. A. Kumari, “Analyzing the performance of stroke prediction using ML classification algorithms,” Int. J. Adv. Comput. Sci. Appl., vol. 12, no. 6, 2021.

B. Akter, et al., “A machine learning approach to detect the brain stroke disease,” in Proc. 4th Int. Conf. Smart Syst. Inventive Technol. (ICSSIT), Jan. 2022, pp. 897–901.

A. Tursynova, et al., “Deep learning-enabled brain stroke classification on computed tomography images,” Comput., Mater. Continua, vol. 75, no. 1, pp. 1431–1446, 2023.

M. Gupta, et al., “Predicting brain stroke using IoT-enabled deep learning and machine learning: Advancing sustainable healthcare,” in Proc. Int. Conf. Sustain. Develop. Mach. Learn., AI IoT. Cham, Switzerland: Springer, 2023, pp. 113–122.

L. Cortés-Ferre, et al., “Deep learning applied to intracranial hemorrhage detection,” J. Imag., vol. 9, no. 2, p. 37, Feb. 2023.