CorrFlow: Exact and Permutation-Equivariant Flow Matching on the Correlation Elliptope
stat.ML, cs.LG, q-fin.RM
Submitted: 2026-10-09
Updated: 2026-10-09
Comments: Accepted at NeurReps 2026, the NeurIPS workshop on Symmetry and Geometry in Neural Representations. Code: https://github.com/gautiermarti/CorrFlow
Code: https://github.com/gautiermarti/CorrFlow
License: http://creativecommons.org/licenses/by/4.0/
The gist: Correlation matrices combine nonlinear positive-definiteness constraints with a simultaneous row-column permutation symmetry.
Terminology
Abstract
Correlation matrices combine nonlinear positive-definiteness constraints with a simultaneous row-column permutation symmetry. Their generation therefore involves two distinct problems. First, every output must remain on the correlation elliptope, irrespective of how its variables are ordered. Second, the generator must learn the domain-specific distribution occupied by empirical data inside that valid space. Existing financial image generators mix these problems: they learn ordered matrix entries and repair invalid draws afterward. We introduce CorrFlow, which separates them. A generalized Fisher transform maps full-rank correlation matrices bijectively to unconstrained coordinates, guaranteeing valid outputs without projection. A graph flow-matching network learns the empirical distribution in those coordinates and is permutation equivariant: reordering the variables only reorders the output. We compare a conditional CorrGAN reimplementation, image diffusion, and CorrFlow on 80 times80 Taiwan equity matrices using five seeds per neural method. CorrFlow produces 100% valid draws without repair and obtains feature sliced-Wasserstein distance.712 plus or minus.141, versus 4.339 plus or minus.252 for CorrGAN and 4.610 plus or minus.191 for image diffusion after their invalid outputs are projected back to the correlation space. It also improves this distance over historical lookup, while lookup retains higher local coverage. These results support the combined construction under the evaluated training budget; they do not establish superiority over fully tuned image models or isolate the contribution of each component.
Sources
- The Generalized Fisher Transformation: Finite-Sample Properties and Inference
- Riemannian Flow Matching for Brain Connectivity Matrices via Pullback Geometry
Related papers
- Behavior of prediction performance metrics with rare events
- Optimal Estimation of Generic Dynamics by Path-Dependent Neural Jump ODEs
- A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors
- Efficient quadratic entropy with distance sketches
- Embedding-Bias in Conditional Independence Testing
- From Log-Odds to Shapley Values: An Explanatory Geometry for the Weighted Naive Bayes Classifier