✗ Failed on benchmark 2026

Distinct-kink complexity regularizer and merger

Implementation & benchmark of arXiv:2607.07778 — A law of robustness for two-layer neural networks with arbitrary weights

Usefulness6/10
Difficulty5/10
Novelty6/10

Source paper: A law of robustness for two-layer neural networks with arbitrary weights arXiv:2607.07778 ⓘ · analyzed Aug 30, 2026

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

Idea description

Replace nominal hidden-unit count by the number of distinct realized ReLU kink hyperplanes, then regularize or prune this effective count. Neurons whose normalized affine boundaries coincide can be exactly merged by summing their canonical coefficients, reducing memory without changing the represented function and aligning the model's complexity measure with the theorem.

Formulas

$$a\,\operatorname{ReLU}(\langle w,x\rangle+b)=\alpha\,\operatorname{ReLU}(\langle u,x\rangle+\beta),\qquad r=\|w\|_2,\quad u=\frac{w}{r},\quad \beta=\frac{b}{r},\quad \alpha=ar.$$
$$f(x)=\sum_{j=1}^{k(f)}\alpha_j\operatorname{ReLU}(\langle u_j,x\rangle+\beta_j)+\langle v,x\rangle+c,\qquad \|u_j\|_2=1,$$
$$\operatorname{Lip}(f)\geq c\,\varepsilon\sqrt{\frac{n}{(k(f)+1)\log(C(k(f)+1)nd/\varepsilon)}}.$$

Mathematical statement

A ReLU neuron $a\,\operatorname{ReLU}(\langle w,x\rangle+b)$ has a kink on the hyperplane $\langle w,x\rangle+b=0$. For $w\neq0$, define $r=\|w\|_2$, $u=w/r$, $\beta=b/r$, and the canonical coefficient $\alpha=ar$; then $a\operatorname{ReLU}(\langle w,x\rangle+b)=\alpha\operatorname{ReLU}(\langle u,x\rangle+\beta)$. If several neurons have the same canonical hyperplane $(u,\beta)$, their coefficients add exactly. Let $k(f)$ be the number of distinct realized kink hyperplanes after merging. The paper's realized-kink theorem replaces $\bar m$ by $k(f)+1$: $\operatorname{Lip}(f)\geq c\varepsilon\sqrt{n/[(k(f)+1)\log(C(k(f)+1)nd/\varepsilon)]}$ for $k(f)\leq n$. Thus redundant parameterization cannot evade the robustness-capacity tradeoff.

Implementation notes

Apply this at optimizer checkpoints or before inference for a two-layer ReLU network. For every nonzero hidden weight, compute $u=w/\|w\|_2$, $\beta=b/\|w\|_2$, and $\alpha=a\|w\|_2$. Canonicalization has a sign subtlety: retain a fixed orientation convention for $(u,\beta)$, and only merge entries whose full normalized pair agrees within tolerance; do not merge $(u,\beta)$ with $(-u,-\beta)$ unless the corresponding affine correction is explicitly derived. Hash rounded $(u,\beta)$ vectors or use locality-sensitive nearest-neighbor search with tolerance $\tau$. For each cluster, replace all members by one neuron with coefficient $\alpha_{\mathrm{cluster}}=\sum_i\alpha_i$, preserving the shared $(u,\beta)$; discard coefficients below a threshold only in approximate compression mode. During training, add $\lambda_k\tilde{k}$ to the loss, where $\tilde{k}=\sum_{p\in\text{clusters}}\operatorname{softplus}((\|\alpha_p\|-\alpha_0)/s)$ is a differentiable proxy, and periodically perform exact discrete merging. Re-estimate $k(f)$ after every merge and log the theorem-inspired lower bound using $k(f)+1$, not raw width. The exact algebraic rescaling and coefficient summation come from the canonical kink representation; clustering tolerance, penalty weights, and pruning thresholds are empirical. First test on noisy 2D regression and CIFAR-10 with a width-1024 shallow ReLU model, comparing raw-$L_0$ pruning and magnitude pruning. Measure exact function discrepancy on a dense probe set, parameter reduction, validation accuracy, Jacobian norm, and robustness to input perturbations. Success means at least 2x fewer hidden units at near-identical predictions, with lower or unchanged Jacobian norm and no worse validation error.

Verification

Failed on benchmark

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗

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 8/10

Implemented canonical ReLU kink normalization and exact same-hyperplane coefficient merging. The algebra check and dense-probe comparison showed only float32-level discrepancies (maximum about 1.7e-6 and 1.4e-6 respectively), while merging reduced width from 48 to 24 and parameters from 147 to 75 without changing validation MSE or mean Jacobian norm. This verifies the claimed exact merger phenomenon, but does not demonstrate that a differentiable regularizer discovers duplicates spontaneously during training.

Agent confidence
8/10
Baseline
Redundant width-48 coefficient-split model: validation MSE 0.0195736885, mean Jacobian norm 1.35254955, 147 parameters.
Idea
Merged width-24 model: validation MSE 0.0195736848, mean Jacobian norm 1.35254955, 75 parameters; 2.0x width reduction and probe RMSE function error 2.11e-7.

Limitations: The experiment uses deliberately constructed duplicate neurons, so it isolates exact merger correctness rather than testing duplicate formation under regularization. It does not test CIFAR-10, magnitude/L0 pruning baselines, approximate clustering tolerance, inference latency, or robustness under adversarial perturbations.

How to run: python3 kink_experiment.py

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

Ahead of the baseline (−0.2%) but not statistically significant (p=0.7563, wins 6 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
8.6173
Idea mean
8.6016
Effect (Δ)
-0.0157 (−0.2%; negative = idea better)
Wins
6 / 8 paired seeds
p-value
0.7563 (permutation test, 20 000 shuffles)
Smallest detectable effect
±2.0%
Mechanism
Not confirmed ✗
Practical effect
No effect
Baseline tuning
swept over 4 configs
Limitations:

Only the registered tabular/Friedman regression track was tested, not vision, sequence, or dynamics. The benchmark used mlp_tiny width 64 and 18 epochs; exact discrete merging and inference memory/latency reduction were not demonstrated because no duplicates formed. The soft proxy is an MVP approximation and was not shown to induce duplicate hyperplanes.

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

Built and ran the matched tabular MLP benchmark using the registered built-in track, with an equal-budget tuned baseline sweep and 8 paired seeds. The idea reached MSE 8.601595 versus baseline 8.617290, but the paired permutation p-value was 0.75635, so the result is not a significant win; trained models formed no duplicate kink clusters and the mechanism signature was not confirmed.

Baseline
8-seed mean test MSE 8.6172895432; best lr=0.01, weight_decay=0.0.
Idea
8-seed mean test MSE 8.6015948653; best lr=0.01, lambda_k=0.0001; paired delta=-0.0156946778; 6/8 wins; p=0.75635.

Benchmark result

Ahead of the baseline (−0.2%) but not statistically significant (p=0.7563, wins 6 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
8.6173
Idea mean
8.6016
Effect (Δ)
-0.0157 (−0.2%; negative = idea better)
Wins
6 / 8 paired seeds
p-value
0.7563 (permutation test, 20 000 shuffles)
Smallest detectable effect
±2.0%
Mechanism
Not confirmed ✗
Practical effect
No effect
Baseline tuning
swept over 4 configs
Limitations:

Only the registered tabular/Friedman regression track was tested, not vision, sequence, or dynamics. The benchmark used mlp_tiny width 64 and 18 epochs; exact discrete merging and inference memory/latency reduction were not demonstrated because no duplicates formed. The soft proxy is an MVP approximation and was not shown to induce duplicate hyperplanes.

How to run: python3 stage2_bench.py

Artifacts

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