Følner-Gated Message Passing
Implementation & benchmark of arXiv:2608.09796 — Fiberwise amenability of étale groupoids
Source paper: Fiberwise amenability of étale groupoids arXiv:2608.09796 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Add an online receptive-field expansion monitor to a graph neural network and use it to gate message-passing depth or invoke graph pooling. For a sampled node set F and propagation neighborhood K, continue fine-scale propagation only while the growth ratio |KF|/|F| is close to one; when it is persistently expansive, replace further propagation with pooling, local attention, or long-range skip messages. This transfers the paper's Følner-versus-paradoxical mechanism into an architecture-level control rule for oversquashing and explosive activation or memory growth.
Formulas
Mathematical statement
For a finite node set F and a finite set K of allowed one-step graph moves, define the K-neighborhood KF={k·x:k∈K,x∈F}, or equivalently KF=N_K(F) in an ordinary graph. The Følner condition is |KF|≤(1+ε)|F|, where ε>0 is a tolerance. Fiberwise amenability asserts that, after enlarging the allowed context to a set L, every finite seed M has an intermediate F satisfying M⊆F⊆LM and the Følner inequality. The contrasting non-amenable condition gives an L such that LM contains n pairwise disjoint translated copies Kx, implying strong expansion and replicated receptive-field demand. In a finite graph, estimate r_K(F)=|N_K(F)|/|F| and its multistep version r_K^(q)(F)=|N_{K^q}(F)|/|F|. The neural controller uses r_K(F)≤1+ε as the low-growth regime and r_K(F)≥1+δ, or repeated growth across q layers, as the expansive regime. This is a finite-graph proxy for the paper's dichotomy rather than a claim that every finite graph has an exact amenability classification.
Implementation notes
1. Integration point: use a residual message-passing GNN with L layers, hidden state h_v^ell, and a lightweight controller before each layer. For a minibatch of seed nodes, sample F from the current receptive-field nodes or from a fixed-size frontier. Let K be the one-hop move set, optionally including self-loops and two-hop edges. Compute an approximate neighborhood F_next=N_K(F) using sparse adjacency expansion, and estimate r=|F_next|/max(1,|F|). Maintain an exponential moving average r_bar and a consecutive-expansion counter. 2. Pseudocode: initialize F from the seed batch and r_bar=1; at layer ell compute F_next=N_K(F), r=|F_next|/max(1,|F|), and r_bar=beta*r_bar+(1-beta)*r. If r_bar≤1+epsilon, execute h^{ell+1}=h^ell+alpha_ell MSG_K(h^ell). Otherwise execute local message passing and a pooled skip: h^{ell+1}=h^ell+g MSG_K(h^ell)+(1-g)U(POOL(F_next)), where g=sigmoid(a((1+delta)-r_bar)). Optionally coarsen F by selecting cluster representatives whenever r_bar>1+delta for q consecutive layers. 3. Computed quantities are sparse frontier cardinalities, sampled expansion ratios, the gate, and pooled features. Estimate expansion on 4–8 random seed sets and report confidence intervals. The paper contributes the operational Følner/paradoxical criterion; the finite-graph estimator, gate, and pooling are the ML adaptation. 4. First cheap experiment: compare a 6-layer GCN or GraphSAGE baseline, fixed pooling, and Følner-gated GraphSAGE on cycle graphs, 2D grids, 3-regular random graphs, and finite 3-regular trees, using node classification with synthetic labels depending on radius-3 neighborhoods. Use identical parameter counts and train for 100 epochs. Measure validation accuracy, peak frontier size, activation memory, and oversmoothing through average pairwise cosine similarity. The quantitative prediction is that cycles and grids will show r_K(F) approaching 1 for sufficiently large sampled F, while trees and expanders will maintain r_K(F)>1+delta; the gate should activate predominantly on the latter. The measured activation boundary should occur near r_bar=1+delta, and gated propagation should reduce peak receptive-field memory by at least 25% on expansive graphs without reducing radius-3 accuracy by more than 2 percentage points.
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 6/10
Built a reproducible Følner-gated six-layer message-passing prototype with frontier expansion monitoring, EMA gating, pooled skips, exact graph sanity checks, and a four-graph classification comparison. The core mechanism manifested: cycle ratios matched 1+2/m exactly, tree-ball ratios matched the exact expansive-growth calculation, and the sigmoid gate's 0.5 boundary matched r=1+delta=1.35 for all tested slopes. The mini-experiment did not show a general accuracy or speed win: gating activated on all finite graphs because the sampled frontier expanded, and accuracy was usually lower than baseline; therefore this is a mechanism-validation success, not evidence of a practical model improvement.
- Agent confidence
- 6/10
- Baseline
- Validation accuracy: cycle 0.967, grid 0.600, random3 0.333, tree3 0.379; mean training time 0.50 s; gate=1.0 by definition.
- Idea
- Validation accuracy: cycle 0.533, grid 0.533, random3 0.467, tree3 0.483; mean training time 0.97 s; gate mean approximately 0 on every graph. Exact checks: cycle maximum absolute ratio error 0; tree maximum error 0; gate boundary observed 1.35 versus predicted 1.35 for slopes 2,4,8,16.
Limitations: The implementation uses dense adjacency matrices and a global pooled skip rather than sparse minibatch graph sampling or learned local pooling. It does not measure true activation memory, FLOPs, peak GPU memory, confidence intervals, fixed pooling, or radius-3 labels as specified. The auxiliary EMA persistence approximation disagreed with the discrete recurrence for r=1.6 and r=2.0, so it was not used to claim a win.
How to run: python3 run_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.09796)
Ahead of the baseline (−8.5%) but not statistically significant (p=0.5225, wins 6 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0006
- Idea mean
- 0.0005
- Effect (Δ)
- -0 (−8.5%; negative = idea better)
- Wins
- 6 / 8 paired seeds
- p-value
- 0.5225 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±22.9%
- Mechanism
- Not confirmed ✗
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 6 configs
The registered dynamics track is a controlled pendulum rollout, so the graph neighborhood monitor is adapted as a temporal receptive-field proxy. No literal graph dataset, sparse graph frontier memory/FLOP measurement, learned pooling hierarchy, or graph radius-3 node-classification task was tested.
How to run: python3 stage2_folner_bench.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 10/10
Implemented Følner-gated recurrent message passing as a local modification of the registered dynamics/rnn_small benchmark system. Across 8 paired seeds, the idea had slightly lower MSE, but the paired permutation test was not significant; the required system verdict is therefore no significant win. The trained-model mechanism signature also failed to confirm the predicted expansive regime.
- Baseline
- Registered track dynamics, model rnn_small; tuned baseline sweep over lr={0.001,0.003,0.01} and weight_decay={0,0.0001}; best lr=0.01, weight_decay=0.0001. Full 8-seed mean test MSE=0.0005577069, std=0.0001623548.
- Idea
- Chosen gated configuration lr=0.01, weight_decay=0.0001, delta=0.5, beta=0.7, slope=8; full 8-seed mean test MSE=0.0005104744, std=0.0001625943. Paired delta=-0.0000472325, 6/8 idea wins, permutation p=0.5225.
- Mechanism signature
- Predicted observed expansion ratio >1.35 and mean gate <0.5; trained-model measurements were mean ratio=1.2843889354 and mean gate=0.8395185024, confirmed=false.
Benchmark result
Ahead of the baseline (−8.5%) but not statistically significant (p=0.5225, wins 6 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0006
- Idea mean
- 0.0005
- Effect (Δ)
- -0 (−8.5%; negative = idea better)
- Wins
- 6 / 8 paired seeds
- p-value
- 0.5225 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±22.9%
- Mechanism
- Not confirmed ✗
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 6 configs
The registered dynamics track is a controlled pendulum rollout, so the graph neighborhood monitor is adapted as a temporal receptive-field proxy. No literal graph dataset, sparse graph frontier memory/FLOP measurement, learned pooling hierarchy, or graph radius-3 node-classification task was tested.
How to run: python3 stage2_folner_bench.py
Artifacts
- bench_report.json 4.7 KB View Raw
- report.md 1.8 KB View
- report_bench_2026-09-01T113846.md 7.0 KB View
- results.json 5.1 KB View Raw
- run_experiment.py 6.7 KB View Raw
- stage2_folner_bench.py 4.5 KB View Raw