Towards a personalized pregnancy experience: Forecasting symptoms using graph neural networks and digital health technologies

Rui Zhu, Jennifer Yu, Stephen H. Friend, Sarah M. Goodday, Bo Wang, Anna Goldenberg
Proceedings of the 4th Machine Learning for Health Symposium, PMLR 259:1104-1120, 2025.

Abstract

Pregnancy is an intricate process involving substantial physiological changes that impact both maternal and fetal health. In this study, we demonstrate the ability to predict two common symptoms during the third trimester of pregnancy—edema and fatigue—using physiological measures from wearable devices and self-reported daily surveys from mobile apps. Our approach employs a Graph Neural Network (GNN) framework, enhanced with a modified weighted Cross-Entropy loss to improve prediction performance. The model achieved AUC scores of 77.27% for edema and 69.70% for fatigue. Additionally, we aligned data from self-reported pregnancy symptoms with clinical examinations to carefully select participant cohorts for our experiments. We also explored how various features identified by the GNN are linked to these symptoms, gaining deeper insights into the relationship between physiological measures and pregnancy symptoms. Our findings indicate that heart rate variability plays a significant role in predicting symptoms of edema and fatigue, and features related to low-intensity activity also have a notable impact. Some of our findings closely align with previous studies on pregnancy. Our research serves as a proof of concept that symptoms can be predicted using wearable data, which may enhance the immediate well-being of expectant mothers and potentially personalize the overall pregnancy experience.

Cite this Paper


BibTeX
@InProceedings{pmlr-v259-zhu25a, title = {Towards a personalized pregnancy experience: Forecasting symptoms using graph neural networks and digital health technologies}, author = {Zhu, Rui and Yu, Jennifer and Friend, Stephen H. and Goodday, Sarah M. and Wang, Bo and Goldenberg, Anna}, booktitle = {Proceedings of the 4th Machine Learning for Health Symposium}, pages = {1104--1120}, year = {2025}, editor = {Hegselmann, Stefan and Zhou, Helen and Healey, Elizabeth and Chang, Trenton and Ellington, Caleb and Mhasawade, Vishwali and Tonekaboni, Sana and Argaw, Peniel and Zhang, Haoran}, volume = {259}, series = {Proceedings of Machine Learning Research}, month = {15--16 Dec}, publisher = {PMLR}, pdf = {https://raw.githubusercontent.com/mlresearch/v259/main/assets/zhu25a/zhu25a.pdf}, url = {https://proceedings.mlr.press/v259/zhu25a.html}, abstract = {Pregnancy is an intricate process involving substantial physiological changes that impact both maternal and fetal health. In this study, we demonstrate the ability to predict two common symptoms during the third trimester of pregnancy—edema and fatigue—using physiological measures from wearable devices and self-reported daily surveys from mobile apps. Our approach employs a Graph Neural Network (GNN) framework, enhanced with a modified weighted Cross-Entropy loss to improve prediction performance. The model achieved AUC scores of 77.27% for edema and 69.70% for fatigue. Additionally, we aligned data from self-reported pregnancy symptoms with clinical examinations to carefully select participant cohorts for our experiments. We also explored how various features identified by the GNN are linked to these symptoms, gaining deeper insights into the relationship between physiological measures and pregnancy symptoms. Our findings indicate that heart rate variability plays a significant role in predicting symptoms of edema and fatigue, and features related to low-intensity activity also have a notable impact. Some of our findings closely align with previous studies on pregnancy. Our research serves as a proof of concept that symptoms can be predicted using wearable data, which may enhance the immediate well-being of expectant mothers and potentially personalize the overall pregnancy experience.} }
Endnote
%0 Conference Paper %T Towards a personalized pregnancy experience: Forecasting symptoms using graph neural networks and digital health technologies %A Rui Zhu %A Jennifer Yu %A Stephen H. Friend %A Sarah M. Goodday %A Bo Wang %A Anna Goldenberg %B Proceedings of the 4th Machine Learning for Health Symposium %C Proceedings of Machine Learning Research %D 2025 %E Stefan Hegselmann %E Helen Zhou %E Elizabeth Healey %E Trenton Chang %E Caleb Ellington %E Vishwali Mhasawade %E Sana Tonekaboni %E Peniel Argaw %E Haoran Zhang %F pmlr-v259-zhu25a %I PMLR %P 1104--1120 %U https://proceedings.mlr.press/v259/zhu25a.html %V 259 %X Pregnancy is an intricate process involving substantial physiological changes that impact both maternal and fetal health. In this study, we demonstrate the ability to predict two common symptoms during the third trimester of pregnancy—edema and fatigue—using physiological measures from wearable devices and self-reported daily surveys from mobile apps. Our approach employs a Graph Neural Network (GNN) framework, enhanced with a modified weighted Cross-Entropy loss to improve prediction performance. The model achieved AUC scores of 77.27% for edema and 69.70% for fatigue. Additionally, we aligned data from self-reported pregnancy symptoms with clinical examinations to carefully select participant cohorts for our experiments. We also explored how various features identified by the GNN are linked to these symptoms, gaining deeper insights into the relationship between physiological measures and pregnancy symptoms. Our findings indicate that heart rate variability plays a significant role in predicting symptoms of edema and fatigue, and features related to low-intensity activity also have a notable impact. Some of our findings closely align with previous studies on pregnancy. Our research serves as a proof of concept that symptoms can be predicted using wearable data, which may enhance the immediate well-being of expectant mothers and potentially personalize the overall pregnancy experience.
APA
Zhu, R., Yu, J., Friend, S.H., Goodday, S.M., Wang, B. & Goldenberg, A.. (2025). Towards a personalized pregnancy experience: Forecasting symptoms using graph neural networks and digital health technologies. Proceedings of the 4th Machine Learning for Health Symposium, in Proceedings of Machine Learning Research 259:1104-1120 Available from https://proceedings.mlr.press/v259/zhu25a.html.

Related Material