16 papers
Scaling Generative Foundation Models for Chest Radiography with Rectified Flow Transformers
Fabio De Sousa Ribeiro, Emma A. M. Stanley, Charles Jones +7
We introduce the first generative foundation model for chest radiograph synthesis trained from scratch at the billion-parameter scale. Existing radiographic AI models often suffer…
Counterfactual Stress Testing for Image Classification Models
Moritz Stammel, Fabio De Sousa Ribeiro, Raghav Mehta +3
Deep learning models in medical imaging often fail when deployed in new clinical environments due to distribution shifts in demographics, scanner hardware, or acquisition protocols…
Factored Classifier-Free Guidance
Tian Xia, Fabio De Sousa Ribeiro, Rajat R Rasal +3
Counterfactual generation aims to simulate realistic hypothetical outcomes under causal interventions. Diffusion models have emerged as a powerful tool for this task, combining DDI…
Counterfactual Identifiability via Dynamic Optimal Transport
Fabio De Sousa Ribeiro, Ainkaran Santhirasekaram, Ben Glocker
We address the open question of counterfactual identification for high-dimensional multivariate outcomes from observational data. Pearl (2000) argues that counterfactuals must be i…
Positional Segmentor-Guided Counterfactual Fine-Tuning for Spatially Localized Image Synthesis
Tian Xia, Matthew Sinclair, Andreas Schuh +8
Counterfactual image generation enables controlled data augmentation, bias mitigation, and disease modeling. However, existing methods guided by external classifiers or regressors…
Latent Causal Modeling for 3D Brain MRI Counterfactuals
Wei Peng, Tian Xia, Fabio De Sousa Ribeiro +5
The number of samples in structural brain MRI studies is often too small to properly train deep learning models. Generative models show promise in addressing this issue by effectiv…