Developing predictive models for COVID-19 positive tests based on the XGBoost and random forest algorithms with internet search data

作者：Yikun Chang, Jinwei Chen, Xiaoxuan Chen, Yueqian Wu, Hui Tang, Gonghua Wu, Jie Sun, Yuyi Liao, Haolin Chen, Senyao Cai, Yuantao Hao, Wangjian Zhang, Zhicheng Du · 发表于：BMC Public Health · 年份：2025 · DOI：10.1186/s12889-025-25569-w · 被引用次数：2 · 研究领域：Data-Driven Disease Surveillance、COVID-19 epidemiological studies、COVID-19 diagnosis using AI

BACKGROUND: Although strategies for COVID-19 have shifted towards normalized measures globally, establishing predictive models based on Internet search data remains crucial for swiftly controlling and preventing future outbreaks. This study aims to utilize Internet search data for early epidemic surveillance and warning. METHODS: We collected the daily number of COVID-19 positive tests and the daily Baidu Search Index (BSI) of COVID-19 related keywords. First, we screened keywords with a maximum correlation coefficient exceeding 0.9 by time-lagged correlation analysis. Then, we used the original and lagged BSI to construct XGBoost and Random Forest (RF) models for short-term prediction of the COVID-19, respectively. Next, we selected top 5 important predictors according to the importance gain in XGBoost model and constructed a comprehensive search index (CSI) weighted by the importance gain. Finally, we used the distributed lagged nonlinear model (DLNM) to evaluate the relationship between the CSI and the number of COVID-19 positive tests. RESULTS: We identified 20 keywords had a maximum correlation coefficient exceeding 0.9 with lag days of 1-10 days. Then, we found that the predictive performance of the XGBoost models was better than that of the RF models. And XGBoost models using lagged BSI (compared to original BSI) had a better predictive performance for forecasting 3 days, with an RMSE of 803.85 and a MAPE of 9.96%. Finally, we observed that the CSI was statistically as...