Unverified 2026

Averaged Periodic Preconditioner

Implementation & benchmark of arXiv:2608.09319 — Averaging Principle and Pullback Attractor Convergence for McKean--Vlasov Stochastic Reaction--Diffusion Equations

Usefulness6/10
Difficulty4/10
Novelty6/10

Source paper: Averaging Principle and Pullback Attractor Convergence for McKean--Vlasov Stochastic Reaction--Diffusion Equations arXiv:2608.09319 ⓘ · analyzed Sep 1, 2026

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

Idea description

Use a rapidly cycling preconditioner or learning-rate vector during optimization, but construct a static averaged optimizer with the same mean update. When the parameter dynamics are locally contractive, the averaged optimizer should track the periodic optimizer while requiring less schedule bookkeeping and potentially fewer expensive state updates.

Formulas

$$\dot\theta_{\varepsilon}(t)=-P\left(\frac{t}{\varepsilon}\right)\nabla\ell\left(\theta_{\varepsilon}(t)\right),\qquad \bar P=\frac{1}{T}\int_0^T P(s)\,ds.$$
$$\dot{\bar\theta}(t)=-\bar P\nabla\ell\left(\bar\theta(t)\right).$$
$$P(s)\succeq p_{\min}I,\quad \nabla^2\ell(\theta)\succeq mI\quad\Longrightarrow\quad \|\theta(t)-\tilde\theta(t)\|_{\bar P^{-1}}\leq e^{-mp_{\min}(t-t_0)}\|\theta(t_0)-\tilde\theta(t_0)\|_{\bar P^{-1}}.$$
$$\lim_{\varepsilon\to0}\sup_{0\leq t\leq L}\mathbb{E}\left\|\theta_{\varepsilon}(t)-\bar\theta(t)\right\|^2=0.$$

Mathematical statement

Let $\theta\in\mathbb{R}^{p}$ be model parameters, $\ell(\theta)$ the loss, and $P(s)$ a positive-definite periodic preconditioner with period $T$. The fast gradient flow is $\dot\theta_{\varepsilon}=-P(t/\varepsilon)\nabla\ell(\theta_{\varepsilon})$. Its averaged flow is $\dot{\bar\theta}=-\bar P\nabla\ell(\bar\theta)$, where $\bar P=T^{-1}\int_0^T P(s)\,ds$. If $P(s)\succeq p_{\min}I$ and the loss Hessian satisfies $\nabla^2\ell(\theta)\succeq mI$ in the visited region, then the dynamics are contractive with rate at least $mp_{\min}$ in the appropriate preconditioned metric. The averaging principle predicts finite-window convergence as the cycle period tends to zero, and contraction prevents accumulated trajectory errors from growing over long training intervals.

Implementation notes

Implement this as an optimizer wrapper around a differentiable model. Choose $M$ positive diagonal preconditioners $P_0,\ldots,P_{M-1}$, or periodic learning-rate vectors, and apply $\theta\leftarrow\theta-\eta P_k g$ at phase $k$. Construct the averaged baseline $\bar P=M^{-1}\sum_{k=0}^{M-1}P_k$ and apply $\theta\leftarrow\theta-\eta\bar P g$ at each step. Keep the average update, batch order, and total gradient evaluations identical. The paper contributes the fast-coefficient averaging and contraction mechanism; strong convexity is only a local approximation in deep networks. Estimate local curvature with Hessian-vector products on a parameter block, estimate $p_{\min}$ from sampled Rayleigh quotients or the smallest diagonal preconditioner entry, and measure contraction by running two nearby parameter replicas with identical minibatches. The first experiment should use a two-layer MLP on MNIST or CIFAR-10, comparing SGD, AdamW, the periodic preconditioner, and its averaged version. Test cycle lengths $M\in\{1,2,4,8,16,32\}$ and synchronize trajectories after each cycle. The prediction is that, in a locally contractive phase, RMS parameter distance between periodic and averaged methods decreases as the physical oscillation period decreases, and replica distances have fitted decay slope near $-mp_{\min}$. When the measured local contraction ratio exceeds one, the approximation should cease to be reliable and trajectory error should begin growing.

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.