English
Related papers

Related papers: SurvivalPFN: Amortizing Survival Prediction via In…

200 papers

Survival models are used in various fields, such as the development of cancer treatment protocols. Although many statistical and machine learning models have been proposed to achieve accurate survival predictions, little attention has been…

Machine Learning · Computer Science 2020-03-26 Hrushikesh Loya , Pranav Poduval , Deepak Anand , Neeraj Kumar , Amit Sethi

Censoring is the central problem in survival analysis where either the time-to-event (for instance, death), or the time-tocensoring (such as loss of follow-up) is observed for each sample. The majority of existing machine learning-based…

Machine Learning · Statistics 2024-07-22 Weijia Zhang , Chun Kai Ling , Xuanhui Zhang

Survival analysis is a widely used statistical framework for modeling time-to-event data under censoring. Classical methods, such as the Cox proportional hazards (Cox PH) model, offer a semiparametric approach to estimating the effects of…

Machine Learning · Statistics 2026-04-23 Yang Xu , Wenbin Lu , Rui Song

Survival analysis is a crucial semi-supervised task in machine learning with numerous real-world applications, particularly in healthcare. Currently, the most common approach to survival analysis is based on Cox's partial likelihood, which…

Machine Learning · Computer Science 2023-04-27 Andre Vauvelle , Benjamin Wild , Aylin Cakiroglu , Roland Eils , Spiros Denaxas

Adaptive experimentation enables efficient estimation of causal effects, but existing methods are not designed for survival data with censoring, where event times are only partially observed (e.g., overall survival in cancer trials but with…

Machine Learning · Computer Science 2026-05-19 Yuxin Wang , Dennis Frauen , Jonas Schweisthal , Maresa Schröder , Emil Javurek , Stefan Feuerriegel

One straightforward metric to evaluate a survival prediction model is based on the Mean Absolute Error (MAE) -- the average of the absolute difference between the time predicted by the model and the true event time, over all subjects.…

Machine Learning · Computer Science 2023-06-05 Shi-ang Qi , Neeraj Kumar , Mahtab Farrokh , Weijie Sun , Li-Hao Kuan , Rajesh Ranganath , Ricardo Henao , Russell Greiner

We propose a deep generative approach to nonparametric estimation of conditional survival and hazard functions with right-censored data. The key idea of the proposed method is to first learn a conditional generator for the joint conditional…

Statistics Theory · Mathematics 2022-05-20 Xingyu Zhou , Wen Su , Changyu Liu , Yuling Jiao , Xingqiu Zhao , Jian Huang

Simulation-based inference (SBI) offers a flexible and general approach to performing Bayesian inference: In SBI, a neural network is trained on synthetic data simulated from a model and used to rapidly infer posterior distributions for…

Machine Learning · Computer Science 2025-10-28 Julius Vetter , Manuel Gloeckler , Daniel Gedon , Jakob H. Macke

Bayesian inference provides a natural way of incorporating prior beliefs and assigning a probability measure to the space of hypotheses. Current solutions rely on iterative routines like Markov Chain Monte Carlo (MCMC) sampling and…

Machine Learning · Computer Science 2025-02-11 Sarthak Mittal , Niels Leif Bracher , Guillaume Lajoie , Priyank Jaini , Marcus Brubaker

Prior-data fitted networks (PFNs) were recently proposed as a new paradigm for machine learning. Instead of training the network to an observed training set, a fixed model is pre-trained offline on small, simulated training sets from a…

Machine Learning · Statistics 2023-05-19 Thomas Nagler

Prior-data fitted networks (PFNs) have recently emerged as a powerful approach for Bayesian prediction tasks, approximating the posterior predictive distribution (PPD) through in-context learning. Despite their strong empirical performance…

Machine Learning · Statistics 2026-05-27 Gyeonghun Kang , Changwoo J. Lee , Xiang Cheng

In this paper, we propose a flexible model for survival analysis using neural networks along with scalable optimization algorithms. One key technical challenge for directly applying maximum likelihood estimation (MLE) to censored data is…

Machine Learning · Statistics 2021-12-07 Weijing Tang , Jiaqi Ma , Qiaozhu Mei , Ji Zhu

The diversity of time series applications and scarcity of domain-specific data highlight the need for time-series models with strong few-shot learning capabilities. In this work, we propose a novel training scheme and a transformer-based…

Machine Learning · Computer Science 2025-02-25 Ege Onur Taga , M. Emrullah Ildiz , Samet Oymak

Accurate time-to-event prediction is integral to decision-making, informing medical guidelines, hiring decisions, and resource allocation. Survival analysis, the quantitative framework used to model time-to-event data, accounts for patients…

Machine Learning · Computer Science 2025-08-08 Vincent Jeanselme , Brian Tom , Jessica Barrett

Prediction methods for time-to-event outcomes often utilize survival models that rely on strong assumptions about noninformative censoring or on how individual-level covariates and survival functions are related. When the main interest is…

Methodology · Statistics 2024-02-29 Mahsa Ashouri , Nicholas C. Henderson

Interval-censored data analysis is important in biomedical statistics for any type of time-to-event response where the time of response is not known exactly, but rather only known to occur between two assessment times. Many clinical trials…

Methodology · Statistics 2019-06-12 Weichi Yao , Halina Frydman , Jeffrey S. Simonoff

Traditional statistical and machine learning methods typically assume that the training and test data follow the same distribution. However, this assumption is frequently violated in real-world applications, where the training data in the…

Methodology · Statistics 2025-07-08 Hanxuan Ye , Hongzhe Li

We describe a new approach to estimating relative risks in time-to-event prediction problems with censored data in a fully parametric manner. Our approach does not require making strong assumptions of constant proportional hazard of the…

Machine Learning · Computer Science 2021-06-10 Chirag Nagpal , Xinyu Rachel Li , Artur Dubrawski

Spatial survival analysis has received a great deal of attention over the last 20 years due to the important role that geographical information can play in predicting survival. This paper provides an introduction to a set of programs for…

Computation · Statistics 2018-04-25 Haiming Zhou , Timothy Hanson , Jiajia Zhang

This paper considers doing quantile regression on censored data using neural networks (NNs). This adds to the survival analysis toolkit by allowing direct prediction of the target variable, along with a distribution-free characterisation of…

Machine Learning · Statistics 2023-02-07 Tim Pearce , Jong-Hyeon Jeong , Yichen Jia , Jun Zhu