A MULTI-LAYER HYBRID MACHINE LEARNING MODEL FOR AUTOMATIC SLEEP STAGE CLASSIFICATION

摘要

Sleep Stage Classification (SSC) is a standard process in the Polysomnography (PSG) for studying sleep patterns and events. The SSC provides sleep stage information of a patient throughout an entire sleep test. A physician uses results from SSCs to diagnose sleep disorder symptoms. However, the SSC data processing is time-consuming and requires trained sleep technicians to complete the task. Over the years, researchers attempted to find alternative methods, which are known as Automatic Sleep Stage Classification (ASSC), to perform the task faster and more efficiently. Proposed ASSC techniques usually derived from existing statistical methods and machine learning (ML) techniques. The objective of this study is to develop a new hybrid ASSC technique, Multi-Layer Hybrid Machine Learning Model (MLHM), for classifying sleep stages. The MLHM blends two baseline ML techniques, Decision Tree (DT) and Support Vector Machine (SVM). It operates on a newly developed multi-layer architecture. The multi-layer architecture consists of three layers for classifying W, R and N1, N2, N3 in different epoch lengths. Our experiment design compares MLHM and baseline ML techniques and other research works. The dataset used in this study was derived from the ISRUC-Sleep database comprising of 100 subjects. The classification performances were thoroughly reviewed using the hold-out and the 10-fold cross-validation method in both subject-specific and subject-independent classifications. The MLHM achieved a certain satisfactory classification results. It gained 0.694±0.22 of accuracy (AUC=0.822±0.31) in subject-specific classification and 0.942±0.02 of accuracy (AUC=0.920±0.17) in subject-independent classification. The pros and cons of the MLHM with the multi-layer architecture were thoroughly discussed. The effect of class imbalance was rationally discussed towards the classification results.