In clinical medicine, the accuracy of sleep staging directly affects the diagnosis and treatment of sleep disorders. However, existing methods have significant limitations in characterizing dynamic spatiotemporal associations of the brain during sleep, especially in modeling nonlinear interactions across brain regions. To overcome these limitations, we develop STHGNN, a hypergraph-based neural network model that captures spatiotemporal features for automated sleep staging. The model fuses temporal and spatial features extracted from multi-modal data, and thus performs classification task effectively. In our experiments, we compared our method on the ISRUC-S1 and ISRUC-S3 datasets with the latest methods. Experimental results on the ISRUC-S3 dataset show that the overall classification accuracy of this method reaches 82.8%, F1-score and Cohen kappa reach 81.1% and 77.8%, respectively. Compared with existing baseline classification methods, this method has better classification performance.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Automatic Sleep Staging with Dynamic Hypergraph Neural Networks Using Spatiotemporal Features

  • Hailu Fan,
  • Yubin Chen,
  • Lucong Wang,
  • Xinyang Li

摘要

In clinical medicine, the accuracy of sleep staging directly affects the diagnosis and treatment of sleep disorders. However, existing methods have significant limitations in characterizing dynamic spatiotemporal associations of the brain during sleep, especially in modeling nonlinear interactions across brain regions. To overcome these limitations, we develop STHGNN, a hypergraph-based neural network model that captures spatiotemporal features for automated sleep staging. The model fuses temporal and spatial features extracted from multi-modal data, and thus performs classification task effectively. In our experiments, we compared our method on the ISRUC-S1 and ISRUC-S3 datasets with the latest methods. Experimental results on the ISRUC-S3 dataset show that the overall classification accuracy of this method reaches 82.8%, F1-score and Cohen kappa reach 81.1% and 77.8%, respectively. Compared with existing baseline classification methods, this method has better classification performance.