arXiv:2603.29475cs.LG2026-03被引 2

用合成数据预训练模型,实现无需调参的精准生存预测

Survival In-Context: Amortized Bayesian Survival Analysis via Prior-Fitted Networks

  • 基于可控制的合成生存数据构建先验,支持灵活生成
  • 单次前向传播即完成个体化生存预测,无需微调
  • 小样本场景下性能优于传统与深度模型,适合医疗数据分析

生存分析在医疗领域至关重要,但受限于数据稀少、删失现象及表格式协变量异质性,对现代机器学习构成挑战。尽管基于先验拟合的范式(通过大规模合成数据集预训练)已推动分类与回归类表格式基础模型发展,其在生存分析中的适用性尚不明确。本文提出一种灵活的生存数据生成框架,定义了具有显式协变量与事件时间分布控制能力的丰富生存先验。基于此先验,我们引入生存上下文学习(Survival In-Context, SIC),一种仅在合成数据上预训练的先验拟合上下文学习模型。SIC被训练以逼近合成生存先验下的贝叶斯后验预测分布,可在单次前向传播中实现个体化生存预测,无需任务特定训练或超参数调优。在多个真实世界生存数据集上的广泛评估显示,SIC在小样本和中等规模数据场景下表现优于或媲美经典与深度生存模型,凸显先验拟合范式在生存分析中的潜力。代码与预训练模型将在发表后公开。

原文摘要 · Abstract (English)

Survival analysis is crucial for many medical applications, but remains challenging for modern machine learning due to limited data, censoring, and the heterogeneity of tabular covariates. While the prior-fitted paradigm, which relies on pretraining models on large collections of synthetic datasets, has recently facilitated tabular foundation models for classification and regression, its suitability for time-to-event modeling remains unclear. We propose a flexible survival data generation framework that defines a rich survival prior with explicit control over covariates and time-event distributions. Building on this prior, we introduce Survival In-Context (SIC), a prior-fitted in-context learning model for survival analysis that is pretrained exclusively on synthetic data. SIC is trained to approximate Bayesian posterior predictive inference under the synthetic survival prior, enabling individualized survival prediction in a single forward pass, requiring no task-specific training or hyperparameter tuning. Across a broad evaluation on real-world survival datasets, SIC achieves competitive or superior performance compared to classical and deep survival models, particularly in small and medium-sized data regimes, highlighting the promise of a prior-fitted paradigm for survival analysis. The code and pretrained models will be made available upon publication.

生存分析先验拟合医疗AI合成数据

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