医学纵向数据预测新模型 LongAgent 发布
LongAgent: History-Guided Agentic Search for Longitudinal Outcome Prediction
医学领域的新模型,能自主搜索变量组合,提升预测精度。
针对医疗数据异构性难题,提出 LongAgent 模型。该模型能自主搜索变量组合和时序窗口,利用历史搜索记录指导后续探索。在合成数据上,其预测均方根误差达 1.7376,优于最强非代理基线 0.0151。
LongAgent: History-Guided Agentic Search for Longitudinal Outcome Prediction
Extracting informative representations from longitudinal data that can predict future outcomes remains a critical challenge in medicine. Medical datasets are inherently heterogeneous, consisting of a large number of variables collected from different sources, sampled with different temporal spacings, and representing different aspects of human health status. This requires identifying those variables with predictive value, processing longitudinal information, and integrating multiple variables for outcome prediction. Here, we propose a novel agent-based approach, LongAgent, that can autonomously search over combinations of variable sets, temporal windows and longitudinal aggregation functions, and identify candidates with promising predictive performance. LongAgent utilises a history memory of previous searches and numerical evidence to guide subsequent exploration. On synthetic data, LongAgent achieves a mean prediction RMSE of 1.7376 and improves over the strongest non-agent baseline by 0.0151 (95% CI: [0.0045,0.0260]; p=0.0273). On a real clinical dataset, it performs comparably to the best baseline.