arXiv:2601.19286cs.CL2026-01中稿 · WSDM 2026

用大模型重写病历数据,让预测更准

ReToP: Learning to Rewrite Electronic Health Records for Clinical Prediction

  • 让大模型同时学习重写病历和做临床预测,端到端训练
  • 在MIMIC-IV上三个任务均超越现有方法,提升明显
  • 适合医疗AI研究者,尤其关注病历建模与可解释性

电子健康记录(EHR)是临床决策的关键信息源,但其高维、异构和稀疏特性使预测任务困难。尽管大语言模型(LLMs)通过引入医学知识提升了EHR表示能力,但现有方法大多任务无关,仅将LLM当作编码器或补全模块,未能充分融合预测任务信号,影响性能。本文提出重构-预测框架ReToP,通过端到端训练病历重写器与临床预测器协同优化。针对缺乏重写标注数据的问题,采用基于临床特征选择的策略生成合成伪标签,构建多样化的患者重写样本用于微调。引入新型分类器监督贡献(CSC)评分,引导重写器生成直接增强预测的临床相关改写内容。ReToP在MIMIC-IV数据集上三个临床预测任务中均优于强基线模型。分析显示,该框架具备良好泛化能力,可在未见数据集与任务上仅经少量微调即保持忠实重写,并聚焦任务相关特征。

原文摘要 · Abstract (English)

Electronic Health Records (EHRs) provide crucial information for clinical decision-making. However, their high-dimensionality, heterogeneity, and sparsity make clinical prediction challenging. Large Language Models (LLMs) allowed progress towards addressing this challenge by leveraging parametric medical knowledge to enhance EHR data for clinical prediction tasks. Despite the significant achievements made so far, most of the existing approaches are fundamentally task-agnostic in the sense that they deploy LLMs as EHR encoders or EHR completion modules without fully integrating signals from the prediction tasks. This naturally hinders task performance accuracy. In this work, we propose Rewrite-To-Predict (ReToP), an LLM-based framework that addresses this limitation through an end-to-end training of an EHR rewriter and a clinical predictor. To cope with the lack of EHR rewrite training data, we generate synthetic pseudo-labels using clinical-driven feature selection strategies to create diverse patient rewrites for fine-tuning the EHR rewriter. ReToP aligns the rewriter with prediction objectives using a novel Classifier Supervised Contribution (CSC) score that enables the EHR rewriter to generate clinically relevant rewrites that directly enhance prediction. Our ReToP framework surpasses strong baseline models across three clinical tasks on MIMIC-IV. Moreover, the analysis of ReToP shows its generalizability to unseen datasets and tasks with minimal fine-tuning while preserving faithful rewrites and emphasizing task-relevant predictive features.

医疗AI大模型病历生成预测模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。