Specializing CNN Models for Sleep Staging Based on Heart Rate
Research output: Contribution to book/Conference proceedings/Anthology/Report › Conference contribution › Contributed › peer-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 language | German |
---|---|
Title of host publication | 2020 Computing in Cardiology |
Publisher | Wiley-IEEE Press |
Pages | 1-4 |
Number of pages | 4 |
ISBN (print) | 978-1-7281-1105-6 |
Publication status | Published - 16 Sept 2020 |
Peer-reviewed | Yes |
Conference
Title | 2020 Computing in Cardiology |
---|---|
Abbreviated title | CinC 2020 |
Conference number | 47 |
Duration | 13 - 16 September 2020 |
Website | |
Degree of recognition | International event |
Location | Rimini Palacongressi & online |
City | Rimini |
Country | Italy |
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