English

DeepHeart: Semi-Supervised Sequence Learning for Cardiovascular Risk Prediction

Machine Learning 2018-02-08 v1 Artificial Intelligence Machine Learning

Abstract

We train and validate a semi-supervised, multi-task LSTM on 57,675 person-weeks of data from off-the-shelf wearable heart rate sensors, showing high accuracy at detecting multiple medical conditions, including diabetes (0.8451), high cholesterol (0.7441), high blood pressure (0.8086), and sleep apnea (0.8298). We compare two semi-supervised train- ing methods, semi-supervised sequence learning and heuristic pretraining, and show they outperform hand-engineered biomarkers from the medical literature. We believe our work suggests a new approach to patient risk stratification based on cardiovascular risk scores derived from popular wearables such as Fitbit, Apple Watch, or Android Wear.

Keywords

Cite

@article{arxiv.1802.02511,
  title  = {DeepHeart: Semi-Supervised Sequence Learning for Cardiovascular Risk Prediction},
  author = {Brandon Ballinger and Johnson Hsieh and Avesh Singh and Nimit Sohoni and Jack Wang and Geoffrey H. Tison and Gregory M. Marcus and Jose M. Sanchez and Carol Maguire and Jeffrey E. Olgin and Mark J. Pletcher},
  journal= {arXiv preprint arXiv:1802.02511},
  year   = {2018}
}

Comments

Presented at AAAI 2018

R2 v1 2026-06-23T00:14:46.038Z