Decoupled Classifier-Free Guidance for Counterfactual Diffusion Models
Xia, Tian, Ribeiro, Fabio De Sousa, Rasal, Rajat R, Kori, Avinash, Mehta, Raghav, Glocker, Ben
–arXiv.org Artificial Intelligence
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 of CFG for counterfactual generation: it prescribes a global guidance scale for all attributes, leading to significant spurious changes in inferred counterfactuals. To mitigate this, we propose Decoupled Classifier-Free Guidance (DCFG), a flexible and model-agnostic guidance technique that enables attribute-wise control following a causal graph. DCFG is implemented via a simple attribute-split embedding strategy that disentangles semantic inputs, enabling selective guidance on user-defined attribute groups.
arXiv.org Artificial Intelligence
Oct-1-2025
- Country:
- Europe (0.28)
- Genre:
- Research Report > New Finding (0.92)
- Industry:
- Health & Medicine > Diagnostic Medicine > Imaging (1.00)
- Technology: