acceptodds
Under review as a conference paper at ICLR 2027

Fitting Multilinear Polynomials for Logic Gate Networks

Abstract

We study learnable logic gate networks that stack layers of 2-input Boolean gates to build combinational circuits. Every 2-input gate has a unique multilinear polynomial with 4 coefficients, so the 16 Boolean gates form a codebook of prototypes in a 4-dimensional space, reducing training to a vector-quantization problem. The baseline method, Soft-Mix, learns a 16-dimensional softmax over gate identities, but the codebook has rank 4: 11 of 15 simplex directions carry nullspace gradient, and at uniform initialization the backward signal vanishes exactly. We prove that no affine product reparameterization fixes the resulting interaction-coefficient starvation under STE, and show that the covariance Jacobian of soft-VQ selection bypasses it by coupling the starved coefficient to the always-active constant channel. Working in the 4-dimensional polynomial space reduces each neuron from 16 to 4 parameters. On seven datasets, with every accuracy measured on the deployed strict-Boolean (zero-multiply-accumulate) circuit, our primary method, Multilinear-CovJac, matches or exceeds 16-parameter Soft-Mix on every dataset, and its advantage over STE grows monotonically with interaction demand. At depth, Soft-Mix collapses (pp on CIFAR-10 at 12 layers) while CovJac holds (pp on CIFAR-10, stable on MNIST).

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.