Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification

The pivotal role of sleep has led to extensive research endeavors aimed at automatic sleep stage classification. However, existing methods perform poorly when classifying small groups or individuals, and these results are often considered outliers in terms of overall performance. These outliers may...

Full description

Saved in:
Bibliographic Details
Main Authors: Wei Zhou, Hangyu Zhu, Wei Chen, Chen Chen, Jun Xu
Format: Article
Language:English
Published: MDPI AG 2024-12-01
Series:Bioengineering
Subjects:
Online Access:https://www.mdpi.com/2306-5354/11/12/1226
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850240499774914560
author Wei Zhou
Hangyu Zhu
Wei Chen
Chen Chen
Jun Xu
author_facet Wei Zhou
Hangyu Zhu
Wei Chen
Chen Chen
Jun Xu
author_sort Wei Zhou
collection DOAJ
description The pivotal role of sleep has led to extensive research endeavors aimed at automatic sleep stage classification. However, existing methods perform poorly when classifying small groups or individuals, and these results are often considered outliers in terms of overall performance. These outliers may introduce bias during model training, adversely affecting feature selection and diminishing model performance. To address the above issues, this paper proposes an ensemble-based sequential convolutional neural network (E-SCNN) that incorporates a clustering module and neural networks. E-SCNN effectively ensembles machine learning and deep learning techniques to minimize outliers, thereby enhancing model robustness at the individual level. Specifically, the clustering module categorizes individuals based on similarities in feature distribution and assigns personalized weights accordingly. Subsequently, by combining these tailored weights with the robust feature extraction capabilities of convolutional neural networks, the model generates more accurate sleep stage classifications. The proposed model was verified on two public datasets, and experimental results demonstrate that the proposed method obtains overall accuracies of 84.8% on the Sleep-EDF Expanded dataset and 85.5% on the MASS dataset. E-SCNN can alleviate the outlier problem, which is important for improving sleep quality monitoring for individuals.
format Article
id doaj-art-6efa2fac58214505b6aec6cfbe00e3d2
institution OA Journals
issn 2306-5354
language English
publishDate 2024-12-01
publisher MDPI AG
record_format Article
series Bioengineering
spelling doaj-art-6efa2fac58214505b6aec6cfbe00e3d22025-08-20T02:00:51ZengMDPI AGBioengineering2306-53542024-12-011112122610.3390/bioengineering11121226Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage ClassificationWei Zhou0Hangyu Zhu1Wei Chen2Chen Chen3Jun Xu4Jiangsu Key Laboratory of Intelligent Medical Image Computing, Nanjing 210044, ChinaCenter for Intelligent Medical Electronics (CIME), School of Information Science and Engineering, Fudan University, Shanghai 200433, ChinaSchool of Biomedical Engineering, The University of Sydney, Sydney, NSW 2006, AustraliaCenter for Medical Research and Innovation, Shanghai Pudong Hosptial, Fudan University Pudong Medical Center, Shanghai 201203, ChinaJiangsu Key Laboratory of Intelligent Medical Image Computing, Nanjing 210044, ChinaThe pivotal role of sleep has led to extensive research endeavors aimed at automatic sleep stage classification. However, existing methods perform poorly when classifying small groups or individuals, and these results are often considered outliers in terms of overall performance. These outliers may introduce bias during model training, adversely affecting feature selection and diminishing model performance. To address the above issues, this paper proposes an ensemble-based sequential convolutional neural network (E-SCNN) that incorporates a clustering module and neural networks. E-SCNN effectively ensembles machine learning and deep learning techniques to minimize outliers, thereby enhancing model robustness at the individual level. Specifically, the clustering module categorizes individuals based on similarities in feature distribution and assigns personalized weights accordingly. Subsequently, by combining these tailored weights with the robust feature extraction capabilities of convolutional neural networks, the model generates more accurate sleep stage classifications. The proposed model was verified on two public datasets, and experimental results demonstrate that the proposed method obtains overall accuracies of 84.8% on the Sleep-EDF Expanded dataset and 85.5% on the MASS dataset. E-SCNN can alleviate the outlier problem, which is important for improving sleep quality monitoring for individuals.https://www.mdpi.com/2306-5354/11/12/1226sleep stagingdeep learningelectroencephalograph
spellingShingle Wei Zhou
Hangyu Zhu
Wei Chen
Chen Chen
Jun Xu
Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification
Bioengineering
sleep staging
deep learning
electroencephalograph
title Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification
title_full Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification
title_fullStr Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification
title_full_unstemmed Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification
title_short Outlier Handling Strategy of Ensembled-Based Sequential Convolutional Neural Networks for Sleep Stage Classification
title_sort outlier handling strategy of ensembled based sequential convolutional neural networks for sleep stage classification
topic sleep staging
deep learning
electroencephalograph
url https://www.mdpi.com/2306-5354/11/12/1226
work_keys_str_mv AT weizhou outlierhandlingstrategyofensembledbasedsequentialconvolutionalneuralnetworksforsleepstageclassification
AT hangyuzhu outlierhandlingstrategyofensembledbasedsequentialconvolutionalneuralnetworksforsleepstageclassification
AT weichen outlierhandlingstrategyofensembledbasedsequentialconvolutionalneuralnetworksforsleepstageclassification
AT chenchen outlierhandlingstrategyofensembledbasedsequentialconvolutionalneuralnetworksforsleepstageclassification
AT junxu outlierhandlingstrategyofensembledbasedsequentialconvolutionalneuralnetworksforsleepstageclassification