Identity-Exact Multi-Bias Activations: Breaking Optimization Symmetry Without Changing the Pretrained Function
Abstract
A nominally three-branch activation can collapse to only two learned knot locations. In the multi-bias parameterization studied here, identity initialization ties its two coefficient- branches. Because these branches evaluate the same feature and receive equal gradients, AdamW's normalized update is identical for the two branches up to its term, preserving the tie throughout optimization. This optimizer-induced branch collapse helps explain why additional branches can match or underperform a same-site control despite the model class's greater expressivity. A multi-bias unit replaces with along a shared preactivation. With learned real coefficients, a ReLU multi-bias unit is a one-dimensional spline, represents XOR with , and, after thresholding, is Boolean complete on finite hypercubes; an exhaustive exact-arithmetic certificate further shows that some five-input Boolean function requires at least eight knots, . The deployed fixed-sign parameterization occupies a degenerate corner of this family: alternating coefficients restrict the available slope increments, while tied biases create the optimizer symmetry. Separating the knots breaks this symmetry but, in a controlled audit, perturbs the pretrained activation by up to in relative . We resolve this conflict with identity-exact nesting, in which one branch starts with coefficient one and additional branches start with coefficient zero at distinct knots. The resulting network exactly preserves the pretrained function at initialization, while the added branches receive distinct coefficient gradients on the first optimization step. Across four frozen ResNet-50 transfer tasks, nested significantly improves over its matched-site control by – accuracy points over three seeds and achieves higher training accuracy in every case. On CIFAR-100, a matched fixed-sign replication is significantly worse than that control ( points), while nested is points better: parameterization decides the effect's sign. Nested is never significantly worse than this control in any tested cell: ViT-B/16 is a four-seed statistical tie, and a 30-epoch sweep peaks at . These results show that expressivity and optimizer reachability can diverge even within a single activation family.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.