Adaptation Interfaces for In-Context Tabular Foundation Models in Time-to-Event Prediction
Not on the map. Why?
Abstract (arXiv)
Tabular foundation models (TabFMs) achieve strong performance on structured data, particularly for standard classification and regression problems. Yet, extending them to censored time-to-event prediction is challenging because it requires properly handling censoring and event-time dynamics. Building on our prior work, we further link TabFMs with CoxPH and DeepHit and revise the context-resampled training procedure. We evaluate temporal zero-shot reformulation, classification-based fine-tuning, and survival-head adaptation using frozen TabFM backbones on 74 single-risk data sets, and we additionally study 4 competing-risk data sets. Zero-shot inference is effective on smaller single-risk data sets, whereas supervised adaptation becomes increasingly advantageous as data sets scale. Cox provides the most reliably strong interface, especially for Integrated Brier Score (IBS) on larger data sets. DeepHit is relatively stronger for the time-dependent Concordance Index than for IBS, while cause-specific MTLR ranks highest among the TabFM survival heads in the four-data-set competing-risk analysis. Classification fine-tuning becomes more competitive with zero-shot inference as data sets grow but remains weaker for probabilistic prediction. Overall, our results indicate that effective TabFM transfer depends on the data regime and on the statistical structure represented by the chosen adaptation interface. The implementation scripts used for this work are available at https://github.com/kaylode/survival-fm.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 12 of its 27 references that are in the map. First posted on arXiv 4 Sep 2026.
Builds on
12 of 27 references.
References · 12
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
- 2019Time-to-Event Prediction with Neural Networks and Cox Regressioncited by 175
- 2024TabDPT: Scaling Tabular Foundation Models on Real Datacited by 113
- 2025From Tables to Time: Extending TabPFN-v2 to Time Series Forecastingcited by 60
- 2021SurvTRACE: Transformers for Survival Analysis with Competing Eventscited by 41
- 2025TabPFN Unleashed: A Scalable and Effective Solution to Tabular Classification Problemscited by 36
- 2026Beyond IID: How General Are Tabular Foundation Models, Really?cited by 20
- 2022SurvSet: An open-source time-to-event dataset repositorycited by 17
- 2026Tabular Foundation Models Can Do Survival Analysiscited by 9
- 2026Survival In-Context: Amortized Bayesian Survival Analysis via Prior-Fitted Networkscited by 6
- 2026Retrieval-aligned Tabular Foundation Models Enable Robust Clinical Risk Prediction in Electronic Health Records Under Real-world Constraintscited by 4
- 2026SurvivalPFN: Amortizing Survival Prediction via In-Context Bayesian Inferencecited by 3
Built on by
Not cited yet.