Brain Disease Classification Research Articles

Comparison of Machine Learning Models for Brain Disease Classification Based on CT and MRI Scans

The complexity and severity of brain diseases have led to increased focus on their diagnosis and treatment. Due to the inherent drawbacks of manual medical diagnosis, such as error-prone and costly, and the recent widespread use of Artificial Intelligence (AI) in medicine, it is a worthy topic to explore machine learning in diagnosing brain diseases utilizing Computed Tomography (CT) and Magnetic Resonance Imaging (MRI). In this paper, three brain diseases using CT or MRI datasets from Kaggle were merged and processed, and then three models of decision tree (DT), random forest (RF) and K-Nearest Neighbors (KNN) were used to make classification prediction. Furthermore, their performance was compared through classification reports and confusion matrix to analysis results. The results showed that DT performed worse than the other two models in this task, with an accuracy of 0.91, whereas RF and KNN performed similar overall, each achieving an overall accuracy of 0.96. Notably, RF exhibited less confusion, especially between some similar categories, which indicates the ability of handle complex data more effectively. Additionally, the strengths and weaknesses of the three models are discussed in the paper. These experimental results show that machine learning model, particularly RF and KNN, is a useful tool for diagnosing brain diseases with high accuracy, which could substantially assist clinical practices by providing reliable and efficient diagnostic support.

CNN based Detection and Classification of Brain Diseases

CNN based Detection and Classification of Brain Diseases

A Siamese network optimization using genetic algorithm for brain diseases

AbstractThe complex nature of human brain tissues is important in ensuring accurate diagnosis to save human lives. Research on early detection of brain diseases has gained significant prominence within medical intelligence using highly complex model architectures with only a single label that cannot be verified. This paper introduces an innovative approach to Siamese Network Genetic Algorithm (SN‐GA) leveraging Siamese contrastive learning for classifying brain images across diverse diseases. Our core architecture is a Bi‐Convolutional Neural Network (Bi‐CNN) optimized by a genetic algorithm to enhance brain image classification. Specifically, five widely recognized transfer learning‐based architectures, namely AlexNet, Efficient‐B0, VGG‐16, ResNet‐50, and Inception‐v3, have been incorporated to evaluate the effectiveness of the proposed SN‐GA system. The performance of these models has been rigorously analyzed and compared using two distinct datasets: Brain tumors and Alzheimer's datasets. The experimental results robustly affirm the efficacy of the proposed Siamese model, yielding exceptional levels of accuracy, precision, and recall, all peaking at 97%. These findings underscore the potential and resilience of the optimized Siamese network in the context of brain disease classification, emphasizing its significance in advancing the field of medical imaging and diagnosis, with implications for early intervention and patient care.

The Use of Generative Adversarial Network and Graph Convolution Network for Neuroimaging-Based Diagnostic Classification.

Functional connectivity (FC) obtained from resting-state functional magnetic resonance imaging has been integrated with machine learning algorithms to deliver consistent and reliable brain disease classification outcomes. However, in classical learning procedures, custom-built specialized feature selection techniques are typically used to filter out uninformative features from FC patterns to generalize efficiently on the datasets. The ability of convolutional neural networks (CNN) and other deep learning models to extract informative features from data with grid structure (such as images) has led to the surge in popularity of these techniques. However, the designs of many existing CNN models still fail to exploit the relationships between entities of graph-structure data (such as networks). Therefore, graph convolution network (GCN) has been suggested as a means for uncovering the intricate structure of brain network data, which has the potential to substantially improve classification accuracy. Furthermore, overfitting in classifiers can be largely attributed to the limited number of available training samples. Recently, the generative adversarial network (GAN) has been widely used in the medical field for its generative aspect that can generate synthesis images to cope with the problems of data scarcity and patient privacy. In our previous work, GCN and GAN have been designed to investigate FC patterns to perform diagnosis tasks, and their effectiveness has been tested on the ABIDE-I dataset. In this paper, the models will be further applied to FC data derived from more public datasets (ADHD, ABIDE-II, and ADNI) and our in-house dataset (PTSD) to justify their generalization on all types of data. The results of a number of experiments show the powerful characteristic of GAN to mimic FC data to achieve high performance in disease prediction. When employing GAN for data augmentation, the diagnostic accuracy across ADHD-200, ABIDE-II, and ADNI datasets surpasses that of other machine learning models, including results achieved with BrainNetCNN. Specifically, in ADHD, the accuracy increased from 67.74% to 73.96% with GAN, in ABIDE-II from 70.36% to 77.40%, and in ADNI, reaching 52.84% and 88.56% for multiclass and binary classification, respectively. GCN also obtains decent results, with the best accuracy in ADHD datasets at 71.38% for multinomial and 75% for binary classification, respectively, and the second-best accuracy in the ABIDE-II dataset (72.28% and 75.16%, respectively). Both GAN and GCN achieved the highest accuracy for the PTSD dataset, reaching 97.76%. However, there are still some limitations that can be improved. Both methods have many opportunities for the prediction and diagnosis of diseases.

Graph Convolutional Network with Self-supervised Learning for Brain Disease Classification.

Brain functional network (BFN) analysis has become a popular method for identifying neurological diseases at their early stages and revealing sensitive biomarkers related to these diseases. Due to the fact that BFN is a graph with complex structure, graph convolutional networks (GCNs) can be naturally used in the identification of BFN, and can generally achieve an encouraging performance if given large amounts of training data. In practice, however, it is very difficult to obtain sufficient brain functional data, especially from subjects with brain disorders. As a result, GCNs usually fail to learn a reliable feature representation from limited BFNs, leading to overfitting issues. In this paper, we propose an improved GCN method to classify brain diseases by introducing a self-supervised learning (SSL) module for assisting the graph feature representation. We conduct experiments to classify subjects with mild cognitive impairment (MCI) and autism spectrum disorder (ASD) respectively from normal controls (NCs). Experimental results on two benchmark databases demonstrate that our proposed scheme tends to obtain higher classification accuracy than the baseline methods.

LCGNet: Local Sequential Feature Coupling Global Representation Learning for Functional Connectivity Network Analysis with fMRI.

Analysis of functional connectivity networks (FCNs) derived from resting-state functional magnetic resonance imaging (rs-fMRI) has greatly advanced our understanding of brain diseases, including Alzheimer's disease (AD) and attention deficit hyperactivity disorder (ADHD). Advanced machine learning techniques, such as convolutional neural networks (CNNs), have been used to learn high-level feature representations of FCNs for automated brain disease classification. Even though convolution operations in CNNs are good at extracting local properties of FCNs, they generally cannot well capture global temporal representations of FCNs. Recently, the transformer technique has demonstrated remarkable performance in various tasks, which is attributed to its effective self-attention mechanism in capturing the global temporal feature representations. However, it cannot effectively model the local network characteristics of FCNs. To this end, in this paper, we propose a novel network structure for Local sequential feature Coupling Global representation learning (LCGNet) to take advantage of convolutional operations and self-attention mechanisms for enhanced FCN representation learning. Specifically, we first build a dynamic FCN for each subject using an overlapped sliding window approach. We then construct three sequential components (i.e., edge-to-vertex layer, vertex-to-network layer, and network-to-temporality layer) with a dual backbone branch of CNN and transformer to extract and couple from local to global topological information of brain networks. Experimental results on two real datasets (i.e., ADNI and ADHD-200) with rs-fMRI data show the superiority of our LCGNet.

Brain tumor classification by CNN

In this paper, we proposed a disease classification method for brain images using CNN. The research dataset collects images of normal, blastoma, meningioma, adenoma, and glioblastoma by brain tumor disease from journals such as NEJM and AuntMinnie and files them in Exam_Brain.Zip. As a result of CNN processing the images for each file folder in the Exam_Brain.zip file 10 times, the AUC was found to be 0.8825. As a result of the experiment, the brain disease classification accuracy of brain magnetic resonance imaging was found to be 88.25%. These results indicate that the results of this study can be used to classify other diseases once the data set is established, and can also be used to classify objects in other industries.This study has the following limitations. A challenge in automatically classifying diseases is the vastness and integrity of the data sets. If the data is not prepared, the accuracy of classification will be problematic. Additionally, in the medical field, a doctor's decision-making (diagnosis) is not limited to image data but is made based on comprehensive data, and there is a limitation in that medical decisions cannot be made by machines without the intervention of a doctor. Determining the decision threshold in AI decision-making problems is also an important limitation. This is because setting reference points for sensitivity and specificity is not the domain of AI experts, but the domain of experts in the field. Therefore, field medical staff must participate in the process of developing AI in medical settings. Future research tasks include developing applications that improve performance by developing the input stage of the program, and continuing research in connection with medical big data servers.

Utilizing deep learning via the 3D U-net neural network for the delineation of brain stroke lesions in MRI image

The segmentation of acute stroke lesions plays a vital role in healthcare by assisting doctors in making prompt and well-informed treatment choices. Although Magnetic Resonance Imaging (MRI) is a time-intensive procedure, it produces high-fidelity images widely regarded as the most reliable diagnostic tool available. Employing deep learning techniques for automated stroke lesion segmentation can offer valuable insights into the precise location and extent of affected tissue, enabling medical professionals to effectively evaluate treatment risks and make informed assessments. In this research, a deep learning approach is introduced for segmenting acute and sub-acute stroke lesions from MRI images. To enhance feature learning through brain hemisphere symmetry, pre-processing techniques are applied to the data. To tackle the class imbalance challenge, we employed a strategy of using small patches with balanced sampling during training, along with a dynamically weighted loss function that incorporates f1-score and IOU-score (Intersection over Union). Furthermore, the 3D U-Net architecture is used to generate predictions for complete patches, employing a high degree of overlap between patches to minimize the requirement for subsequent post-processing steps. The 3D U-Net model, utilizing ResnetV2 as the pre-trained encoder for IOU-score and Seresnext101 for f1-score, stands as the leading state-of-the-art (SOTA) model for segmentation tasks. However, recent research has introduced a novel model that surpasses these metrics and demonstrates superior performance compared to other backbone architectures. The f1-score and IOU-score were computed for various backbones, with Seresnext101 achieving the highest f1-score and ResnetV2 performing the highest IOU-score. These calculations were conducted using a threshold value of 0.5. This research proposes a valuable model based on transfer learning for the classification of brain diseases in MRI scans. The achieved f1-score using the recommended classifiers demonstrates the effectiveness of the approach employed in this study. The findings indicate that Seresnext101 attains the highest f1-score of 0.94226, while ResnetV2 achieves the best IOU-score of 0.88342, making it the preferred architecture for segmentation methods. Furthermore, the study presents experimental results of the 3D U-Net model applied to brain stroke lesion segmentation, suggesting prospects for researchers interested in segmenting brain strokes and enhancing 3D U-Net models.

Automated classification of brain diseases using the Restricted Boltzmann Machine and the Generative Adversarial Network

Background:Early diagnosis of brain diseases is very important. Brain disease classification is a common and complex topic in biomedical engineering. Therefore, machine learning methods are the most reliable and common methods for the automatic detection of brain diseases. Materials and methods:The brain diseases dataset consists of four classes: atrophy, normal, white matter density, and ischemia. First, data augmentation and normalization is applied to the dataset. Then, deep feature extraction is applied to this preprocessed dataset with the Restricted Boltzmann Machine (RBM) method. Deep feature data are given as input to the generator unit of the Generative Adversarial Network (GAN) method. Finally, classification is performed by Tree, Linear discriminant, Naïve Bayes, Support Vector Machine (SVM), K-Nearest Neighbors (KNN), Ensemble, Neural Network classifiers and K-mean. Results:In the classification applied before feature extraction with the RBM method, the highest accuracy is calculated in the SVM classifier with 97.94%, and the lowest accuracy is calculated in the Naïve Bayes classifier with 80.43%. After applying the RBM feature extraction, the highest accuracy is calculated as 99.65% in the SVM classifier and 83.74% in the lowest Naïve Bayes classifier. Conclusion:This study shows that the performance criteria values of the presented methods is improved with RBM-GAN.

Optimal transport based pyramid graph kernel for autism spectrum disorder diagnosis

Brain network, which characterizes the functional and structural interactions of brain regions with graph theory, has been widely utilized to diagnose brain diseases, such as autism spectrum disorder (ASD). It is a challenge to measure the network (or graph) similarity in brain network analysis. Graph kernel (i.e., kernel defined on graphs) offers an efficient tool for measuring the similarity of paired brain networks and yields the excellent classification performance in brain disease diagnosis. However, most of the existing graph kernels neglected the hierarchical architecture information of brain networks. To address this problem, in this paper, we propose an optimal transport based pyramid graph kernel for measuring brain network similarity and then apply it to brain disease classification. The main idea is to transform brain networks into pyramid structures, which reflect the hierarchical architecture information of the brain network with multi-resolution histograms. The optimal transport distance in pyramid structures is calculated for measuring transport costs between paired brain networks. Finally, the optimal transport based pyramid graph kernel is computed based on this optimal transport distance. To evaluate the effectiveness of the proposed optimal transport based pyramid graph kernel, the extensive experiments are performed in functional magnetic resonance imaging data of brain disease from the Autism Brain Imaging Data Exchange database. The experimental results show that our proposed optimal transport based pyramid graph kernel outperforms the state-of-the-art methods in ASD classification tasks.

Deep-Stacked Convolutional Neural Networks for Brain Abnormality Classification Based on MRI Images.

An automated diagnosis system is crucial for helping radiologists identify brain abnormalities efficiently. The convolutional neural network (CNN) algorithm of deep learning has the advantage of automated feature extraction beneficial for an automated diagnosis system. However, several challenges in the CNN-based classifiers of medical images, such as a lack of labeled data and class imbalance problems, can significantly hinder the performance. Meanwhile, the expertise of multiple clinicians may be required to achieve accurate diagnoses, which can be reflected in the use of multiple algorithms. In this paper, we present Deep-Stacked CNN, a deep heterogeneous model based on stacked generalization to harness the advantages of different CNN-based classifiers. The model aims to improve robustness in the task of multi-class brain disease classification when we have no opportunity to train single CNNs on sufficient data. We propose two levels of learning processes to obtain the desired model. At the first level, different pre-trained CNNs fine-tuned via transfer learning will be selected as the base classifiers through several procedures. Each base classifier has a unique expert-like character, which provides diversity to the diagnosis outcomes. At the second level, the base classifiers are stacked together through neural network, representing the meta-learner that best combines their outputs and generates the final prediction. The proposed Deep-Stacked CNN obtained an accuracy of 99.14% when evaluated on the untouched dataset. This model shows its superiority over existing methods in the same domain. It also requires fewer parameters and computations while maintaining outstanding performance.

The effect of node features on GCN-based brain network classification: an empirical study.

Brain functional network (BFN) analysis has become a popular technique for identifying neurological/mental diseases. Due to the fact that BFN is a graph, a graph convolutional network (GCN) can be naturally used in the classification of BFN. Different from traditional methods that directly use the adjacency matrices of BFNs to train a classifier, GCN requires an additional input-node features. To our best knowledge, however, there is no systematic study to analyze their influence on the performance of GCN-based brain disorder classification. Therefore, in this study, we conduct an empirical study on various node feature measures, including (1) original fMRI signals, (2) one-hot encoding, (3) node statistics, (4) node correlation, and (5) their combination. Experimental results on two benchmark databases show that different node feature inputs to GCN significantly affect the brain disease classification performance, and node correlation usually contributes higher accuracy compared to original signals and manually extracted statistical features.

Nonlinear Feature Extraction Methods Based on Dual-Tree Complex Wavelet Transform Subimages of Brain Magnetic Resonance Imaging for the Classification of Multiple Diseases.

It has been a long time since we use magnetic resonance imaging (MRI) to detect brain diseases and many useful techniques have been developed for this task. However, there is still a potential for further improvement of classification of brain diseases in order to be sure of the results. In this research we presented, for the first time, a non-linear feature extraction method from the MRI sub-images that are obtained from the three levels of the two-dimensional Dual tree complex wavelet transform (2D DT-CWT) in order to classify multiple brain disease. After extracting the non-linear features from the sub-images, we used the spectral regression discriminant analysis (SRDA) algorithm to reduce the classifying features. Instead of using the deep neural networks that are computationally expensive, we proposed the Hybrid RBF network that uses the k-means and recursive least squares (RLS) algorithm simultaneously in its structure for classification. To evaluate the performance of RBF networks with hybrid learning algorithms, we classify nine brain diseases based on MRI processing using these networks, and compare the results with the previously presented classifiers including, supporting vector machines (SVM) and K-nearest neighbour (KNN). Comprehensive comparisons are made with the recently proposed cases by extracting various types and numbers of features. Our aim in this paper is to reduce the complexity and improve the classifying results with the hybrid RBF classifier and the results showed 100 percent classification accuracy in both the two class and the multiple classification of brain diseases in 8 and 10 classes. In this paper, we provided a low computational and precise method for brain MRI disease classification. the results show that the proposed method is not only accurate but also computationally reasonable.

The classification of brain network for major depressive disorder patients based on deep graph convolutional neural network.

The early diagnosis of major depressive disorder (MDD) is very important for patients that suffer from severe and irreversible consequences of depression. It has been indicated that functional connectivity (FC) analysis based on functional magnetic resonance imaging (fMRI) data can provide valuable biomarkers for clinical diagnosis. However, previous studies mainly focus on brain disease classification in small sample sizes, which may lead to dramatic divergences in classification accuracy. This paper attempts to address this limitation by applying the deep graph convolutional neural network (DGCNN) method on a large multi-site MDD dataset. The resting-state fMRI data are acquired from 830 MDD patients and 771 normal controls (NC) shared by the REST-meta-MDD consortium. The DGCNN model trained with the binary network after thresholding, identified MDD patients from normal controls and achieved an accuracy of 72.1% with 10-fold cross-validation, which is 12.4%, 9.8%, and 7.6% higher than SVM, RF, and GCN, respectively. Moreover, the process of dataset reading and model training is faster. Therefore, it demonstrates the advantages of the DGCNN model with low time complexity and sound classification performance. Based on a large, multi-site dataset from MDD patients, the results expressed that DGCNN is not an extremely accurate method for MDD diagnosis. However, there is an improvement over previous methods with our goal of better understanding brain function and ultimately providing a biomarker or diagnostic capability for MDD diagnosis.

Machine learning and artificial neural network for data mining classification and prediction of brain diseases

Machine learning and artificial neural network for data mining classification and prediction of brain diseases

Machine Learning and Artificial Neural Network for Data Mining Classification and Prediction of Brain Diseases

Machine Learning and Artificial Neural Network for Data Mining Classification and Prediction of Brain Diseases

Bio-Inspired Feature Selection in Brain Disease Detection via an Improved Sparrow Search Algorithm

The timely diagnosis and treatment of brain diseases have always been an essential part of saving patients with encephalopathy. At the same time, medical image analysis is a significant step in diagnosing brain diseases. This article proposes a brain disease classification approach based on an improved sparrow search algorithm (ISSA). Specifically, a set of image features is first extracted to compose a pool of features for selection from five kinds of brain hemorrhage image datasets and one of the brain tumor image dataset. Second, we design an objective function to jointly reduce the number of the selected features and improve the classification accuracy. Finally, the proposed ISSA is utilized to solve the objective function. ISSA introduces three improvement mechanisms: the tent chaotic initialization, a novel local search strategy, and adaptive crossover operation to enhance the performance of conventional SSA. Moreover, a binary operator is used to solve the discrete feature selection problems. Besides, four machine learning algorithms, which are the K-nearest neighbor (KNN), support vector machine, decision tree, and random forest, are used to calculate the classification accuracy. Experimental results show that the proposed ISSA with KNN classifier eliminates about 65.9% of useless features, while the classification accuracy rate reaches more than 85%, which indicates that the proposed algorithm is more suitable for brain disease analysis than other comparison algorithms and other classifiers. Moreover, the effectiveness of the proposed improved mechanisms and the portability of different classifiers are evaluated by tests, and convolutional neural networks are compared with ISSA to classify the brain computed tomography (CT) images.

Interpretable brain disease classification and relevance-guided deep learning

Deep neural networks are increasingly used for neurological disease classification by MRI, but the networks’ decisions are not easily interpretable by humans. Heat mapping by deep Taylor decomposition revealed that (potentially misleading) image features even outside of the brain tissue are crucial for the classifier’s decision. We propose a regularization technique to train convolutional neural network (CNN) classifiers utilizing relevance-guided heat maps calculated online during training. The method was applied using T1-weighted MR images from 128 subjects with Alzheimer’s disease (mean age = 71.9 ± 8.5 years) and 290 control subjects (mean age = 71.3 ± 6.4 years). The developed relevance-guided framework achieves higher classification accuracies than conventional CNNs but more importantly, it relies on less but more relevant and physiological plausible voxels within brain tissue. Additionally, preprocessing effects from skull stripping and registration are mitigated. With the interpretability of the decision mechanisms underlying CNNs, these results challenge the notion that unprocessed T1-weighted brain MR images in standard CNNs yield higher classification accuracy in Alzheimer’s disease than solely atrophy.

A novel generation adversarial network framework with characteristics aggregation and diffusion for brain disease classification and feature selection.

Imaging genetics provides unique insights into the pathological studies of complex brain diseases by integrating the characteristics of multi-level medical data. However, most current imaging genetics research performs incomplete data fusion. Also, there is a lack of effective deep learning methods to analyze neuroimaging and genetic data jointly. Therefore, this paper first constructs the brain region-gene networks to intuitively represent the association pattern of pathogenetic factors. Second, a novel feature information aggregation model is constructed to accurately describe the information aggregation process among brain region nodes and gene nodes. Finally, a deep learning method called feature information aggregation and diffusion generative adversarial network (FIAD-GAN) is proposed to efficiently classify samples and select features. We focus on improving the generator with the proposed convolution and deconvolution operations, with which the interpretability of the deep learning framework has been dramatically improved. The experimental results indicate that FIAD-GAN can not only achieve superior results in various disease classification tasks but also extract brain regions and genes closely related to AD. This work provides a novel method for intelligent clinical decisions. The relevant biomedical discoveries provide a reliable reference and technical basis for the clinical diagnosis, treatment and pathological analysis of disease.

Diagnosis after zooming in: A multilabel classification model by imitating doctor reading habits to diagnose brain diseases.

Computed tomography (CT) has the advantages of being low cost and noninvasive and is a primary diagnostic method for brain diseases. However, it is a challenge for junior radiologists to diagnose CT images accurately and comprehensively. It is necessary to build a system that can help doctors diagnose and provide an explanation of the predictions. Despite the success of deep learning algorithms in the field of medical image analysis, the task of brain disease classification still faces challenges: Researchers lack attention to complex manual labeling requirements and the incompleteness of prediction explanations. More importantly, most studies only measure the performance of the algorithm, but do not measure the effectiveness of the algorithm in the actual diagnosis ofdoctors. In this paper, we propose a model called DrCT2 that can detect brain diseases without using image-level labels and provide a more comprehensive explanation at both the slice and sequence levels. This model achieves reliable performance by imitating human expert reading habits: targeted scaling of primary images from the full slice scans and observation of suspicious lesions for diagnosis. We evaluated our model on two open-access data sets: CQ500 and the RSNA Intracranial Hemorrhage Detection Challenge. In addition, we defined three tasks to comprehensively evaluate model interpretability by measuring whether the algorithm can select key images with lesions. To verify the algorithm from the perspective of practical application, three junior radiologists were invited to participate in the experiments, comparing the effects before and after human-computer cooperation in differentaspects. The method achieved F1-scores of 0.9370 on CQ500 and 0.8700 on the RSNA data set. The results show that our model has good interpretability under the premise of good performance. Human radiologist evaluation experiments have proven that our model can effectively improve the accuracy of the diagnosis and improveefficiency. We proposed a model that can simultaneously detect multiple brain diseases. The report generated by the model can assist doctors in avoiding missed diagnoses, and it has good clinical applicationvalue.