A novel machine learning-based feature extraction method for classifying intracranial hemorrhage computed tomography images

Abstract

One of the most serious forms of brain stroke is intracranial hemorrhage (ICH). When an artery bursts, the brain and the tissue around the artery start bleeding. This study proposes a joint feature selection strategy to classify computed tomography (CT) images of intracranial hemorrhage. The joint feature set is composed of transform and texture features. Joint features are constructed from a combination of grey level co-occurrence matrix (GLCM) features, discrete wavelet features (DWT), and discrete cosine features (DCT). Brain hemorrhage CT image classification uses ensemble-based machine learning (ML) techniques. On the training dataset, a Synthetic Minority Over-Sampling Technique (SMOTE) is applied to treat the problem of oversampling by adding fresh data. Additionally, the sequential forward feature selection technique is used to obtain feature subsets. The classification accuracy is further examined for varied feature vector sizes. Confusion matrix, precision, and recall in categorization are employed as performance evaluation measurements. The ML-based ensemble classifiers can produce highly accurate results with the aid of the proposed novel feature extraction mechanism. When taking into consideration a crucial feature set consisting of six features, it can be seen that Random Forest obtained the greatest accuracy, which is 87.22%.