Double-Geometric Layerwise ES
Implementation & benchmark of arXiv:2608.23714 — Integer Natural Evolution Strategies
Source paper: Integer Natural Evolution Strategies arXiv:2608.23714 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace Gaussian perturbations in a low-dimensional neural-network optimizer with independent double-geometric integer mutations and adapt each mutation scale using its exponential-family natural gradient. Apply the method to layerwise quantization scales, adapter coefficients, pruning thresholds, or other integer/discrete hyperparameters rather than to every individual weight.
Formulas
Mathematical statement
The paper starts from the ES mutation rule \(\mathbf{x}^{(t+1)}=\mathbf{x}^{(t)}+\mathbf{z}\), where \(\mathbf{x}^{(t)}\) is the current search point and \(\mathbf{z}\) is a random mutation. For one integer coordinate, use the double-geometric distribution \(p_q(z)=\frac{1-q}{1+q}q^{|z|}\) for \(z\in\mathbb{Z}\) and \(0<q<1\). Writing \(\eta=\log q\), its sufficient statistic is \(T(z)=|z|\), and its score is \(\partial_\eta\log p_\eta(z)=|z|-\mu(q)\), where \(\mu(q)=\mathbb{E}_q|z|=\frac{2q}{1-q^2}\). For a minimization objective \(F\), the natural-gradient direction for the log-scale parameter is \(g_\eta=\operatorname{Cov}(F(\mathbf{x}+\mathbf{z}),|z|)/\operatorname{Var}(|z|)\); the covariance is estimated from sampled candidates and the denominator is the Fisher information \(\operatorname{Var}(|z|)\). A fading-memory evolution path estimates this signal online: \(h_t=\beta h_{t-1}+(1-\beta)\widehat g_t\), followed by \(\eta_{t+1}=\eta_t-\rho h_t\) for minimization, with \(q_{t+1}=\exp(\eta_{t+1})\) clipped to \([q_{\min},q_{\max}]\). For multiple coordinates or layerwise search variables, use one independent \(q_j\), one centered statistic \(|z_j|-\mu(q_j)\), and one path \(h_j\) per coordinate.
Implementation notes
Integrate this into a zero-order optimizer for a neural network whose trainable search variables are low-dimensional integer quantities: for example, one integer quantization offset or bit-allocation variable per layer, one integer rank per adapter block, or one integer perturbation coefficient per channel group. Do not mutate millions of weights initially; define a vector \(x\in\mathbb{Z}^d\) with \(d\) equal to the number of layerwise or groupwise variables, and decode \(x\) into the network before evaluating a minibatch loss or validation score. At iteration t, maintain \(q_j\) and evolution path \(h_j\) for each coordinate. Sample K mutations independently using inverse-CDF or sequential sampling from the double-geometric law: sample sign uniformly when the magnitude is nonzero, and sample magnitude m from the geometric distribution with parameter \(1-q_j\). Evaluate \(F(x+z_k)\), rank the K candidates, and assign centered utilities \(u_k\) with zero mean, such as normalized negative ranks for minimization. Compute the numerator in the displayed natural-gradient estimator using the sampled \(|z_{k,j}|\), divide by the empirical variance plus \(\varepsilon\), update the fading path with \(\beta\approx0.8\!\text{--\!}0.95\), and update \(\eta_j\) with step size \(\rho\). Clip q to prevent either no mutations or extremely heavy tails. The mathematical quantities are the double-geometric probability, analytic mean \(\mu(q)\), and Fisher normalization; utilities, empirical covariance, and q bounds are engineering choices. First test on a 2-layer MLP or small Transformer with post-training integer quantization: optimize per-layer integer zero-points and scale indices using 16--32 candidates per iteration, comparing against Gaussian ES and fixed-radius integer random search at equal validation-loss evaluations. Success is faster validation-loss reduction, fewer collapsed zero-mutation coordinates, and improved accuracy at the same quantization budget.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗
Stage 2 · Practical benchmark result: Inconclusive
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 7/10
Built Double-Geometric Layerwise ES with independent double-geometric integer mutations, analytic Fisher normalization, fading-memory natural-gradient scale adaptation, clipping, and rank utilities. The score/covariance identity matched a finite-difference derivative with absolute error 3.2e-9. On a six-coordinate layerwise-style integer quadratic using 8 fixed seeds and 2160 evaluations per run, the method reached the exact optimum in 100% of runs by iteration 20, versus 25% for Gaussian ES and 87.5% for fixed-radius integer search at the full budget; this is a promising but toy-scale signal. The learned q values reached the lower clip (0.08), so scale adaptation was effective but somewhat collapsed toward sparse mutations.
- Agent confidence
- 7/10
- Baseline
- Gaussian ES: final objective 0.700 +/- 0.415, exact fraction 25%; fixed-radius integer search: 0.100 +/- 0.265, exact fraction 87.5%.
- Idea
- Double-geometric ES: final objective 0.000 +/- 0.000, exact fraction 100%; mean objective was 0.0 by iteration 20 across 8 seeds. Score identity absolute error: 3.2e-9.
Limitations: The mini-experiment is a low-dimensional weighted integer quadratic rather than an actual neural-network quantization or adapter task. Candidate selection is greedy-best rather than a full ES recombination/update, only 8 seeds were used, and the adaptive q values hit the lower bound; no wall-clock, GPU, quantization accuracy, or larger-dimensional scaling study was performed.
How to run: python3 run_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.23714)
No significant difference vs. the tuned baseline (+9.8%, p=0.129, wins 3 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 9.2576
- Idea mean
- 10.1677
- Effect (Δ)
- 0.91 (+9.8%; negative = idea better)
- Wins
- 3 / 8 paired seeds
- p-value
- 0.129 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±12.3%
- Mechanism
- Not confirmed ✗
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 6 configs
The mechanism signature did not confirm the stage-1 quantitative prediction because per-candidate covariance/update diagnostics were not persisted in the report. The idea sweep contains a duplicated lr=0.006 configuration rather than three distinct learning rates, although all tested idea learning rates were present in the baseline sweep. Only the structurally matched tabular track and mlp_tiny model were tested; no quantization-specific deployment or larger layerwise search was evaluated.
How to run: cd /home/maxwelhelp/all/math2nn/experiments/exp107_double-geometric-layerwise-es && /home/maxwelhelp/main/bin/python3 bench_run.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
Implemented Double-Geometric Layerwise ES as a layerwise integer-controlled optimizer for the matched tabular MLP benchmark and ran the full 8-seed paired protocol. The baseline Adam sweep achieved test MSE 9.258 ± 0.795, while the idea achieved 10.168 ± 1.859; paired delta was +0.910 with permutation p=0.129, so there was no significant win and the system did not work on this benchmark.
- Baseline
- Adam, best lr=0.006 and weight_decay=0.0001: test MSE 9.257637917995453 ± 0.7946203059794043
- Idea
- Double-geometric layerwise optimizer, selected lr=0.006, rho=0.12, beta=0.85, K=6: test MSE 10.167674660682678 ± 1.8586175283157877
Benchmark result
No significant difference vs. the tuned baseline (+9.8%, p=0.129, wins 3 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 9.2576
- Idea mean
- 10.1677
- Effect (Δ)
- 0.91 (+9.8%; negative = idea better)
- Wins
- 3 / 8 paired seeds
- p-value
- 0.129 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±12.3%
- Mechanism
- Not confirmed ✗
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 6 configs
The mechanism signature did not confirm the stage-1 quantitative prediction because per-candidate covariance/update diagnostics were not persisted in the report. The idea sweep contains a duplicated lr=0.006 configuration rather than three distinct learning rates, although all tested idea learning rates were present in the baseline sweep. Only the structurally matched tabular track and mlp_tiny model were tested; no quantization-specific deployment or larger layerwise search was evaluated.
How to run: cd /home/maxwelhelp/all/math2nn/experiments/exp107_double-geometric-layerwise-es && /home/maxwelhelp/main/bin/python3 bench_run.py
Artifacts
- bench_report.json 3.2 KB View Raw
- bench_run.py 5.9 KB View Raw
- report.md 1.7 KB View
- report_bench_2026-09-03T014209.md 5.2 KB View
- results.json 42.6 KB View Raw
- run_experiment.py 4.4 KB View Raw