In plain words: A neural network that reads weeks of wearable heart rate data in order first learns heart-rate patterns without labels, then is tuned to flag several conditions at once. It beat hand-crafted medical biomarkers, reaching 0.85 at spotting diabetes.
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.
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
arXiv:1802.02511 · cs.LG, cs.AI, stat.ML · submitted Feb 7, 2018
abstract · pdf · html · Presented at AAAI 2018