SurvivalPFN: Amortizing Survival Prediction via In-Context Bayesian Inference

Note This repository is a placeholder for the upcoming arXiv release. Code, pretrained checkpoints, and reproduction instructions will be added shortly.

Authors

• Shi-ang Qi
• Vahid Balazadeh
• Michael Cooper
• Russell Greiner
• Rahul G. Krishnan

Abstract

Coming soon.

Paper

• arXiv: link coming soon

Repository status

Item	Status
Paper (arXiv)	TBD
Code release	TBD
Pretrained checkpoints	TBD
Reproduction instructions	TBD

Installation

Coming soon.

# git clone https://github.com/rgklab/SurvivalPFN.git
# cd SurvivalPFN
# pip install -e .

Quick start

Coming soon.

Citation

If you use this work, please cite:

@article{qi2026survivalpfn,
  title   = {SurvivalPFN: Amortizing Survival Prediction via In-Context Bayesian Inference},
  author  = {Qi, Shi-ang and Balazadeh, Vahid and Cooper, Michael and Greiner, Russell and Krishnan, Rahul G.},
  journal = {arXiv preprint},
  year    = {2026}
}

License

Released under the Apache License 2.0.