3 papers
cs.LG2026
SurvivalPFN: Amortizing Survival Prediction via In-Context Bayesian Inference
Shi-ang Qi, Vahid Balazadeh, Michael Cooper +2
Survival analysis provides a powerful statistical framework for modeling time-to-event outcomes in the presence of censoring. However, selecting an appropriate estimator from the m…
cs.CL2025
Red Teaming Large Language Models for Healthcare
Vahid Balazadeh, Michael Cooper, David Pellow +32
We present the design process and findings of the pre-conference workshop at the Machine Learning for Healthcare Conference (2024) entitled Red Teaming Large Language Models for He…
cs.LG2025
Diverse Prototypical Ensembles Improve Robustness to Subpopulation Shift
Minh Nguyen Nhat To, Paul F RWilson, Viet Nguyen +6
The subpopulationtion shift, characterized by a disparity in subpopulation distributibetween theween the training and target datasets, can significantly degrade the performance of…