Autism Diagnosis from EEG Signals Using Machine Learning Algorithms and Convolutional Neural Networks

نوع مقاله : مقاله پژوهشی

نویسندگان

1 گروه مهندسی برق، دانشگاه فنی و حرفه ای، تهران، ایران

2 گروه مهندسی برق، دانشگاه اردکان، اردکان، ایران

3 گروه مهندسی برق، دانشگاه یزد، یزد، ایران

چکیده

A neurodevelopmental disorder recognized by insufficiency in social communication and repetitive behaviours is expressed as an autism spectrum disorder (ASD). One of the most useful tools for diagnosing autism is the use of electroencephalography (EEG) signals because these signals accurately represent the brain's function. The recorded EEG of each person contains a lot of information that is very difficult to study and check visually. The main goal of machine learning algorithms is to train the machine in such a way that it finally has a diagnosis close to that of the human brain. This paper evaluates the appropriate strategies for further exploitation of deep learning capabilities in the feature extraction block of autism diagnosis without using classical feature extraction methods. In this research, a convolutional neural network (CNN) structure is used to check the available data in order to extract features. Classification has been done with five machine learning classifiers, covering support vector machine (SVM), linear discriminant analysis (LDA), decision tree (DT), simple Bayes classification (GNB), and random forest (RF). The accuracy obtained from the use of classifiers with SVM methods is 100%, LDA 82%, DT 80.5%, GNB 100%, and RF 100%. The proposed idea of convolutional neural networks for feature extraction and classification with different machine learning methods has provided high-accuracy results that are equal to the other amazing methods for autism diagnosis.

کلیدواژه‌ها

موضوعات


عنوان مقاله [English]

Autism Diagnosis from EEG Signals Using Machine Learning Algorithms and Convolutional Neural Networks

نویسندگان [English]

  • Zohreh Sharifi 1
  • Hajar Momeni 2
  • Habib Adabi Ardakani 3
1 Department of Electrical Engineering, Technical and Vocational University, Tehran, Iran
2 Department of Electrical Engineering, Engineering Faculty, Ardakan University, Ardakan, Iran
3 Department of Electrical Engineering, Yazd University, Yazd, Iran
چکیده [English]

A neurodevelopmental disorder recognized by insufficiency in social communication and repetitive behaviours is expressed as an autism spectrum disorder (ASD). One of the most useful tools for diagnosing autism is the use of electroencephalography (EEG) signals because these signals accurately represent the brain's function. The recorded EEG of each person contains a lot of information that is very difficult to study and check visually. The main goal of machine learning algorithms is to train the machine in such a way that it finally has a diagnosis close to that of the human brain. This paper evaluates the appropriate strategies for further exploitation of deep learning capabilities in the feature extraction block of autism diagnosis without using classical feature extraction methods. In this research, a convolutional neural network (CNN) structure is used to check the available data in order to extract features. Classification has been done with five machine learning classifiers, covering support vector machine (SVM), linear discriminant analysis (LDA), decision tree (DT), simple Bayes classification (GNB), and random forest (RF). The accuracy obtained from the use of classifiers with SVM methods is 100%, LDA 82%, DT 80.5%, GNB 100%, and RF 100%. The proposed idea of convolutional neural networks for feature extraction and classification with different machine learning methods has provided high-accuracy results that are equal to the other amazing methods for autism diagnosis.

کلیدواژه‌ها [English]

  • Autistic disorder
  • Convolutional Neural Network
  • Electroencephalography
  • Machine learning
  • Deep learning