CorrFlow: Exact and Permutation-Equivariant Flow Matching on the Correlation Elliptope

arXiv:2610.13606 · stat.ML, cs.LG, q-fin.RM · Submitted 2026-10-09 · Read on arXiv

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

Related papers