Toward Interpretable Sleep Stage Classification Using Cross-Modal Transformers.
IEEE Trans Neural Syst Rehabil Eng
; 32: 2893-2904, 2024.
Article
en En
| MEDLINE
| ID: mdl-39102323
ABSTRACT
Accurate sleep stage classification is significant for sleep health assessment. In recent years, several machine-learning based sleep staging algorithms have been developed, and in particular, deep-learning based algorithms have achieved performance on par with human annotation. Despite improved performance, a limitation of most deep-learning based algorithms is their black-box behavior, which have limited their use in clinical settings. Here, we propose a cross-modal transformer, which is a transformer-based method for sleep stage classification. The proposed cross-modal transformer consists of a cross-modal transformer encoder architecture along with a multi-scale one-dimensional convolutional neural network for automatic representation learning. The performance of our method is on-par with the state-of-the-art methods and eliminates the black-box behavior of deep-learning models by utilizing the interpretability aspect of the attention modules. Furthermore, our method provides considerable reductions in the number of parameters and training time compared to the state-of-the-art methods. Our code is available at https//github.com/Jathurshan0330/Cross-Modal-Transformer. A demo of our work can be found at https//bit.ly/Cross_modal_transformer_demo.
Texto completo:
1
Colección:
01-internacional
Base de datos:
MEDLINE
Asunto principal:
Fases del Sueño
/
Algoritmos
/
Redes Neurales de la Computación
/
Aprendizaje Profundo
Límite:
Adult
/
Female
/
Humans
/
Male
Idioma:
En
Revista:
IEEE Trans Neural Syst Rehabil Eng
Asunto de la revista:
ENGENHARIA BIOMEDICA
/
REABILITACAO
Año:
2024
Tipo del documento:
Article
Pais de publicación:
Estados Unidos