Scale-Quotient Covariance Alignment
Implementation & benchmark of arXiv:2608.27209 — Connecting Riemannian Geometry and Statistical Inference for Correlation Matrices
Source paper: Connecting Riemannian Geometry and Statistical Inference for Correlation Matrices arXiv:2608.27209 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace Euclidean covariance matching with a discrepancy that identifies covariance matrices differing only by per-channel positive rescaling. Apply it to minibatch feature covariances in a representation-alignment, domain-adaptation, style-transfer, or multi-view objective so that the network is penalized for changing correlation structure but not arbitrary channel units.
Formulas
Mathematical statement
The paper models the manifold of full-rank correlation matrices as the quotient \(\operatorname{Cor}_{+}(p)=\operatorname{SPD}(p)/\operatorname{Diag}^{+}(p)\), where \(\operatorname{SPD}(p)\) is the set of \(p\times p\) symmetric positive-definite covariance matrices and \(\operatorname{Diag}^{+}(p)\) is the group of positive diagonal matrices. The action is \(\Sigma\mapsto D\Sigma D\), with \(D\) a positive diagonal matrix, so all covariance matrices differing by channel-wise scale belong to one equivalence class. The affine-invariant metric on \(\operatorname{SPD}(p)\) is \(G_{\Sigma}(A,B)=\operatorname{tr}(\Sigma^{-1}A\Sigma^{-1}B)\), where \(A,B\) are symmetric tangent perturbations. The induced quotient distance can be implemented as \(d_Q([\Sigma_1],[\Sigma_2])=\min_{D\in\operatorname{Diag}^{+}(p)} d_{AI}(\Sigma_1,D\Sigma_2D)\), using \(d_{AI}(X,Y)=\|\log(X^{-1/2}YX^{-1/2})\|_F\), where \(\log\) is the matrix logarithm and \(\|\cdot\|_F\) is the Frobenius norm. The minimization removes nuisance marginal variances while retaining correlation differences.
Implementation notes
(1) Integrate this into a covariance-alignment loss. Choose a layer producing centered activations \(H_x\in\mathbb{R}^{m\times p}\), with batch size \(m\), channel dimension \(p\), and two groups or views \(a,b\). Form \(\widehat\Sigma_x=(H_x^{\mathsf T}H_x)/(m-1)+\varepsilon I\), where \(\varepsilon>0\) ensures positive definiteness. Add \(\lambda\mathcal{L}_{\mathrm{quot}}\) to the task loss. (2) Compute the loss with a small number of Newton or gradient iterations over unconstrained variables \(u\in\mathbb{R}^p\), setting \(D=\operatorname{diag}(\exp u)\). Initialize \(u=0\); at each inner iteration compute \(M=\widehat\Sigma_a^{-1/2}D\widehat\Sigma_bD\widehat\Sigma_a^{-1/2}\), eigendecompose \(M=V\operatorname{diag}(s)V^{\mathsf T}\), evaluate \(\|\log s\|_2^2\), backpropagate through the eigendecomposition, update \(u\leftarrow u-\eta_u\nabla_u\|\log s\|_2^2\), and detach or stop after 3--5 iterations. Use the resulting value as the outer loss. (3) The quotient action and affine-invariant metric come from the paper; the minimizing diagonal \(D\), covariance estimates, and matrix eigendecomposition are computed numerically. For a cheaper approximation, replace the inner minimization by correlation normalization \(R_{ij}=\Sigma_{ij}/\sqrt{\Sigma_{ii}\Sigma_{jj}}\), then use the affine-invariant distance between \(R_a+\varepsilon I\) and \(R_b+\varepsilon I\). (4) First test on CIFAR-10 or Office-31 with a small ResNet-18, using supervised classification plus covariance alignment between two augmented views or source and target domains. Compare against no alignment, Frobenius covariance loss, and standard correlation-matrix loss. Measure target accuracy, robustness under random per-channel activation rescaling, training variance across seeds, and wall-clock overhead. Success is higher accuracy or domain-transfer performance under scale perturbations at comparable task loss, without the covariance term destabilizing optimization.
Verification
Stage 1 · Toy mechanism gate: Failed ✗
Stage 2 · Mechanism transferred to benchmark: Not tested
Stage 2 · Practical benchmark result: Not run
Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Built an exact diagonal-quotient affine-invariant covariance distance with unrolled log-scale optimization, plus a cheaper correlation-normalized proxy used in the training benchmark. The numerical check passed strongly: same-class quotient distance was about 1e-29 despite raw AI distance 22.45, while a correlation perturbation gave 0.0022. On the two-seed synthetic domain-shift task, the idea matched baseline target accuracy (0.800 vs 0.800) but reduced source accuracy and rescaled-target accuracy (0.749 vs 0.760); it greatly reduced correlation MSE (0.00121 vs 0.0253), but this did not translate into an accuracy win.
- Agent confidence
- 9/10
- Baseline
- No alignment: target accuracy 0.800, random-channel-rescaled target accuracy 0.760, correlation MSE 0.0253, mean runtime 4.81 s
- Idea
- Correlation-normalized AI proxy: target accuracy 0.800, random-channel-rescaled target accuracy 0.749, correlation MSE 0.00121, mean runtime 3.93 s; exact quotient math check same-class distance 1.28e-29 and changed-correlation distance 0.00221
Limitations: Only a tiny synthetic two-domain problem and two random seeds were tested. The training benchmark uses the permitted correlation-normalized approximation rather than the expensive exact inner quotient optimization; no CIFAR/Office-31 data, larger networks, hyperparameter tuning, FLOP-matched study, or multiple perturbation strengths were evaluated.
How to run: python3 experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- experiment.py 4.9 KB View Raw
- report.md 1.6 KB View
- results.json 2.4 KB View Raw