arXiv · 2609.15647
MedDiME: Efficient Latent Diffusion with Adaptive Masking for Medical Counterfactual Generation
Abstract
Medical counterfactual generation modifies images to change model predictions for interpretability. However, existing diffusion-based approaches are often prohibitively slow and memory-intensive, making them difficult to apply in high-resolution settings. Moreover, existing masking strategies are tightly coupled with pixel-space representations, making them incompatible with latent-space diffusion editing. To address these challenges, we propose MedDiME, a latent-space classifier-guided diffusion framework that reduces computational and memory overhead while introducing a latent-compatible, gradient-driven adaptive masking mechanism for spatially precise medical counterfactual generation. Extensive experiments demonstrate that MedDiME achieves high-quality counterfactual generation with significant efficiency gains compared to prior classifier-guided diffusion baselines, achieving up to 40 times faster inference and 13 times lower peak GPU memory usage.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Yan Zeng, Changlu Guo, Anders Nymark Christensen, Morten Rieger Hannemose, Anders Bjorholm Dahl. 2026-09-14. MedDiME: Efficient Latent Diffusion with Adaptive Masking for Medical Counterfactual Generation. https://arxiv.org/abs/2609.15647
Cite the original work for its findings. Save a collection to share your selection of sources.