English

CardioComposer: Leveraging Differentiable Geometry for Compositional Control of Anatomical Diffusion Models

Image and Video Processing 2026-03-18 v3 Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning

Abstract

Generative models of 3D cardiovascular anatomy can synthesize informative structures for clinical research and medical device evaluation, but face a trade-off between geometric controllability and realism. We propose CardioComposer: a programmable, inference-time framework for generating multi-class anatomical label maps from interpretable ellipsoidal primitives. These primitives represent geometric attributes such as the size, shape, and position of discrete substructures. We specifically develop differentiable measurement functions based on voxel-wise geometric moments, enabling loss-based gradient guidance during diffusion model sampling. We demonstrate that these losses can constrain individual geometric attributes in a disentangled manner and provide compositional control over multiple substructures. Finally, we show that our method is compatible with a broad range of anatomical systems containing non-convex substructures, spanning cardiac, vascular, and skeletal organs. We release our code at https://github.com/kkadry/CardioComposer.

Keywords

Cite

@article{arxiv.2509.08015,
  title  = {CardioComposer: Leveraging Differentiable Geometry for Compositional Control of Anatomical Diffusion Models},
  author = {Karim Kadry and Shoaib Goraya and Ajay Manicka and Abdalla Abdelwahed and Naravich Chutisilp and Farhad Nezami and Elazer Edelman},
  journal= {arXiv preprint arXiv:2509.08015},
  year   = {2026}
}

Comments

10 pages, 16 figures

R2 v1 2026-07-01T05:28:55.740Z