△ Mechanism confirmed, baseline not beaten 2026

Phenotype-Rao-Blackwellized ES

Implementation & benchmark of arXiv:2608.02073 — Accelerating Evolutionary Strategy via Rao-Blackwellizing Realization of Uncertain Input

Usefulness7/10
Difficulty5/10
Novelty8/10

Source paper: Accelerating Evolutionary Strategy via Rao-Blackwellizing Realization of Uncertain Input arXiv:2608.02073 ⓘ · analyzed Aug 31, 2026

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

Idea description

Modify an evolutionary-strategy gradient estimator so that the observed phenotype or trajectory is used to infer the conditional mean of the latent ES perturbation. Instead of multiplying fitness by the raw perturbation, multiply it by the posterior mean perturbation given the realized input; this remains unbiased and has variance no greater than the ordinary ES estimator when the conditional model is correct.

Formulas

$$F(\theta):=\mathbb{E}_{p_{\theta}(x)}\left[f(x)\right].$$
$$g_{\mathrm{ES}}=\frac{f(x)}{\sigma}z,\qquad z\sim\mathcal{N}(0,I_d).$$
$$g_{\mathrm{RB}}=\frac{f(x)}{\sigma}\,\mathbb{E}[z\mid x],\qquad \mathbb{E}[g_{\mathrm{RB}}]=\mathbb{E}[g_{\mathrm{ES}}].$$
$$\operatorname{Var}(g_{\mathrm{ES}})=\operatorname{Var}(g_{\mathrm{RB}})+\mathbb{E}\left[\operatorname{Var}(g_{\mathrm{ES}}\mid x)\right],\qquad \mathbb{E}[z\mid x]=\frac{\sigma}{\sigma^2+\tau^2}(x-\theta),\qquad g_{\mathrm{RB}}=\frac{f(x)(x-\theta)}{\sigma^2+\tau^2}.$$

Mathematical statement

The paper defines the objective as $F(\theta):=\mathbb{E}_{p_{\theta}(x)}[f(x)]$, where $\theta$ parameterizes an input distribution $p_{\theta}$, $x$ is the realized input, and $f(x)$ is its evaluation. For a Gaussian ES perturbation $z\sim\mathcal{N}(0,I_d)$ and perturbed parameter $\theta+\sigma z$, the ordinary score estimator is $g_{\mathrm{ES}}=f(x)z/\sigma$. If $x$ is generated conditionally from the perturbed parameter, the Rao-Blackwellized estimator is $g_{\mathrm{RB}}=f(x)\,\mathbb{E}[z\mid x]/\sigma$. The law of total expectation gives $\mathbb{E}[g_{\mathrm{RB}}]=\mathbb{E}[g_{\mathrm{ES}}]$, while the Rao-Blackwell variance identity gives $\operatorname{Var}(g_{\mathrm{ES}})=\operatorname{Var}(g_{\mathrm{RB}})+\mathbb{E}[\operatorname{Var}(g_{\mathrm{ES}}\mid x)]$, so the RB estimator cannot have larger variance. For the directly implementable Gaussian realization model $z\sim\mathcal{N}(0,I_d)$ and $x=\theta+\sigma z+\tau\xi$, with $\xi\sim\mathcal{N}(0,I_d)$ independent, the posterior mean is $\mathbb{E}[z\mid x]=\frac{\sigma}{\sigma^2+\tau^2}(x-\theta)$, yielding $g_{\mathrm{RB}}=f(x)(x-\theta)/(\sigma^2+\tau^2)$. Here $d$ is the parameter or input dimension, $\sigma$ is the ES perturbation scale, and $\tau$ is realization-noise scale.

Implementation notes

Integrate this into an ES optimizer for a stochastic policy, a noisy input layer, or a MoE router whose perturbed parameters produce an observable phenotype. Begin with the explicit Gaussian testbed, where the optimized vector is $\theta\in\mathbb{R}^d$, the latent ES perturbation is $z$, and the observed realization is $x=\theta+\sigma z+\tau\xi$. For each population member, sample $z_i\sim\mathcal{N}(0,I)$, construct $x_i=\theta+\sigma z_i+\tau\xi_i$, evaluate the network or task to obtain scalar fitness $r_i=f(x_i)$, and accumulate either the baseline or RB gradient. Pseudocode: g=0; for i in 1..N: z=randn(d); xi=randn(d); x=theta+sigma*z+tau*xi; r=f(x); if baseline: g += r*z/sigma; else: g += r*(x-theta)/(sigma*sigma+tau*tau); theta += eta*g/N. Use antithetic pairs and identical random seeds for both methods. The Gaussian posterior is analytic. In RL or MoE settings, replace it with a learned posterior mean $m_\psi(x)\approx\mathbb{E}[z\mid x]$, trained on sampled $(z,x)$ pairs by minimizing $\|m_\psi(x)-z\|^2$, then use $r_i m_\psi(x_i)/\sigma$ in the update. Estimate empirical variance from the population and monitor bias against a large-sample finite-difference gradient. First experiment: optimize a small MLP on noisy quadratic regression and CartPole with population sizes 32 and 128, comparing vanilla ES, antithetic ES, and RB-ES at equal environment evaluations. Log coordinate variance, gradient signal-to-noise ratio, and return versus evaluations. Success means at least 25% lower estimator variance and faster return improvement at equal evaluation budget; if posterior misspecification increases bias, use a validation criterion to interpolate between RB and baseline estimators.

Verification

Mechanism confirmed, baseline not beaten

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Confirmed ✓

Stage 2 · Practical benchmark result: No measurable effect

Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works

Stage 1 — Mechanism check agent confidence 9/10

Built Phenotype-Rao-Blackwellized ES with the analytic Gaussian posterior mean, variance-law verification, unbiasedness checks, and a paired antithetic noisy-quadratic optimizer comparison. The predicted variance ratio sigma²/(sigma²+tau²) matched observed ratios within 0.00032 across tau values; the residual variance identity also matched closely, and nonlinear-estimator mean differences stayed within 1.21 joint standard errors. RB improved distance at 20 steps from 0.2511 to 0.2249 and final distance from 0.000159 to 0.000106, so the mechanism manifested and showed a modest optimization benefit.

Agent confidence
9/10
Baseline
Antithetic ES: distance at step 20 = 0.25105, step 40 = 0.02085, final mean = 0.0001594 (SD 0.0000585).
Idea
Phenotype-RB ES: distance at step 20 = 0.22492, step 40 = 0.01719, final mean = 0.0001055 (SD 0.0000202); variance ratios observed/predicted across tau = 1.0000/1.0000, 0.8003/0.8000, 0.5002/0.5000, 0.2001/0.2000, and 0.05883/0.05882.

Limitations: Only the analytic Gaussian realization model and a small noisy quadratic were tested; no learned posterior, neural network, RL environment, MoE router, wall-clock benchmark, or population/FLOP-normalized large-scale comparison was evaluated.

How to run: python3 rb_es_experiment.py

Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.02073)

No significant difference vs. the tuned baseline (+0.7%, p=0.1424, wins 3 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_med
Paired seeds
8
Baseline mean
228.4837
Idea mean
230.1002
Effect (Δ)
1.6165 (+0.7%; negative = idea better)
Wins
3 / 8 paired seeds
p-value
0.1424 (permutation test, 20 000 shuffles)
Smallest detectable effect
±1.0%
Mechanism
Confirmed ✓
Practical effect
No effect
Baseline tuning
swept over 3 configs
Limitations:

Only the structurally matched tabular track was tested; no vision, sequence, or dynamics track was run. The ES loop used standardized paired fitness differences for numerical stability, so wall-clock speedups and learned posterior means were not evaluated.

How to run: python3 stage2_bench.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 Phenotype-Rao-Blackwellized ES as a local optimizer modification and ran the full paired 8-seed tabular benchmark with matched mlp_med systems, baseline sweep, and shared hyperparameter grid. The trained-model mechanism signature confirmed the predicted variance ratio, but RB-ES had worse test MSE than baseline and the difference was not significant.

Baseline
Tabular Friedman#1 regression with mlp_med; tuned ES cfg lr=0.01, sigma=0.1, tau=0.2; 8-seed test MSE mean 228.483749, std 11.990675.
Idea
Phenotype-RB ES with the same task/model/budget and cfg lr=0.01, sigma=0.1, tau=0.2; 8-seed test MSE mean 230.100216, std 11.617225; paired delta +1.616467 and permutation p=0.14245.
Mechanism signature
Observed trained-model fitness-weighted variance ratio 0.200282 versus predicted 0.2; posterior MSE 0.799645 versus predicted 0.8; confirmed=true.

Benchmark result

No significant difference vs. the tuned baseline (+0.7%, p=0.1424, wins 3 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_med
Paired seeds
8
Baseline mean
228.4837
Idea mean
230.1002
Effect (Δ)
1.6165 (+0.7%; negative = idea better)
Wins
3 / 8 paired seeds
p-value
0.1424 (permutation test, 20 000 shuffles)
Smallest detectable effect
±1.0%
Mechanism
Confirmed ✓
Practical effect
No effect
Baseline tuning
swept over 3 configs
Limitations:

Only the structurally matched tabular track was tested; no vision, sequence, or dynamics track was run. The ES loop used standardized paired fitness differences for numerical stability, so wall-clock speedups and learned posterior means were not evaluated.

How to run: python3 stage2_bench.py

Artifacts

Implementation overview ⬇ Download all as ZIP 7 files · code, reports and structured results