Specializing CNN Models for Sleep Staging Based on Heart Rate

Publikation: Beitrag in Buch/Konferenzbericht/Sammelband/GutachtenBeitrag in KonferenzbandBeigetragenBegutachtung

Beitragende

Abstract

This work aims to classify sleep stages based on tachograms using Convolutional Neural Networks (CNNs) and investigate advantages of specialized classifiers. The tachograms of 5422 patients were extracted from the Sleep Heart Health Study. A CNN was trained to classify each 30 s epoch into four distinct sleep stages. The patients were divided into four subgroups by Apnoe-Hypopnoe-Index (AHI). From each subgroup, 20 % of pa-tients were held out as test data. One general model was trained on all training patients and four narrowed models were each trained on one subgroup. Furthermore, the general model was retrained on the subgroups, yielding four additional transfer learning models. Our general model gained an average Cohen's Kappa score of 0.53. The general model outperformed the narrowed models on each test subset. From the narrowed models, training on the subgroup with AHI 5-15 achieved best overall performance. However, a correlation exists between the size of train sets and classification quality. Transfer learning did not improve the results. CNN models are capable of learning features from tachograms with very good classification performance compared to other works using heart rate only. However, the pursued strategies for specializing classifiers did not yield any advantages over our general model.

Details

OriginalspracheDeutsch
Titel2020 Computing in Cardiology
Herausgeber (Verlag)Wiley-IEEE Press
Seiten1-4
Seitenumfang4
ISBN (Print)978-1-7281-1105-6
PublikationsstatusVeröffentlicht - 16 Sept. 2020
Peer-Review-StatusJa

Konferenz

Titel47th Computing in Cardiology Conference
KurztitelCinC 2020
Veranstaltungsnummer47
Dauer13 - 16 September 2020
Webseite
BekanntheitsgradInternationale Veranstaltung
OrtRimini Palacongressi & online
StadtRimini
LandItalien

Externe IDs

Scopus 85100947358
ORCID /0000-0003-2126-290X/work/142250133

Schlagworte

Schlagwörter

  • Training, Heart rate, Transfer learning, Time series analysis, Training data, Sleep apnea, Task analysis