Antlion re-sampling based deep neural network model for classification of imbalanced multimodal stroke dataset

• Published in: Multimedia tools and applications Vol. 81; no. 29; pp. 41429 - 41453
• Main Authors: G, Thippa Reddy, Bhattacharya, Sweta, Maddikunta, Praveen Kumar Reddy, Hakak, Saqib, Khan, Wazir Zada, Bashir, Ali Kashif, Jolfaei, Alireza, Tariq, Usman
• Format: Journal Article
• Language: English
• Published: New York Springer US 01.12.2022; Springer Nature B.V
• Subjects: Algorithms; Artificial neural networks; Computer Communication Networks; Computer Science; Data points; Data Structures and Information Theory; Datasets; Decision making; Homogeneity; Machine learning; Multimedia Information Systems; Neural networks; Optimization; Oversampling; Paralysis; Resampling; Special Purpose and Application-Based Systems; Stroke; Imbalanced dataset; Re-sampling; Deep neural networks; Antlion optimization; Stroke prediction
• DOI: 10.1007/s11042-020-09988-y
• ISSN: 1380-7501

Stroke is enlisted as one of the leading causes of death and serious disability affecting millions of human lives across the world with high possibilities of becoming an epidemic in the next few decades. Timely detection and prompt decision making pertinent to this disease, plays a major role which can reduce chances of brain death, paralysis and other resultant outcomes. Machine learning algorithms have been a popular choice for the diagnosis, analysis and predication of this disease but there exists issues related to data quality as they are collected cross-institutional resources. The present study focuses on improving the quality of stroke data implementing a rigorous pre-processing technique. The present study uses a multimodal stroke dataset available in the publicly available Kaggle repository. The missing values in this dataset are replaced with attribute means and LabelEncoder technique is applied to achieve homogeneity. However the dataset considered was observed to be imbalanced which reflect that the results may not represent the actual accuracy and would be biased. In order to overcome this imbalance, resampling technique was used. In case of oversampling, some data points in the minority class are replicated to increase the cardinality value and rebalance the dataset. transformed and oversampled data is further normalized using Standardscalar technique. Antlion optimization (ALO) algorithm is implemented on the deep neural network (DNN) model to select optimal hyperparameters in minimal time consumption. The proposed model consumed only 38.13% of the training time which was also a positive aspect. The experimental results proved the superiority of proposed model.