Classification and transfer learning of sleep spindles based on convolutional neural networks

Jun Liang, Abdelkader Nasreddine Belkacem, Yanxin Song, Jiaxin Wang, Zhiguo Ai, Xuanqi Wang, Jun Guo, Lingfeng Fan, Changming Wang, Bowen Ji, Zengguang Wang

Research output: Contribution to journalArticlepeer-review

Abstract

Background: Sleep plays a critical role in human physiological and psychological health, and electroencephalography (EEG), an effective sleep-monitoring method, is of great importance in revealing sleep characteristics and aiding the diagnosis of sleep disorders. Sleep spindles, which are a typical phenomenon in EEG, hold importance in sleep science. Methods: This paper proposes a novel convolutional neural network (CNN) model to classify sleep spindles. Transfer learning is employed to apply the model trained on the sleep spindles of healthy subjects to those of subjects with insomnia for classification. To analyze the effect of transfer learning, we discuss the classification results of both partially and fully transferred convolutional layers. Results: The classification accuracy for the healthy and insomnia subjects’ spindles were 93.68% and 92.77%, respectively. During transfer learning, when transferring all convolutional layers, the classification accuracy for the insomnia subjects’ spindles was 91.41% and transferring only the first four convolutional layers achieved a classification result of 92.80%. The experimental results demonstrate that the proposed CNN model can effectively classify sleep spindles. Furthermore, the features learned from the data of the normal subjects can be effectively applied to the data for subjects with insomnia, yielding desirable outcomes. Discussion: These outcomes underscore the efficacy of both the collected dataset and the proposed CNN model. The proposed model exhibits potential as a rapid and effective means to diagnose and treat sleep disorders, thereby improving the speed and quality of patient care.

Original languageEnglish
Article number1396917
JournalFrontiers in Neuroscience
Volume18
DOIs
Publication statusPublished - 2024

Keywords

  • convolutional neural network
  • electroencephalogram
  • polysomnography
  • sleep spindles
  • transfer learning

ASJC Scopus subject areas

  • General Neuroscience

Cite this