kth.sePublications
System disruptions
We are currently experiencing disruptions on the search portals due to high traffic. We are working to resolve the issue, you may temporarily encounter an error message.
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
A Machine Learning-Based Method to Identify Bipolar Disorder Patients
KTH, School of Electrical Engineering and Computer Science (EECS), Electrical Engineering, Electromagnetic Engineering.ORCID iD: 0000-0002-4900-4788
Show others and affiliations
2022 (English)In: Circuits, systems, and signal processing, ISSN 0278-081X, E-ISSN 1531-5878, Vol. 41, no 4, p. 2244-2265Article in journal (Refereed) Published
Abstract [en]

Bipolar disorder is a serious psychiatric disorder characterized by periodic episodes of manic and depressive symptomatology. Due to the high percentage of people suffering from severe bipolar and depressive disorders, the modelling, characterisation, classification and diagnostic analysis of these mental disorders are of vital importance in medical research. Electroencephalogram (EEG) records offer important information to enhance clinical diagnosis and are widely used in hospitals. For this reason, EEG records and patient data from the Virgen de la Luz Hospital were used in this work. In this paper, an extreme gradient boosting (XGB) machine learning (ML) method involving an EEG signal is proposed. Four supervised ML algorithms including a k-nearest neighbours (KNN), decision tree (DT), Gaussian Naïve Bayes (GNB) and support vector machine (SVM) were compared with the proposed XGB method. The performance of these methods was tested implementing a standard 10-fold cross-validation process. The results indicate that the XGB has the best prediction accuracy (94%), high precision (> 0.94) and high recall (> 0.94). The KNN, SVM, and DT approaches also present moderate prediction accuracy (> 87), moderate recall (> 0.87) and moderate precision (> 0.87). The GNB algorithm shows relatively low classification performance. Based on these results for classification performance and prediction accuracy, the XGB is a solid candidate for a correct classification of patients with bipolar disorder. These findings suggest that XGB system trained with clinical data may serve as a new tool to assist in the diagnosis of patients with bipolar disorder. 

Place, publisher, year, edition, pages
Springer Nature , 2022. Vol. 41, no 4, p. 2244-2265
Keywords [en]
Biomedical signals, Bipolar disorders, Extreme gradient boosting, Machine learning, Adaptive boosting, Biomedical signal processing, Decision trees, Electroencephalography, Forecasting, Hospital data processing, Hospitals, Support vector machines, Biomedical signal, Bipolar disorder, Classification performance, Gaussians, Gradient boosting, Learning-based methods, Machine-learning, Prediction accuracy, Support vectors machine, Nearest neighbor search
National Category
Other Engineering and Technologies Information Systems Communication Systems
Identifiers
URN: urn:nbn:se:kth:diva-313196DOI: 10.1007/s00034-021-01889-1ISI: 000722804800001Scopus ID: 2-s2.0-85119997618OAI: oai:DiVA.org:kth-313196DiVA, id: diva2:1665765
Note

QC 20220608

Available from: 2022-06-08 Created: 2022-06-08 Last updated: 2025-02-18Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Quevedo-Teruel, Oscar

Search in DiVA

By author/editor
Quevedo-Teruel, Oscar
By organisation
Electromagnetic Engineering
In the same journal
Circuits, systems, and signal processing
Other Engineering and TechnologiesInformation SystemsCommunication Systems

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 89 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf