Background <p>Machine learning provides a powerful framework to model the complex patterns underlying migraine attack onset from real-world high dimensional datasets. In this study, we used machine learning to forecast headache days using mobile health (mHealth) data from a migraine biofeedback treatment app.</p> Method <p>This was a machine learning analysis of data from the BioCer clinical trial (NCT05616741) evaluating app-based biofeedback for preventive treatment of episodic migraine. Participants completed three months of daily biofeedback sessions with wearables measuring trapezius muscle tension, heart rate variability, and peripheral skin temperature. Input data for the models included summary metrics from the biofeedback sessions and daily headache diary entries. The outcome of interest was the presence of a moderate-to-severe headache (defined as an intensity of 4 or higher on an 11-point scale of 0–10) on the next calendar day and the next three calendar days. The dataset was randomly split into training, validation, and test sets. Multiple standard machine learning architectures, foundation models, and time-series models were trained and optimized using the area under the receiver operating characteristics curve (AUC) as the primary scoring metric. Among these three classes of machine learning models, the best optimized model in each class identified during training was applied on the unseen test set. Permutation feature importance (PFI) was created for model explainability.</p> Results <p>146 individuals, with a total of 21,550 headache days, were included in the forecasting models. For the next calendar day predictions, the top performing standard machine learning approach (decision tree) and foundation model achieved a test set AUC of 0.59 (95% CI 0.56 to 0.61) and 0.55 (95% CI 0.55 to 0.56), respectively. The best time-series model achieved a test set AUC of 0.84 (95% CI 0.82 to 0.85). For the three-calendar day forecasting window, the test set performances were 0.55 (95% CI 0.53 to 0.56), 0.55 (95% CI 0.54 to 0.57), and 0.76 (95% CI 0.74 to 0.77), respectively. The most important features were headache intensity, duration of the headache, and heart rate scores.</p> Conclusion <p>Time-series machine learning models using a relatively large dataset could forecast moderate-to-severe headaches with good accuracy in patients with episodic migraine.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Forecasting migraine with time-series machine learning from mobile health data

  • Fahim Faisal,
  • Amalie C. Poole,
  • Antonios Danelakis,
  • Tore Wergeland,
  • Marte Bjørk,
  • Andrej N. Khanevski,
  • Espen Saxhaug Kristoffersen,
  • Tjaša Kumelj,
  • Iben C. K. Larsen,
  • Oda V. Lunder,
  • Manjit Matharu,
  • Parashkev Nachev,
  • Melanie R. Simpson,
  • Erling Tronvik,
  • Kjersti G. Vetvik,
  • Bendik S. Winsvold,
  • Lise R. Øie,
  • Ane B. Øvrevik,
  • Anker Stubberud

摘要

Background

Machine learning provides a powerful framework to model the complex patterns underlying migraine attack onset from real-world high dimensional datasets. In this study, we used machine learning to forecast headache days using mobile health (mHealth) data from a migraine biofeedback treatment app.

Method

This was a machine learning analysis of data from the BioCer clinical trial (NCT05616741) evaluating app-based biofeedback for preventive treatment of episodic migraine. Participants completed three months of daily biofeedback sessions with wearables measuring trapezius muscle tension, heart rate variability, and peripheral skin temperature. Input data for the models included summary metrics from the biofeedback sessions and daily headache diary entries. The outcome of interest was the presence of a moderate-to-severe headache (defined as an intensity of 4 or higher on an 11-point scale of 0–10) on the next calendar day and the next three calendar days. The dataset was randomly split into training, validation, and test sets. Multiple standard machine learning architectures, foundation models, and time-series models were trained and optimized using the area under the receiver operating characteristics curve (AUC) as the primary scoring metric. Among these three classes of machine learning models, the best optimized model in each class identified during training was applied on the unseen test set. Permutation feature importance (PFI) was created for model explainability.

Results

146 individuals, with a total of 21,550 headache days, were included in the forecasting models. For the next calendar day predictions, the top performing standard machine learning approach (decision tree) and foundation model achieved a test set AUC of 0.59 (95% CI 0.56 to 0.61) and 0.55 (95% CI 0.55 to 0.56), respectively. The best time-series model achieved a test set AUC of 0.84 (95% CI 0.82 to 0.85). For the three-calendar day forecasting window, the test set performances were 0.55 (95% CI 0.53 to 0.56), 0.55 (95% CI 0.54 to 0.57), and 0.76 (95% CI 0.74 to 0.77), respectively. The most important features were headache intensity, duration of the headache, and heart rate scores.

Conclusion

Time-series machine learning models using a relatively large dataset could forecast moderate-to-severe headaches with good accuracy in patients with episodic migraine.