kth.sePublikationer KTH
Ändra sökning
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf
A Machine Learning-Based Method to Identify Bipolar Disorder Patients
KTH, Skolan för elektroteknik och datavetenskap (EECS), Elektroteknik, Elektroteknisk teori och konstruktion.ORCID-id: 0000-0002-4900-4788
Visa övriga samt affilieringar
2022 (Engelska)Ingår i: Circuits, systems, and signal processing, ISSN 0278-081X, E-ISSN 1531-5878, Vol. 41, nr 4, s. 2244-2265Artikel i tidskrift (Refereegranskat) 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. 

Ort, förlag, år, upplaga, sidor
Springer Nature , 2022. Vol. 41, nr 4, s. 2244-2265
Nyckelord [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
Nationell ämneskategori
Annan teknik Systemvetenskap, informationssystem och informatik Kommunikationssystem
Identifikatorer
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
Anmärkning

QC 20220608

Tillgänglig från: 2022-06-08 Skapad: 2022-06-08 Senast uppdaterad: 2025-02-18Bibliografiskt granskad

Open Access i DiVA

Fulltext saknas i DiVA

Övriga länkar

Förlagets fulltextScopus

Person

Quevedo-Teruel, Oscar

Sök vidare i DiVA

Av författaren/redaktören
Quevedo-Teruel, Oscar
Av organisationen
Elektroteknisk teori och konstruktion
I samma tidskrift
Circuits, systems, and signal processing
Annan teknikSystemvetenskap, informationssystem och informatikKommunikationssystem

Sök vidare utanför DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetricpoäng

doi
urn-nbn
Totalt: 141 träffar
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf