Specializing CNN Models for Sleep Staging Based on Heart Rate

Research output: Contribution to book/Conference proceedings/Anthology/ReportConference contributionContributedpeer-review

Contributors

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

Original languageGerman
Title of host publication2020 Computing in Cardiology
PublisherWiley-IEEE Press
Pages1-4
Number of pages4
ISBN (print)978-1-7281-1105-6
Publication statusPublished - 16 Sept 2020
Peer-reviewedYes

Conference

Title2020 Computing in Cardiology
Abbreviated titleCinC 2020
Conference number47
Duration13 - 16 September 2020
Website
Degree of recognitionInternational event
LocationRimini Palacongressi & online
CityRimini
CountryItaly

External IDs

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

Keywords

Keywords

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