Regularity-Matched Random Fourier Layer
Implementation & benchmark of arXiv:2609.03401 — Spectral Convergence of Random Feature Method in Multiple Dimensions
Source paper: Spectral Convergence of Random Feature Method in Multiple Dimensions arXiv:2609.03401 ⓘ · analyzed Sep 4, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace the usual isotropic Gaussian random Fourier features with a frequency distribution matched to the expected spectral regularity of the target function. For coordinate fields, operator-learning maps, or PDE solution surrogates, this should place more features where the target Fourier energy lies and improve approximation at the same feature count. Stabilize the resulting feature matrix with whitening or ridge regression because spectral accuracy can create severe ill-conditioning.
Formulas
Mathematical statement
Let N be the number of sampled features, delta in (0,1) the failure probability, d the input dimension, and L_N := N / log(N/delta). The paper's rate table states that Sobolev targets of smoothness s, measured in an error norm indexed by t, use d tau_Sob(w) proportional to (1+|w|^2)^(-bar{s}) d w and achieve rate L_N^(-(s-t)/d). Gevrey targets with parameter s >= 1 use d tau_Gevrey(w) proportional to exp(-2 bar{kappa}|w|^(1/s)) d w and achieve exp(-c L_N^(1/(s d))). Ultra-analytic targets with s > 1 use d tau_ultra(w) proportional to exp(-2 bar{kappa}|w|^s) d w and achieve exp(-c L_N^(1/d) log L_N). Here bar{s}, bar{kappa}, and c are positive constants determined by the distribution and target class. Implement phi_w(x) = sqrt(2) cos(w^T x + b), with b uniform on [0,2 pi], and train a head on the resulting feature matrix. The extracted text states that the same spectral mechanism causing high accuracy also causes severe condition-number growth, so empirical whitening is part of the adaptation.
Implementation notes
1. Integration point: use this as the frozen input projection in a coordinate MLP, neural operator, PINN, or implicit neural representation. Given inputs X in R^(B by d), sample N frequencies once at initialization, compute Z_bn = sqrt(2) cos(X_b dot w_n + b_n), and feed Z to a trainable linear or MLP head. 2. Pseudocode: choose a target regularity mode. For Sobolev, sample each w_n from the normalized density proportional to (1+||w||^2)^(-bar{s}); for Gevrey, sample from the density proportional to exp(-2 bar{kappa} ||w||^(1/s)); for ultra-analytic, sample from exp(-2 bar{kappa} ||w||^s). Draw b_n uniformly from [0, 2 pi], form Z, estimate the empirical Gram matrix G = Z^T Z / B + epsilon I, and replace Z by Z G^(-1/2) using an eigendecomposition or QR factorization. Train the downstream head first and optionally unfreeze frequencies later. 3. Compute the distributions and L_N exactly from the paper; estimate the Gram spectrum empirically and choose epsilon = 10^(-5) trace(G)/N. 4. First experiment: fit 2D Sobolev, Gevrey, and bandlimited synthetic functions on a 64 by 64 grid using a linear head with 256, 512, and 1024 features. Compare Gaussian RFF, uniform-window RFF, and matched distributions at equal feature count and FLOPs, with and without whitening. Measure test L2 error, convergence slope versus N, condition number, and wall-clock stability. Success means lower error at fixed N, a steeper log-error curve for smooth targets, and no instability after whitening.
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.