Physics-Informed Transfer Learning to Enhance Sleep Staging

Abstract

Objective: At-home sleep staging using wearable medical sensors poses a viable alternative to in-hospital polysomnography due to its lower cost and lower disruption to the daily lives of patients, especially in the case of long-term monitoring. Machine learning with wearables however is difficult due to the paucity of data from wearable sensors, making automation a challenge. Transfer learning from hospital polysomnograms can boost performance, but is still hindered by differences between wearable and in-hospital EEG resulting in part from differing electrode placement. We improve transfer learning performance by using electrophysiological models of a human head to generate synthetic EEG resembling EEG from a wearable sensor.

Methods: The data generation method utilizes Low-Resolution Electromagnetic Tomography Analysis (LORETA). Real EEG from standard in- hospital recordings is first mapped to point currents within the brain using LORETA, after which the point currents are used to estimate EEG that would have been recorded using a wearable sensor at any given point on the head.

Results: Augmenting source datasets with synthetic data statistically significantly boosted accuracy on a wearable sleep staging task from 80.8% to 81.3% on average, depending on the transfer learning parameters and data sources.

Conclusion: Machine learning performance can be improved using data synthesized using physical models.

Significance: Our approach represents a new form of transfer learning and demonstrates that incorporating domain knowledge of electrophysiological modeling can improve machine learning results for sleep staging tasks. We expect this approach to be particularly useful for EEG data which is hard to collect, or which is obtained using unusual electrode configurations.