← Search

Avinash Kori

7 accepted papers

2026

Factored Classifier-Free Guidance

ICML 2026poster

Counterfactual generation aims to simulate realistic hypothetical outcomes under causal interventions. Diffusion models have emerged as a powerful tool for this task, combining DDIM inversion with conditional generation and classifier-free guidance (CFG). In this work, we identify a key limitation o…

Cited by 0SourceScholar
2025

Continuous Bayesian Model Selection for Multivariate Causal Discovery

ICML 2025poster

Current causal discovery approaches require restrictive model assumptions in the absence of interventional data to ensure structure identifiability. These assumptions often do not hold in real-world applications leading to a loss of guarantees and poor performance in practice. Recent work has shown…

Cited by 1SourcePDFScholar
2025

Diffusion Counterfactual Generation with Semantic Abduction

ICML 2025poster

Counterfactual image generation presents significant challenges, including preserving identity, maintaining perceptual quality, and ensuring faithfulness to an underlying causal model. While existing auto-encoding frameworks admit semantic latent spaces which can be manipulated for causal control, t…

2025

Flow Stochastic Segmentation Networks

ICCV 2025poster

We propose the Flow Stochastic Segmentation Network (Flow-SSN), a generative model for probabilistic segmentation featuring discrete-time autoregressive and modern continuous-time flow parameterisations. We prove fundamental limitations of the low-rank parameterisation of previous methods and show t…

2024

Grounded Object-Centric Learning

ICLR 2024poster

The extraction of object-centric representations for downstream tasks is an emerging area of research. Learning grounded representations of objects that are guaranteed to be stable and invariant promises robust performance across different tasks and environments. Slot Attention (SA) learns object-ce…

Cited by 8SourcePDFScholar
2024

Identifiable Object-Centric Representation Learning via Probabilistic Slot Attention

NeurIPS 2024poster

Learning modular object-centric representations is said to be crucial for systematic generalization. Existing methods show promising object-binding capabilities empirically, but theoretical identifiability guarantees remain relatively underdeveloped. Understanding when object-centric representations…