Leibenson Nonlinear Diffusion Layer
Implementation & benchmark of arXiv:2608.15168 — Leibenson's equation on graphs
Source paper: Leibenson's equation on graphs arXiv:2608.15168 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a standard graph-convolution propagation step with a short time integration of the nonlinear graph flow \(\partial_t u=\Delta_p(u^q)\). The pointwise power \(q\) and gradient exponent \(p\) create state- and edge-gradient-dependent propagation: small signals can be suppressed or amplified by \(q\), while large graph discrepancies receive nonlinear diffusion controlled by \(p\). Use nonnegative feature states and conservative edge fluxes so the layer inherits positivity and total-mass preservation.
Formulas
Mathematical statement
The central equation is Leibenson's nonlinear diffusion equation \(\partial_t u=\Delta_p u^q\), where \(u_i(t)\ge 0\) is the scalar state at graph node \(i\), \(p>1\) is the p-Laplacian exponent, and \(q>0\) is a pointwise power. The discrete p-Laplacian is defined by \(\Delta_pu:=\operatorname{div}(|\nabla u|^{p-2}\nabla u)\). For a weighted undirected graph with edge weights \(w_{ij}\ge0\), use \((\Delta_p f)_i=\sum_{j\in\mathcal N(i)}w_{ij}|f_j-f_i|^{p-2}(f_j-f_i)\), with \(f_i=u_i^q\). The edge flux \(J_{ij}=w_{ij}|u_j^q-u_i^q|^{p-2}(u_j^q-u_i^q)\) is antisymmetric, \(J_{ij}=-J_{ji}\), so the sum of node values is preserved on an undirected graph without boundary. The paper proves global existence for integrable initial data and obtains smoothing, propagation, and finite-time extinction results under graph inequalities. The neural adaptation exploits the explicit conservative flux, while the theoretical qualitative behaviors motivate using the flow as a stable message-passing operator and, for suitable \(p,q\), as a sparsifying propagation mechanism.
Implementation notes
(1) Integration point: insert the layer after a graph encoder or between GraphSAGE/GAT blocks. For input features \(X\in\mathbb R^{n\times d}\), produce nonnegative states \(U^0=\operatorname{softplus}(XW+b)+\epsilon\), where \(\epsilon>0\) avoids singular derivatives near zero when \(q<1\). Apply diffusion independently to every feature channel. Use the original sparse adjacency with normalized nonnegative weights, or learn symmetric weights \(w_{ij}=\operatorname{softplus}(a_{ij})\). (2) Pseudocode: U = softplus(X @ W + b) + eps; for k=1..K: F=U**q; D=F[dst]-F[src]; J=w*abs(D)**(p-2)*D; div=scatter_add(J,dst)-scatter_add(J,src); U=U+tau*div; return Y=U @ Wo + bo. The opposite signs in the two scatter operations implement the antisymmetric flux. (3) Computed from the mathematics: powered node states, nonlinear edge fluxes, divergence, and Euler recursion. Estimated empirically: choose \(\tau\) with a stability sweep; use \(L\approx q\max_i(U_i+\epsilon)^{q-1}\) and weighted degree \(d_{w,\max}=\max_i\sum_jw_{ij}\), then enforce \(\tau Ld_{w,\max}<0.5\) as a conservative initial rule. Clip or backtrack \(\tau\) if NaNs or feature explosion occur. (4) Cheap experiment: compare a 2-layer GCN, GraphSAGE, and this layer on Cora, Citeseer, and a synthetic homophilic graph, matching parameter count and message-passing FLOPs. Sweep \(p\in\{1.5,2,3\}\), \(q\in\{0.5,1,2\}\), and \(K\in\{1,2,4\}\). Measure validation accuracy, feature Dirichlet energy, total feature-mass drift, gradient norms, and accuracy under random edge deletion. Success means improved noisy-graph accuracy or validation accuracy at equal depth, with mass drift close to numerical precision and fewer exploding or vanishing-gradient failures than the baseline.
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.