Mean-Curvature Relaxation Layer
Implementation & benchmark of arXiv:2608.08901 — Well-posedness for the mean curvature flow on the half-space and on bounded domains
Source paper: Well-posedness for the mean curvature flow on the half-space and on bounded domains arXiv:2608.08901 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Insert a small number of differentiable graphical mean-curvature-flow steps between a neural network's raw vector-field prediction and its task loss. The relaxation performs geometry-aware smoothing rather than isotropic Gaussian smoothing, and it can enforce fixed boundary values after every step.
Formulas
Mathematical statement
For a graph f: \\(\\Omega\\subset\\mathbb{R}^{d}\\to\\mathbb{R}^{N}\\), the paper studies \\(\\partial_t f=\\mathsf{A}[\\nabla f]:\\nabla^2f\\), where \\(\\nabla f\\in\\mathbb{R}^{N\\times d}\\), \\(\\nabla^2f\\) is the spatial Hessian of every output component, and \\(\\mathsf{A}[\\nabla f]=(\\mathrm{Id}+\\sum_{\\alpha=1}^{N}\\nabla f^{\\alpha}\\otimes\\nabla f^{\\alpha})^{-1}\\). The contraction is \\(\\mathsf{A}:\\nabla^2f=\\sum_{i,j=1}^{d}\\mathsf{A}_{ij}\\partial_i\\partial_jf\\), applied componentwise. The ellipticity inequality \\( |\\xi|^2/(1+|\\nabla f|^2)\\leq\\xi\\cdot\\mathsf{A}[\\nabla f]\\xi\\leq|\\xi|^2 \\) shows that bounded slope gives uniformly parabolic diffusion. The adaptation uses this explicit metric-dependent diffusion as a post-processing or trainable architecture layer.
Implementation notes
1. Integration point: let a U-Net, CNN, or coordinate MLP produce a tensor u with shape batch by N by H by W, representing optical flow, a deformation field, or multiple surface-height channels. Apply K relaxation steps to u before computing the task loss. If homogeneous Dirichlet values are required, maintain a boundary mask and overwrite those pixels with zero after each update. 2. Pseudocode: set f = u. For k from 1 to K, compute finite-difference gradients g_alpha = grad_h(f_alpha). At each pixel form G = I_d + sum over alpha of outer(g_alpha, g_alpha). Compute A = inverse(G), compute finite-difference Hessians H_alpha[i,j] = D2_ij,h(f_alpha), and update f_alpha = f_alpha + eta times sum over i,j of A[i,j] times H_alpha[i,j]. Reapply the boundary mask. Backpropagate through all steps. 3. The paper-derived quantities are G, A, the anisotropic Hessian contraction, and the ellipticity diagnostic lambda_min(A) >= 1 divided by (1 + maximum slope squared). Estimate the maximum slope directly from finite differences and choose eta conservatively, initially eta <= 0.2 h squared for grid spacing h. 4. First experiment: train a small U-Net on Sintel or FlyingChairs optical flow. Compare the baseline head, isotropic Laplacian relaxation, and this layer for K in {1, 2, 4, 8}, at matched parameter counts. Measure endpoint error, boundary error, high-frequency output energy, gradient norms, and runtime. The falsifiable target is lower endpoint error and less oscillation than isotropic smoothing at equal FLOPs, without unstable gradients.
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.