Covering-Based Interaction Regularization
Implementation & benchmark of arXiv:2607.07742 — Discrete Faà di Bruno via Möbius Inversion
Source paper: Discrete Faà di Bruno via Möbius Inversion arXiv:2607.07742 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Regularize a neural network using exact finite-difference interaction terms at a chosen perturbation scale, while retaining the covering decomposition of a composition f∘g. Instead of penalizing only the total mixed difference, separately penalize selected covering terms containing large subsets or overlapping subsets, which targets higher-order and nonlocal interactions without computing Hessians.
Formulas
Mathematical statement
For maps g:X→Y and f:Y→Z between abelian groups, define the k-direction forward difference by Δ(g;x;u_1,…,u_k)=∑_{S⊆[k]}(-1)^{k-|S|}g(x+∑_{i∈S}u_i), where [k]={1,…,k}. Let Cov(k) be the set of families H of nonempty subsets T⊆[k] whose union is [k]. The paper's discrete Faà di Bruno identity is Δ(f∘g;x;u_1,…,u_k)=∑_{H∈Cov(k)}Δ(f;g(x);(Δ(g;x;u_T))_{T∈H}), where u_T denotes the directions indexed by T. Each outer difference is computed by the same inclusion-exclusion rule over the directions indexed by H. In a neural network, take g to be a feature trunk and f to be its head, x an input, and u_i small structured perturbations. The regularizer penalizes the squared norm of selected covering terms R_H, especially families H containing a block |T|≥2 or overlapping blocks, while optionally preserving singleton-only terms. The identity provides an exact numerical check that the selected decomposition sums to the direct mixed difference.
Implementation notes
Integrate this at the loss level around a feature trunk g and prediction head f. Use k=2 or k=3 and generate perturbations u_i in input space; for images, use independent small Gaussian, color, or spatial perturbations, and for token models use continuous embedding perturbations before the embedding projection. For every minibatch element, evaluate g(x+sum_{i in S}u_i) for all S⊆[k], cache the results, and compute each inner difference by the inclusion-exclusion formula above. Enumerate families H of nonempty subsets whose union is [k]; for k=2 or 3 this is cheap by brute force. For each chosen H, form the list of inner differences Δg(x;u_T), feed these directions to f at base point g(x), and compute the outer finite difference. A practical penalty set is all H containing a block with |T|≥2, or all H with overlapping subsets; set w_H=1/|H| and normalize each term by the product of perturbation norms. Add the penalty to cross-entropy or MSE with λ swept over {1e-4,1e-3,1e-2}. The direct mixed difference and covering sum should agree numerically; log R_check as an implementation invariant. The mathematics determines the inclusion-exclusion and covering enumeration exactly; perturbation scales, penalty weights, and which H are penalized are empirical choices. First test on CIFAR-10 with a small ResNet-18, comparing ordinary weight decay, a standard Jacobian finite-difference penalty, and this covering penalty. Measure clean accuracy, accuracy under composed perturbations, calibration, and training stability across perturbation scales. Success is lower error under two simultaneous perturbations and smoother loss trajectories at equal clean accuracy, without more than roughly 2x training FLOPs for k=2.
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.