arXiv:2603.13275cs.LGcs.AI2026-03

用临床数据增强大模型,精准预测手术时长。

PREBA: Surgical Duration Prediction via PCA-Weighted Retrieval-Augmented LLMs and Bayesian Averaging Aggregation

  • 通过主成分加权检索历史病例,构建真实医疗上下文提示。
  • 融合贝叶斯平均与统计先验,将预测误差降低40%。
  • 无需训练,适合医院快速部署,尤其适合数据少的机构。

准确预测手术时长对医院资源管理至关重要。尽管近期监督学习方法(从机器学习到微调的大语言模型)表现强劲,但受限于高质量标注数据需求和高计算成本。相比之下,零样本大模型推理虽无需训练,却缺乏机构特定临床背景(如本地人口特征与病例结构),导致预测临床不一致且不稳定。为此,我们提出PREBA,一种基于检索增强的框架,结合主成分加权检索与贝叶斯平均聚合,将大模型预测锚定在机构特异性临床证据与统计先验上。其核心是为大模型构造基于证据的提示,包含(1)最相似的历史手术案例;(2)临床统计先验。PREBA首先将异构临床特征编码至统一表示空间,实现系统化检索;随后进行主成分加权检索,选出临床相关历史病例作为提示上下文输入大模型;最后应用贝叶斯平均,融合多轮大模型预测与人群统计先验,生成校准且临床合理的时长估计。我们在两个真实世界临床数据集上,使用Qwen3、DeepSeek-R1和HuatuoGPT-o1三个先进大模型评估PREBA,结果表明其显著提升性能:例如,最大降低40%的平均绝对误差(MAE),并将R²从-0.13提升至0.62,超越零样本推理,并达到与监督学习方法相当的精度,展现出强有效性和泛化能力。

原文摘要 · Abstract (English)

Accurate prediction of surgical duration is pivotal for hospital resource management. Although recent supervised learning approaches-from machine learning (ML) to fine-tuned large language models (LLMs)-have shown strong performance, they remain constrained by the need for high-quality labeled data and computationally intensive training. In contrast, zero-shot LLM inference offers a promising training-free alternative but it lacks grounding in institution-specific clinical context (e.g., local demographics and case-mix distributions), making its predictions clinically misaligned and prone to instability. To address these limitations, we present PREBA, a retrieval-augmented framework that integrates PCA-weighted retrieval and Bayesian averaging aggregation to ground LLM predictions in institution-specific clinical evidence and statistical priors. The core of PREBA is to construct an evidence-based prompt for the LLM, comprising (1) the most clinically similar historical surgical cases and (2) clinical statistical priors. To achieve this, PREBA first encodes heterogeneous clinical features into a unified representation space enabling systematic retrieval. It then performs PCA-weighted retrieval to identify clinically relevant historical cases, which form the evidence context supplied to the LLM. Finally, PREBA applies Bayesian averaging to fuse multi-round LLM predictions with population-level statistical priors, yielding calibrated and clinically plausible duration estimates. We evaluate PREBA on two real-world clinical datasets using three state-of-the-art LLMs, including Qwen3, DeepSeek-R1, and HuatuoGPT-o1. PREBA significantly improves performance-for instance, reducing MAE by up to 40% and raising R^2 from -0.13 to 0.62 over zero-shot inference-and it achieves accuracy competitive with supervised ML methods, demonstrating strong effectiveness and generalization.

手术预测大模型临床推理贝叶斯

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