Factorized Gradients for Scalable Highly-expressive Parametric Diffeomorphisms
NeurIPS 2026 Conference on Neural Information Processing Systems
FG-CPAB separates trajectory integration from parameter-space projection, reducing the gradient cost of CPAB transformations from $\mathcal{O}(dTN)$ to $\mathcal{O}(TN + dC)$. This yields up to $10^4\times$ faster gradients and $500\times$ lower memory, making fine-tessellation diffeomorphisms practical in 2D and 3D.
