Unverified 2026

MP Bulk Conditioning Regularizer

Implementation & benchmark of arXiv:2607.21759 — Marchenko-Pastur law for tensor powers of exchangeable unconditional vectors

Usefulness5/10
Difficulty6/10
Novelty6/10

Source paper: Marchenko-Pastur law for tensor powers of exchangeable unconditional vectors arXiv:2607.21759 ⓘ · analyzed Aug 30, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Add a spectral regularizer that prevents tensorized feature batches from developing covariance outliers or a collapsed lower edge. The target is the Marchenko–Pastur bulk predicted for the current feature-to-sample ratio, rather than an arbitrary identity-covariance penalty that may suppress useful anisotropy.

Formulas

$$C={1\over m}Z^TZ,\qquad \operatorname{ESD}(C)={1\over p}\sum_{j=1}^{p}\delta_{\lambda_j(C)},\qquad c={p\over m}$$
$$a_c=(1-\sqrt{c})^2,\qquad b_c=(1+\sqrt{c})^2$$
$$\mathcal{L}_{\mathrm{MP}}=\alpha\,[a_c+\delta-\widetilde\lambda_{\min}]_+^2+\beta\,[\widetilde\lambda_{\max}-b_c-\delta]_+^2$$
$$\widetilde\lambda_{\max}=\tau\log\sum_j e^{\lambda_j/\tau},\qquad \widetilde\lambda_{\min}=-\tau\log\sum_j e^{-\lambda_j/\tau}$$

Mathematical statement

For a feature matrix Z in R^{m times p}, define the sample covariance C=(1/m)Z^T Z and its empirical spectral distribution ESD(C)=(1/p) sum_j delta_{lambda_j(C)}. The paper's theorem says that tensor features generated from isotropic, exchangeable, unconditional vectors can have the same asymptotic MP ESD as independent-coordinate models under suitable moment assumptions. Let c=p/m, a_c=(1-sqrt(c))^2, and b_c=(1+sqrt(c))^2. Let lambda_min and lambda_max denote the extreme eigenvalues of C. Use soft estimates tilde-lambda_min and tilde-lambda_max, and penalize only violations of the expected bulk with L_MP=alpha[a_c+delta-tilde-lambda_min]_+^2+beta[tilde-lambda_max-b_c-delta]_+^2. Here delta is a tolerance margin, alpha and beta are nonnegative weights, and [r]_+=max(0,r). For c greater than or equal to one, set the lower-edge target to zero because the sample covariance is rank deficient in the overparameterized limit.

Implementation notes

Insert the regularizer after the tensorized embedding and before its following linear projection. (1) For each minibatch, construct Z with rows z_i=phi(x_i), optionally center columns if the baseline layer uses centered features, and define C=Z^T Z/B. For large p, never materialize C. (2) Estimate lambda_max with power iteration: initialize v randomly, repeat v<-Cv/||Cv|| for K=5 iterations, and set lambda_max_hat=v^T C v. Estimate lambda_min with a short Lanczos routine; for a simpler approximation, choose a safe shift mu greater than the current lambda_max and power-iterate mu I-C, then recover lambda_min_hat=mu-v^T(mu I-C)v. (3) Compute c=p/m_eff, where m_eff is the minibatch size or a FIFO feature queue size. If c<1, set the lower target to a_c; if c>=1, set it to zero and retain only the upper-outlier penalty. Add L_MP to the task loss with alpha and beta initially between 1e-3 and 1e-2, use delta around 0.05 times the current median eigenvalue, and stop gradients through a_c and b_c. Use gradient accumulation or a feature queue so m_eff is not extremely small. (4) Log the estimated eigenvalues, MP edges, task loss, regularizer magnitude, and gradient norm. The first cheap experiment is a width-256 degree-2 tensor-feature MLP on CIFAR-10 or synthetic regression, comparing no regularizer, standard whitening, and L_MP at equal wall-clock budget. Success means fewer covariance outliers, a larger minimum-eigenvalue margin, smoother gradients, and equal or better validation accuracy. Failure is increased compute or suppression of task-relevant anisotropy without improved optimization.

Verification

This idea has not been verified yet.

Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.

Artifacts

Artifacts unavailable.