Dynamic clustering via branched deep learning enhances personalization of stress prediction from mobile sensor data

Abstract

Abstract College students experience ever-increasing levels of stress, leading to a wide range of health problems. In this context, monitoring and predicting students’ stress levels is crucial and, fortunately, made possible by the growing support for data collection via mobile devices. However, predicting stress levels from mobile phone data remains a challenging task, and off-the-shelf deep learning models are inapplicable or inefficient due to data irregularity, inter-subject variability, and the “cold start problem”. To overcome these challenges, we developed a platform named Branched CALM-Net that aims to predict students’ stress levels through dynamic clustering in a personalized manner. This is the first platform that leverages the branching technique in a multitask setting to achieve personalization and continuous adaptation. Our method achieves state-of-the-art performance in predicting student stress from mobile sensor data collected as part of the Dartmouth StudentLife study, with a ROC AUC 37% higher and a PR AUC surpassing that of the nearest baseline models. In the cold-start online learning setting, Branched CALM-Net outperforms other models, attaining an average F1 score of 87% with just 1 week of training data for a new student, which shows it is reliable and effective at predicting stress levels from mobile data.

Keywords

• Machine learning
• Multitask learning
• Stress prediction
• Mobile computing