Monotone Resolvent Elimination Layer
Implementation & benchmark of arXiv:2607.07358 — Maximal monotonicity of piecewise polyhedral mappings
Source paper: Maximal monotonicity of piecewise polyhedral mappings arXiv:2607.07358 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Build an implicit layer from a piecewise-linear maximal monotone operator on visible variables z_* and auxiliary variables z_**, then eliminate the auxiliary block rather than exposing it in the network output. Compute the layer through a fixed point of the eliminated component of a nonexpansive resolvent, with damping when the auxiliary map is not strictly contractive.
Formulas
Mathematical statement
Let H=H_*\oplus H_** be a finite-dimensional real Hilbert space and let T:H\rightrightarrows H be maximal monotone with piecewise polyhedral graph. The reduced relation T_*:H_*\rightrightarrows H_* is defined by w_*\in T_*(z_*) if and only if there exists z_\in H_ such that (w_*,0)\in T(z_*,z_**) (equation (1)); the zero in the eliminated output coordinate imposes internal balance. If there exists a witness (z_*,z_**,w_*) with (w_*,0)\in T(z_*,z_**) (equation (2)), then T_* is maximal monotone. Minty's parametrization uses the resolvent R=(I+T)^{-1}; maximal monotonicity makes R single-valued and nonexpansive, so ||R(a)-R(b)||\leq||a-b||. The piecewise-polyhedral graph makes R piecewise affine. The neural adaptation represents R(x,u)=[R_*(x,u),R_**(x,u)], where x\in H_* is the visible input and u\in H_ is an auxiliary state. The auxiliary state is obtained from the fixed-point equation u=R_(x,u), and the returned visible representation is R_*(x,u).
Implementation notes
Integrate this as an implicit residual/MLP block with a retained feature vector x in R^{d_*} and an auxiliary vector u in R^{d_**}. Parameterize one joint piecewise-affine map R(x,u)=[R_*(x,u),R_**(x,u)] using linear layers and ReLU or leaky-ReLU branches. Apply spectral normalization or explicit matrix rescaling to the joint linear operators so that the estimated global Lipschitz constant of R is at most one. In the forward pass, initialize u^0=0, or reuse the previous u for temporally related inputs, and perform K iterations: compute (v_*,v_)=R(x,u^k), then set u^{k+1}=(1-alpha)u^k+alpha v_. Return y=v_* from the final iteration, optionally followed by a learned linear projection and residual skip x+y. Backpropagate through the K iterations for the first implementation; implicit differentiation can be added once fixed-point convergence is reliable. The paper supplies the reduced-operator definition, the resolvent nonexpansiveness target, and the piecewise-affine structure. Estimate the auxiliary contraction factor empirically with finite differences or power iteration: q=max ||R_(x,u+delta)-R_(x,u)||/||delta|| on held-out minibatches. Use alpha=1 if q<0.9; otherwise use alpha=min(1,0.9/(q+1e-6)). Track the fixed-point residual ||u^{k+1}-u^k||, activation norms, gradient norms, and perturbation amplification. First test on a four-layer MNIST or CIFAR-10 MLP against a standard residual MLP with matched parameters and FLOPs. The desired signal is fewer unstable training runs, bounded input perturbation amplification, and equal or better accuracy at a learning rate where the baseline becomes unstable. Also report iteration overhead and test whether smaller auxiliary dimensions reduce activation memory without hurting accuracy.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.