△ Mechanism confirmed, baseline not beaten 2026

Histogram-Controlled Cluster Updates for Iterative GNNs

Implementation & benchmark of arXiv:2607.20130 — Learning to Decode Quantum LDPC Codes via Cluster-Based Sequential Belief Propagation

Usefulness7/10
Difficulty5/10
Novelty7/10

Source paper: Learning to Decode Quantum LDPC Codes via Cluster-Based Sequential Belief Propagation arXiv:2607.20130 ⓘ · analyzed Aug 30, 2026

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

Idea description

Replace node-by-node scheduling in an iterative message-passing network with a learned scheduler that selects one graph cluster at a time, while updating all nodes in that cluster synchronously. The scheduler observes a quantized histogram of local residual weights, making its state invariant to permutations of nodes inside a cluster and independent of cluster cardinality.

Formulas

$$\omega_i=\sum_{c_j\in\mathcal N(v_i)}\delta_j,\qquad \delta_j\in\{0,1\}$$
$$h_{a,r}=\frac{1}{B_a}\sum_{v_i\in\mathcal C_a}\mathbf 1\!\left\{q(\omega_i)=r\right\},\qquad q(\omega)=\left\lfloor \frac{R\omega}{A_{\max}}\right\rfloor$$
$$Q(s,a)\leftarrow Q(s,a)+\alpha\left[r+\gamma\max_{a'}Q(s',a')-Q(s,a)\right]$$
$$\mathbf x_{i,t+1}=F_\theta\!\left(\mathbf x_{i,t},\{\mathbf x_{u,t}:u\in\mathcal N(i)\}\right)\quad\forall v_i\in\mathcal C_{a_t}$$

Mathematical statement

Use a sparse bipartite graph G=(V,C,E), where V are neural state nodes and C are local constraint or aggregation nodes. Let delta_j in {0,1} indicate whether factor c_j currently has a residual, disagreement, or constraint violation. For node v_i, define the local mismatch weight omega_i as the sum of delta_j over neighboring factors. Partition V into clusters C_a. For a cluster of size B_a, define h_{a,r} as the fraction of its nodes whose quantized mismatch weight equals bin r. The quantizer q maps the possible degree range [0,A_max] into R bins. Thus the observation has fixed dimension R rather than dimension B_a. A scheduler selects action a_t, and every node in the selected cluster is updated from the same pre-update state. A Q-learning controller uses the temporal-difference recursion Q(s,a) <- Q(s,a) + alpha [r + gamma max_{a'} Q(s',a') - Q(s,a)].

Implementation notes

Integrate this into a recurrent GNN or neural algorithmic solver that currently performs T full message-passing sweeps. First partition graph nodes into fixed clusters using METIS, BFS blocks, or balanced random blocks, and store node indices for every cluster. At each micro-step, compute a local residual for every factor. For a constraint-solving GNN, use delta_j = 1 when the factor is violated. For a supervised denoising GNN, use delta_j = 1 when the factor residual norm exceeds a threshold tau. Compute omega_i by summing residual indicators over neighboring factors, clip it to [0,A_max], quantize it with q(omega) = floor(R omega / A_max), and form the normalized R-bin histogram for every cluster. Concatenate all cluster histograms, the fraction of active factors, and the previous action, then feed the state to a small scheduler MLP. Select a cluster by argmax Q(s,a), or use epsilon-greedy exploration while training the scheduler. For the selected cluster, gather all neighbor states from the pre-update tensor, run the GNN update for every selected node in one batched operation, and scatter the outputs simultaneously. This preserves within-cluster parallelism and prevents accidental sequential ordering. Use reward r_t = -sum_j delta_{j,t+1} - lambda, where lambda penalizes a scheduling decision. The mathematical quantities are the residual weights, quantizer, histogram, and temporal-difference update; tau, lambda, cluster construction, and exploration rate are empirical choices. The first experiment should use a 10-layer recurrent GNN on synthetic graph-coloring, SAT, or matching instances with 1,000 to 10,000 nodes. Compare full synchronous sweeps, random cluster order, greedy largest-residual selection, and learned scheduling at equal wall-clock budget. Record residual mass, task accuracy, GPU utilization, and the number of sequential decisions. Success means lower residual or higher accuracy with at least 2x fewer sequential scheduling decisions and no instability from stale states.

Verification

Mechanism confirmed, baseline not beaten

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Confirmed ✓

Stage 2 · Practical benchmark result: Significantly worse ✗

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

Stage 1 — Mechanism check agent confidence 8/10

Built a readable toy implementation of histogram observations, quantization, synchronous cluster scattering, and a residual-driven cluster selector. Histogram permutation/cardinality invariance and quantizer thresholds matched exactly; the stale sequential-vs-synchronous error followed the predicted eta^2 law with maximum error 1.1e-16. The mini-comparison ended at residual 74 for both full synchronous and histogram-selected updates, so the mechanism is real but this experiment did not establish a practical speedup or lower residual.

Agent confidence
8/10
Baseline
Full synchronous updates: final residual 74 after 30 sweeps (2,880 node updates).
Idea
Histogram-selected cluster updates: final residual 74 after 240 cluster decisions (2,880 node updates); 8x more sequential decisions than full sweeps, with no observed residual improvement.

Limitations: This is a small synthetic continuous-signal toy, not a trained recurrent GNN or Q-learning scheduler. It does not test METIS/BFS partitioning, GPU wall-clock utilization, learned policies, graph coloring/SAT accuracy, larger graphs, or statistical averages over multiple problem seeds.

How to run: python3 experiment.py

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

Significantly WORSE than the tuned baseline (+59.8%, p=0.0081, wins 0 / 8 paired seeds).

Benchmark
Dynamics forecasting
Model
rnn_small
Paired seeds
8
Baseline mean
0.0038
Idea mean
0.0061
Effect (Δ)
0.0023 (+59.8%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±36.7%
Mechanism
Confirmed ✓
Practical effect
Hurts
Baseline tuning
swept over 3 configs
Limitations:

The scheduler was a deterministic greedy largest-high-bin histogram policy rather than a learned Q-learning controller, so learned-policy training was not tested. Wall-clock speedup, GPU utilization, larger graphs, alternative cluster partitions, and more than three recurrent sweeps were not evaluated. The local intervention uses a compact ring message-passing model compatible with the dynamics structure rather than modifying the bench rnn_small GRU internals directly.

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 a recurrent message-passing dynamics system with synchronous full-sweep baseline and histogram-selected synchronous cluster updates, using the same data, optimizer, architecture components, and shared hyperparameter union. Across 8 paired seeds, the idea had higher test MSE (0.006128 vs 0.003836), with delta_mean=+0.002293 and permutation p=0.0081, constituting a significant loss. The trained-model mechanism signature confirmed the predicted approximately eta² scaling of synchronous-versus-sequential update discrepancy, but this did not translate into better task performance.

Baseline
Dynamics test MSE mean 0.00383558, std 0.00081823; best lr=0.01, epochs=12; per-seed [0.00405838, 0.00342526, 0.00307760, 0.00440629, 0.00449852, 0.00260400, 0.00526552, 0.00334905].
Idea
Histogram-controlled cluster updates test MSE mean 0.00612822, std 0.00194303, best lr=0.01, epochs=12; per-seed [0.00588618, 0.00363651, 0.00578116, 0.01028094, 0.00752147, 0.00445210, 0.00660217, 0.00486525]. Paired delta +0.00229264, 0/8 wins, permutation p=0.0081; mechanism signature confirmed=true with eta² ratio range [0.5408, 0.5880].

Benchmark result

Significantly WORSE than the tuned baseline (+59.8%, p=0.0081, wins 0 / 8 paired seeds).

Benchmark
Dynamics forecasting
Model
rnn_small
Paired seeds
8
Baseline mean
0.0038
Idea mean
0.0061
Effect (Δ)
0.0023 (+59.8%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±36.7%
Mechanism
Confirmed ✓
Practical effect
Hurts
Baseline tuning
swept over 3 configs
Limitations:

The scheduler was a deterministic greedy largest-high-bin histogram policy rather than a learned Q-learning controller, so learned-policy training was not tested. Wall-clock speedup, GPU utilization, larger graphs, alternative cluster partitions, and more than three recurrent sweeps were not evaluated. The local intervention uses a compact ring message-passing model compatible with the dynamics structure rather than modifying the bench rnn_small GRU internals directly.

How to run: python3 stage2_bench.py

Artifacts

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