TY - JOUR
T1 - Multi-Task Learning for Randomized Controlled Trials
T2 - A case study on predicting depression with wearable data
AU - Dai, Ruixuan
AU - Kannampallil, Thomas
AU - Zhang, Jingwen
AU - Lv, Nan
AU - Ma, Jun
AU - Lu, Chenyang
N1 - Publisher Copyright:
© 2022 Owner/Author.
PY - 2022/7
Y1 - 2022/7
N2 - A randomized controlled trial (RCT) is used to study the safety and efficacy of new treatments, by comparing patient outcomes of an intervention group with a control group. Traditionally, RCTs rely on statistical analyses to assess the differences between the treatment and control groups. However, such statistical analyses are generally not designed to assess the impact of the intervention at an individual level. In this paper, we explore machine learning models in conjunction with an RCT for personalized predictions of a depression treatment intervention, where patients were longitudinally monitored with wearable devices. We formulate individual-level predictions in the intervention and control groups from an RCT as a multi-Task learning (MTL) problem, and propose a novel MTL model specifically designed for RCTs. Instead of training separate models for the intervention and control groups, the proposed MTL model is trained on both groups, effectively enlarging the training dataset. We develop a hierarchical model architecture to aggregate data from different sources and different longitudinal stages of the trial, which allows the MTL model to exploit the commonalities and capture the differences between the two groups. We evaluated the MTL approach in an RCT involving 106 patients with depression, who were randomized to receive an integrated intervention treatment. Our proposed MTL model outperforms both single-Task models and the traditional multi-Task model in predictive performance, representing a promising step in utilizing data collected in RCTs to develop predictive models for precision medicine.
AB - A randomized controlled trial (RCT) is used to study the safety and efficacy of new treatments, by comparing patient outcomes of an intervention group with a control group. Traditionally, RCTs rely on statistical analyses to assess the differences between the treatment and control groups. However, such statistical analyses are generally not designed to assess the impact of the intervention at an individual level. In this paper, we explore machine learning models in conjunction with an RCT for personalized predictions of a depression treatment intervention, where patients were longitudinally monitored with wearable devices. We formulate individual-level predictions in the intervention and control groups from an RCT as a multi-Task learning (MTL) problem, and propose a novel MTL model specifically designed for RCTs. Instead of training separate models for the intervention and control groups, the proposed MTL model is trained on both groups, effectively enlarging the training dataset. We develop a hierarchical model architecture to aggregate data from different sources and different longitudinal stages of the trial, which allows the MTL model to exploit the commonalities and capture the differences between the two groups. We evaluated the MTL approach in an RCT involving 106 patients with depression, who were randomized to receive an integrated intervention treatment. Our proposed MTL model outperforms both single-Task models and the traditional multi-Task model in predictive performance, representing a promising step in utilizing data collected in RCTs to develop predictive models for precision medicine.
KW - Depression
KW - Multi-Task Learning
KW - Randomized Controlled Trial
KW - Wearables
UR - http://www.scopus.com/inward/record.url?scp=85134242595&partnerID=8YFLogxK
U2 - 10.1145/3534591
DO - 10.1145/3534591
M3 - Article
AN - SCOPUS:85134242595
SN - 2474-9567
VL - 6
JO - Proceedings of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies
JF - Proceedings of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies
IS - 2
M1 - 50
ER -