Trainable at Depth, Solvable at Small Size: What the Loss Surface of a Polynomial Activation Dictates for Designing, Initializing and Training Networks

cs.LG cs.AI Xiuyi Yang · Oct 14, 2026
Local to this browser
What it does
Training deep networks without normalization remains difficult, and relating their trainability to the exact geometry of the loss surface is rarely attempted. This paper proposes a projective polynomial activation $f(z)=z+\lambda z^3$...
Why it matters
This paper proposes a projective polynomial activation $f(z)=z+\lambda z^3$ paired with either an $\ell_2$ projection onto the unit sphere or an unnormalized identity path, and studies it at two scales: networks that train to depth...
Main concern
The paper makes a methodologically novel and welcome attempt to connect exact algebraic geometry to deep network engineering. The empirical separation of scale control from activation choice within the fully-connected MNIST regime is...
Community signal
0
0 up · 0 down
Sign in to vote with arrows
AI Review AI reviewed
Plain-language introduction

Training deep networks without normalization remains difficult, and relating their trainability to the exact geometry of the loss surface is rarely attempted. This paper proposes a projective polynomial activation $f(z)=z+\lambda z^3$ paired with either an $\ell_2$ projection onto the unit sphere or an unnormalized identity path, and studies it at two scales: networks that train to depth $100$--$300$ on MNIST, and tiny algebraic versions whose square-loss critical points are solved exactly via Gröbner bases and numerical irreducible decomposition. The point is to derive concrete design, initialization, and diagnostic rules from the small model's geometry and validate them at depth.

Critical review
Verdict
Bottom line

The paper makes a methodologically novel and welcome attempt to connect exact algebraic geometry to deep network engineering. The empirical separation of scale control from activation choice within the fully-connected MNIST regime is compelling, and the solvable-tier computations with Bertini are rigorous. However, the scope is extremely narrow, the transfer from small algebraic models to deep trained networks is asserted as causal rather than proven, and the abstract directly contradicts Section 6.3 on whether the output-zero critical point has a vanishing or nonzero Hessian, which undermines confidence in the solvable-tier narrative.

“...and that point is the output-zero network with a vanishing Hessian; either scale-control mechanism makes the critical set positive-dimensional.”
paper · Abstract
“...and the Hessian there is nonzero but indefinite (rank 2, eigenvalues ±3.1600): the single real stationary point is a saddle...”
paper · Section 6.3
“We claim no theorem: the projective view motivates the architecture, and our claims are restricted to measured behaviour.”
paper · Section 2, Related Work
What holds up

The deep-tier experiments convincingly establish that scale control dominates activation choice for trainability at depth. The $\ell_2$ projection rescues the polynomial network from chance to matching ReLU within $0.29$ points at depth $30$, while an unnormalized identity path takes it to $94.50\%$ at depth $100$ and $93.80\%$ at depth $300$ by allowing hidden norms to grow instead of saturating at the clamp bound. On the solvable tier, the numerical irreducible decomposition with Bertini rigorously verifies that the loss is constant on each critical component and that the number of zero Hessian eigenvalues equals the component dimension, giving strong algebraic footing to the geometric intuition. As the authors summarize in Section 6.1, "The activation choice is second-order; scale control is first-order."

“The activation choice is second-order; scale control is first-order.”
paper · Section 6.1
“This is the sharper of the two, and it holds without exception: every dimension-d component has exactly d zero eigenvalues at each of its witness points, the count taken over the trainable parameters only.”
paper · Section 5.5
Main concerns

A glaring internal inconsistency damages the small-model claims. The abstract states the real critical point is "the output-zero network with a vanishing Hessian," yet Section 6.3 explicitly states it has a "nonzero but indefinite (rank 2, eigenvalues $\pm 3.1600$)" Hessian and calls it a saddle. This contradiction makes the solvable-tier exposition hard to trust. More broadly, the paper admits in Section 2 that it "claims no theorem" and that claims are "restricted to measured behaviour," but later sections repeatedly assert that the small models "prescribe" or "dictate" deep-network design, a causal overreach. The evaluation is also confined to MNIST and fully-connected stacks; as noted in Section 6.5, the conclusions "do not automatically transfer to convolutional or attention architectures." Finally, the projection abruptly fails at depth $50$ for reasons the algebraic model does not explain, while the residual alternative permits unbounded growth in hidden norms (up to $15729.8$) whose practical implications are unexplored.

“...and that point is the output-zero network with a vanishing Hessian; either scale-control mechanism makes the critical set positive-dimensional.”
paper · Abstract
“...and the Hessian there is nonzero but indefinite (rank 2, eigenvalues ±3.1600): the single real stationary point is a saddle...”
paper · Section 6.3
“We claim no theorem: the projective view motivates the architecture, and our claims are restricted to measured behaviour.”
paper · Section 2, Related Work
“We study fully-connected stacks; our conclusions about the projection as a substitute for normalization are statements about scale control in a feed-forward stack and do not automatically transfer to convolutional or attention architectures...”
paper · Section 6.5
Evidence and comparison

Within the studied regime, the evidence supports the qualitative claim that the unnormalized polynomial dies by saturation—hidden norms hit the clamp-imposed bound $226.27$—and that both the projection and the residual remedy this, consistent with the algebraic finding that their critical sets become positive-dimensional. However, the projective polynomial remains less accurate than ReLU, GELU, or SiLU at equal depth and parameter count, so the evidence does not support the polynomial as a better nonlinearity. Comparisons to gated units show collapse without normalization but do not fully tabulate their repaired performance once normalized, weakening the claim that gating confers no advantage. The correspondence between the untruncated square-loss ideal and the clamped, cross-entropy-trained deep network is asserted but not rigorously derived.

“The failure it repairs is saturation, not a lost direction: the failing arm ends at the clamp-imposed norm bound 226.27.”
paper · Section 5.3
“...the bare cubic z+λ z^3 lands almost four accuracy points below ReLU, GELU and SiLU, and the parameter-free ℓ_2 projection closes a little over half of that gap without closing it; the remainder is not recovered by increasing the degree or making the polynomial learnable.”
paper · Section 6.1
Reproducibility

Reproducibility is strong in terms of artifacts: complete code, JSON logs, and macro-generation scripts are released under an MIT license, and the authors report that "Every number here is produced by the harness in experiments/ and injected into the text as a LaTeX macro by make_macros.py." Hyperparameters and the harness structure are documented in detail. Weaknesses include that most configurations use a single seed (42), and the collapsed training arms exhibit numerical sensitivity to thread count and accumulation order, producing spreads of several points that the authors explicitly warn should not be read to the last digit.

“Every number here is produced by the harness in experiments/ and injected into the text as a LaTeX macro by make_macros.py...”
paper · Reproducibility subsection
“The core comparisons use a single seed, at a resolution set by the five-seed spread of the headline configuration.”
paper · Section 6.5
“In the saturated, no-normalization configurations the exact accuracy is sensitive to floating-point accumulation order: repeating one seeded run under different thread counts gave 10.3%, 11.4% and 13.3%, a spread of three points that is entirely absent from the residual arm.”
paper · Section 6.5
Abstract

What makes a deep network train, and what shape is its loss surface? We answer both with one construction at two scales: a polynomial activation f (z) = z + {\lambda}z 3 , clamped and projected onto the unit sphere, plus an unnormalized identity path. As a deep stack it trains to one hundred layers on MNIST with no normalization machinery; as a small model its square loss is a polynomial system we solve exactly. The small scale explains the large one: it prescribes how to design, initialize and train the stack.

Challenge the Review

Pick a starting point or write your own. Challenges run in the background, so you can keep reading while the AI investigates.

No challenges yet. Disagree with the review? Ask the AI to revisit a specific claim.