{"id":"arxiv/2602.15022","url":"https://arxiv.org/abs/2602.15022","tags":["cs.LG","cs.AI","math.GR","q-bio.BM"],"score":0,"title":"Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation","domain":"cs.LG","source":"arxiv","authors":["Cai Zhou","Zijie Chen","Zian Li","Jike Wang","Kaiyi Jiang","Pan Li","Rose Yu","Muhan Zhang","Stephen Bates","Tommi Jaakkola"],"reviews":0,"subject":"cs.LG","summary":"Many generative tasks in chemistry and science involve distributions invariant to group symmetries (e.g., permutation and rotation). A common strategy enforces invariance and equivariance through architectural constraints such as equivariant denoisers and invariant priors. In this paper, we challenge this tradition through the alternative canonicalization perspective: first map each sample to an orbit representative with a canonical pose or order, train an unconstrained (non-equivariant) diffusion or flow model on the canonical slice, and finally recover the invariant distribution by…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-02-16","reactions":{}}