Predicting Age with Deep Neural Networks from Polysomnograms

SViT: A Spectral Vision Transformer for the Detection of REM Sleep Behavior Disorder

REM sleep behavior disorder (RBD) is a parasomnia with dream enactment and presence of REM sleep without atonia (RSWA). RBD diagnosed manually via polysomnography (PSG) scoring, which is time intensive. Isolated RBD (iRBD) is also associated with a high probability of conversion to Parkinson's disease. Diagnosis of iRBD is largely based on clinical evaluation and subjective PSG ratings of REM sleep without atonia. Here we show the first application of a novel spectral vision transformer (SViT) to PSG signals for detection of RBD and compare the results to the more conventional convolutional neural network architecture. The vision-based deep learning models were applied to scalograms (30 or 300 s windows) of the PSG data (EEG, EMG and EOG) and the predictions interpreted. A total of 153 RBD (96 iRBD and 57 RBD with PD) and 190 controls were included in the study and 5-fold bagged ensemble was used. Model outputs were analyzed per-patient (averaged), with regards to sleep stage, and the SViT was interpreted using integrated gradients. Models had a similar per-epoch test F1 score. However, the vision transformer had the best per-patient performance, with an F1 score 0.87. Training the SViT on channel subsets, it achieved an F1 score of 0.93 on a combination of EEG and EOG. EMG is thought to have the highest diagnostic yield, but interpretation of our model showed that high relevance was placed on EEG and EOG, indicating these channels could be included for diagnosing RBD.

End-to-end Deep Learning of Polysomnograms for Classification of REM Sleep Behavior Disorder

Rapid eye movement (REM) sleep behavior disorder (RBD) is parasomnia and a prodromal manifestation of Parkinson's disease. The current diagnostic method relies on manual scoring of polysomnograms (PSGs), a procedure that is time and effort intensive, subject to interscorer variability, and requires high level of expertise. Here, we present an automatic and interpretable diagnostic tool for RBD that analyzes PSGs using end-to-end deep neural networks. We optimized hierarchical attention networks in a 5-fold cross validation directly to classify RBD from PSG data recorded in 143 participants with RBD and 147 age-and sex-matched controls. An ensemble model using logistic regression was implemented to fuse decisions from networks trained in various signal combinations. We interpreted the networks using gradient SHAP that attribute relevance of input signals to model decisions. The ensemble model achieved a sensitivity of 91.4 % and a specificity of 86.3 %. Interpretation showed that electroencephalography (EEG) and leg electromyography (EMG) exhibited most patterns with high relevance. This study validates a robust diagnostic tool for RBD and proposes an interpretable and fully automatic framework for end-to-end modeling of other sleep disorders from PSG data. Clinical relevance- This study presents a novel diagnostic tool for RBD that considers neurophysiologic biomarkers in multiple modalities.

A comparison between diffusion tensor imaging and generalized q-sampling imaging in the age prediction of healthy adults via machine learning approaches

OBJECTIVE

Brain age, which is predicted using neuroimaging data, has become an important biomarker in aging research. This study applied diffusion tensor imaging (DTI) and generalized q-sampling imaging (GQI) model to predict age respectively, with the purpose of evaluating which diffusion model is more accurate in estimating age and revealing age-related changes in the brain.

APPROACH

Diffusion MRI data of 125 subjects from two sites were collected. Fractional anisotropy (FA) and quantitative anisotropy (QA) from the two diffusion models were calculated and were used as features of machine learning models. Sequential backward elimination algorithm was used for feature selection. Six machine learning approaches including linear regression, ridge regression, support vector regression (SVR) with linear kernel, quadratic kernel and radial basis function (RBF) kernel and feedforward neural network were used to predict age using FA and QA features respectively.

MAIN RESULTS

Age predictions using FA features were more accurate than predictions using QA features for all the 6 machine learning algorithms. Post-hoc analysis revealed that FA was more sensitive to age-related white matter alterations in the brain. In addition, SVR with RBF kernel based on FA features achieved better performances than the competing algorithms with MAE ranging from 7.74 to 10.54, MSE ranging from 87.79 to 150.86, and nMSE ranging from 0.05 to 0.14 Significance: FA from DTI model was more suitable than QA from GQI model in age prediction. FA metric was more sensitive to age-related white matter changes in the brain and FA of several brain regions could be used as white matter biomarkers in aging.