mlsanityData to 7 Oct 2026

Fitting Multilinear Polynomials for Logic Gate Networks

arXiv 2605.08657 ↗First posted 9 May 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

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, at least one 4-parameter method matches or exceeds Soft-Mix on every dataset; the CovJac advantage over STE grows monotonically with interaction demand across all seven datasets. At depth, Soft-Mix collapses (-37.3pp on CIFAR-10 at 12 layers) while CovJac holds (-0.5pp on CIFAR-10, stable on MNIST).

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 35 references that are in the map. First posted on arXiv 9 May 2026.

Builds on

15 of 35 references.

References · 15

  1. 2017Neural Discrete Representation Learningcited by 4,650
  2. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
  3. 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
  4. 2018DARTS: Differentiable Architecture Searchcited by 1,782
  5. 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
  6. 2017Soft-to-Hard Vector Quantization for End-to-End Learning Compressible Representationscited by 149
  7. 2020Analyzing Differentiable Fuzzy Logic Operatorscited by 100
  8. 2023Straightening Out the Straight-Through Estimator: Overcoming Optimization Challenges in Vector Quantized Networkscited by 80
  9. 2022Deep Differentiable Logic Gate Networkscited by 67
  10. 2024Convolutional Differentiable Logic Gate Networkscited by 37
  11. 2023PolyLUT: Learning Piecewise Polynomials for Ultra-Low Latency FPGA LUT-based Inferencecited by 30
  12. 2024Differentiable Weightless Neural Networkscited by 27
3 more
  1. 2025Light Differentiable Logic Gate Networkscited by 9
  2. 2024Towards Narrowing the Generalization Gap in Deep Boolean Networkscited by 3
  3. 2026Align Forward, Adapt Backward: Closing the Discretization Gap in Logic Gate Networkscited by 2

Built on by

Not cited yet.