Latent Drifting in Diffusion Models for Counterfactual Medical Image Synthesis
Yousef Yeganeh, Azade Farshad, Ioannis Charisiadis, Marta Hasny, Martin Hartenberger, Björn Ommer, Nassir Navab, Ehsan Adeli
Abstract
ing enables diffusion models to be conditioned for medical images fitted for the complex task of counterfactual image generation, which is crucial to investigate how parameters such as gender, age, and adding or removing diseases in a patient would alter the medical images. We evaluate our method on three public longitudinal benchmark datasets of brain MRI and chest X-rays for counterfactual image generation. Our results demonstrate significant performance gains in various scenarios when combined with different fine-tuning schemes.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f69b1e5d-73cd-44a3-bfaf-09e813249eb2Cited by top-tier papers4
- Causal-Adapter: Taming Text-to-Image Diffusion for Faithful Counterfactual GenerationLei Tong, Zhihua Liu, Chaochao Lu, Dino Oglic et al.ICML 2026 · 2 citations
- MaskDiME: Adaptive Masked Diffusion for Precise and Efficient Visual Counterfactual ExplanationsChanglu Guo, Anders Nymark Christensen, Anders Bjorholm Dahl, Morten Rieger HannemoseCVPR 2026 · 2 citations
- Clinically-Grounded Counterfactual Reasoning for Medical Video DiagnosisJianzhe Gao, Churan Wang, Weiyi Zhang, Jianghua Li et al.CVPR 2026
- D2T2 - Multimodal Automated Planning for BrachytherapyLance C. Moore, Aranyo Mitra, Ryan Truong, Karoline Kallis et al.CVPR 2026
Builds on18
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Directly Denoising Diffusion ModelsDan Zhang, Jingjing Wang, Feng LuoICML 2024 · 11,724 citations
- Improved Denoising Diffusion Probabilistic ModelsAlexander Quinn Nichol, Prafulla DhariwalICML 2021 · 5,234 citations
Related papers
- Factored Classifier-Free GuidanceTian Xia, Fabio De Sousa Ribeiro, Rajat Rasal, Avinash Kori et al.ICML 2026
- Counterfactual Generative Models for Time-Varying TreatmentsShenghao Wu, Wenbin Zhou, Minshuo Chen, Shixiang ZhuKDD 2024 · 2 citations
- MAISI-v2: Accelerated 3D High-Resolution Medical Image Synthesis with Rectified Flow and Region-specific Contrastive LossCan Zhao, Pengfei Guo, Dong Yang, Yufan He et al.AAAI 2026 · 11 citations
- Foreground-Background Separation through Concept Distillation from Generative Image Foundation ModelsMischa Dombrowski, Hadrien Reynaud, Matthew Baugh, Bernhard KainzICCV 2023 · 9 citations
- Controllable Sequence Editing for Biological and Clinical TrajectoriesMichelle M. Li, Kevin Li, Yasha Ektefaie, Ying Jin et al.ICLR 2026
