9 papers
Understanding Clustering in Slot Attention via Particle Dynamics
Vasudev Joy, Rajat Rasal, Avinash Kori +2
Studying attention through the lens of interacting particle dynamics has shown how token clustering can emerge from the underlying dynamics. We extend this perspective to slot atte…
Generating Chest X-Ray Counterfactuals by Specialising Foundation Image Models
Xiaodan Xing, Rajat R. Rasal, Julia A. Meister +3
Counterfactual image generation answers questions about how a subject would have looked under retrospective, hypothetical scenarios. Recent methods have improved perceptual quality…
Steering Optimisation Trajectories in Diffusion Representation Learning
Rajat Rasal, Avinash Kori, Tian Xia +1
We study why diffusion autoencoders can achieve similar image quality while learning substantially different latent structures. We trace this behaviour to optimisation dynamics; we…
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…
Segmentor-Guided Counterfactual Fine-Tuning for Locally Coherent and Targeted Image Synthesis
Tian Xia, Matthew Sinclair, Andreas Schuh +8
Counterfactual image generation is a powerful tool for augmenting training data, de-biasing datasets, and modeling disease. Current approaches rely on external classifiers or regre…
Diffusion Counterfactual Generation with Semantic Abduction
Rajat Rasal, Avinash Kori, Fabio De Sousa Ribeiro +2
Counterfactual image generation presents significant challenges, including preserving identity, maintaining perceptual quality, and ensuring faithfulness to an underlying causal mo…
Causal Representation Learning with Observational Grouping for CXR Classification
Rajat Rasal, Avinash Kori, Ben Glocker
Identifiable causal representation learning seeks to uncover the true causal relationships underlying a data generation process. In medical imaging, this presents opportunities to…
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…
Deep Structural Causal Shape Models
Rajat Rasal, Daniel C. Castro, Nick Pawlowski +1
Causal reasoning provides a language to ask important interventional and counterfactual questions beyond purely statistical association. In medical imaging, for example, we may wan…