arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.04086cs.LG
收藏

面向电子健康记录联合疾病预测的自动化多任务学习

Automated Multi-Task Learning for Joint Disease Prediction on Electronic Health Records

The Pennsylvania State University(宾夕法尼亚州立大学)

Suhan Cui, Prasenjit Mitra

AI总结:

针对电子健康记录多任务疾病预测框架依赖人工设计的问题,提出AutoDP自动化方法,用代理模型优化同时搜索任务分组与架构最优配置,性能优于现有方法且搜索成本可行。

摘要:

在大数据与数字医疗领域,电子健康记录(EHR)已成为可改善患者护理与医学研究的丰富信息来源。近年来,用于分析EHR数据以预测患者未来健康状况的机器学习模型大量涌现。其中,部分研究主张采用多任务学习(MTL)联合预测多种目标疾病,以提升单任务学习的预测性能。然而,当前面向EHR数据的MTL框架存在显著局限,因其高度依赖人类专家确定联合训练的任务分组并设计模型架构。为减少人工干预并优化框架设计,我们提出一种名为AutoDP的自动化方法,可同时搜索任务分组与架构的最优配置。为应对包含任务组合与架构的庞大联合搜索空间,我们采用基于代理模型的优化方法,能够高效发现最优解。在真实世界EHR数据上的实验结果证明了所提AutoDP框架的有效性,其性能相较于人工设计与自动化的现有最优方法均有显著提升,同时还保持了可行的搜索成本。源代码可通过链接获取:https://github.com/SH-Src/AutoDP。

英文摘要:

In the realm of big data and digital healthcare, Electronic Health Records (EHR) have become a rich source of information with the potential to improve patient care and medical research. In recent years, machine learning models have proliferated for analyzing EHR data to predict patients future health conditions. Among them, some studies advocate for multi-task learning (MTL) to jointly predict multiple target diseases for improving the prediction performance over single task learning. Nevertheless, current MTL frameworks for EHR data have significant limitations due to their heavy reliance on human experts to identify task groups for joint training and design model architectures. To reduce human intervention and improve the framework design, we propose an automated approach named AutoDP, which can search for the optimal configuration of task grouping and architectures simultaneously. To tackle the vast joint search space encompassing task combinations and architectures, we employ surrogate model-based optimization, enabling us to efficiently discover the optimal solution. Experimental results on real-world EHR data demonstrate the efficacy of the proposed AutoDP framework. It achieves significant performance improvements over both hand-crafted and automated state-of-the-art methods, also maintains a feasible search cost at the same time. Source code can be found via the link: \url{https://github.com/SH-Src/AutoDP}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑