DeepHeart: Semi-Supervised Sequence Learning for Cardiovascular Risk Prediction

  • 2018-02-07 16:31:50
  • Brandon Ballinger, Johnson Hsieh, Avesh Singh, Nimit Sohoni, Jack Wang, Geoffrey H. Tison, Gregory M. Marcus, Jose M. Sanchez, Carol Maguire, Jeffrey E. Olgin, Mark J. Pletcher
  • 29


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


Introduction (beta)



Conclusion (beta)