期刊
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS
卷 25, 期 6, 页码 1949-1963出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/JBHI.2020.3037693
关键词
Sleep; Brain modeling; Electroencephalography; Task analysis; Feature extraction; Training; Data models; Sleep stage classification; meta-learning; pre-trained EEG; transfer learning; convolutional neural network
类别
资金
- PTT Public Company Limited
- Thailand Science Research and Innovation [SRI62W1501]
- Office of National Higher Education Science Research and Innovation Policy Council [C10F630057]
- SCB Public Company Limited
This study proposes a novel transfer learning framework MetaSleepLearner based on Meta-Learning (MAML), achieving a range of 5.4% to 17.7% improvement in comparison to the conventional approach. The model interpretation after adaptation to each subject confirms a reasonable learning performance, showcasing potential for human-machine collaboration in sleep stage classification and reducing clinicians' workload.
Identifying bio-signals based-sleep stages requires time-consuming and tedious labor of skilled clinicians. Deep learning approaches have been introduced in order to challenge the automatic sleep stage classification conundrum. However, the difficulties can be posed in replacing the clinicians with the automatic system due to the differences in many aspects found in individual bio-signals, causing the inconsistency in the performance of the model on every incoming individual. Thus, we aim to explore the feasibility of using a novel approach, capable of assisting the clinicians and lessening the workload. We propose the transfer learning framework, entitled MetaSleepLearner, based on Model Agnostic Meta-Learning (MAML), in order to transfer the acquired sleep staging knowledge from a large dataset to new individual subjects (source code is available at https://github.com/IoBT-VISTEC/MetaSleepLearner). The framework was demonstrated to require the labelling of only a few sleep epochs by the clinicians and allow the remainder to be handled by the system. Layer-wise Relevance Propagation (LRP) was also applied to understand the learning course of our approach. In all acquired datasets, in comparison to the conventional approach, MetaSleepLearner achieved a range of 5.4% to 17.7% improvement with statistical difference in the mean of both approaches. The illustration of the model interpretation after the adaptation to each subject also confirmed that the performance was directed towards reasonable learning. MetaSleepLearner outperformed the conventional approaches as a result from the fine-tuning using the recordings of both healthy subjects and patients. This is the first work that investigated a non-conventional pre-training method, MAML, resulting in a possibility for human-machine collaboration in sleep stage classification and easing the burden of the clinicians in labelling the sleep stages through only several epochs rather than an entire recording.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据