Lipschitz-Free Metric Pooling
Implementation & benchmark of arXiv:2607.06049 — Analytic integration of metric-valued functions in Lipschitz free spaces
Source paper: Analytic integration of metric-valued functions in Lipschitz free spaces arXiv:2607.06049 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace coordinate-wise mean pooling of metric-valued items with a finite representation of their free integral. Each item x in a pointed metric space M is represented through evaluations of learned Lipschitz probes, and the pooled feature is the weighted integral of those probe values. A dual Lipschitz critic estimates the free-space norm of differences between pooled groups, making the representation sensitive to metric geometry while remaining permutation-invariant.
Formulas
Mathematical statement
The paper identifies the dual of the Lipschitz-free space with Lipschitz functions vanishing at a base point: \(\mathcal{F}(M)^*\cong\mathrm{Lip}_0(M)\). For a pointed metric space \((M,d,\theta)\), \(\mathrm{Lip}_0(M)\) consists of real functions \(\varphi\) with \(\varphi(\theta)=0\) and finite Lipschitz constant \(\mathrm{Lip}(\varphi)=\sup_{x\neq y}|\varphi(x)-\varphi(y)|/d(x,y)\). Corollary 3.36 gives \((I^{*}\varphi)(f)=\int_{\Omega}\varphi\circ f\,d\mu\), where \((\Omega,\mu)\) is a measure space, \(f:\Omega\to M\) is free-integrable, and \(I(f)=F\text{-}\int_{\Omega}f\,d\mu\in\mathcal{F}(M)\). Therefore evaluation of the free integral by any Lipschitz probe equals the integral of pointwise probe evaluations. For a finite weighted batch \(x_1,\ldots,x_n\in M\), with \(w_i\geq0\) and \(\sum_iw_i=1\), use the molecule \(m=\sum_iw_i(\delta_{x_i}-\delta_{\theta})\in\mathcal{F}(M)\). Its dual norm is \(\|m\|_{\mathcal{F}(M)}=\sup_{\mathrm{Lip}(\varphi)\leq1,\ \varphi(\theta)=0}\sum_iw_i\varphi(x_i)\). Approximate this supremum using K learned 1-Lipschitz probes.
Implementation notes
Integrate the layer immediately after a variable-size set, point-cloud neighborhood, or attention-selected token group. Let the input items be x_i, with a fixed differentiable metric d(x,y) or a pretrained metric encoder whose output is treated as the metric space M; choose a base point theta, usually a learned anchor or the zero embedding. Build K scalar probe networks phi_k(x), each constrained to vanish at theta. Enforce approximate 1-Lipschitzness by spectral-normalizing every linear layer and adding a sampled penalty max(0, |phi_k(x)-phi_k(y)|-d(x,y))^2 over item pairs. Compute attention or router weights w_i with softmax, evaluate all probes, and form z_k=sum_i w_i phi_k(x_i), producing z in R^K. For two groups X and Y, compute D_hat=max_k |sum_i w_i phi_k(x_i)-sum_j v_j phi_k(y_j)|. A training loop is: encode items; compute weights; evaluate probes; pool probe values; compute the task loss; optionally update probes for several ascent steps on D_hat or a supervised metric-matching loss; then update the main network by descent, stopping probe gradients during the main-model update if critic instability appears. The paper supplies the dual integration identity; the finite probe bank, spectral normalization, and critic optimization are approximations. First test on ModelNet40 point-cloud classification or synthetic sets on a circle with nonlinear coordinate distortions. Compare mean pooling, DeepSets, and attention pooling at equal parameter count. Success means higher accuracy under metric distortions, improved set-retrieval recall, and graceful performance degradation when 20-50% of items are removed.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗
Stage 2 · Practical benchmark result: Significantly worse ✗
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 8/10
Built a reproducible Lipschitz-free probe pooling MVP with finite-metric LP verification and a matched-parameter toy classification experiment. The exact distance-to-anchor probe was valid in all 30 trials and matched the exact dual free norm; across five seeds, probe pooling improved clean accuracy from 0.497 to 0.691 and retained 0.644 accuracy after removing 50% of items versus 0.496 for mean pooling. This is a clear signal on the constructed synthetic task, but not evidence of broad superiority.
- Agent confidence
- 8/10
- Baseline
- Mean pooling: 42 parameters, clean accuracy 0.4969±0.0245, 50% removal accuracy 0.4957±0.0193
- Idea
- Lipschitz probe pooling: 42 parameters, clean accuracy 0.6910±0.0496, 50% removal accuracy 0.6438±0.0403; exact math check valid-probe fraction 1.0 and probe/exact norm ratio 1.0
Limitations: Only a small synthetic 2D radial/angular classification task was tested. The experiment did not test ModelNet40, retrieval, nonlinear metric distortions, learned attention/router weights, explicit pairwise Lipschitz penalties, or comparisons against DeepSets and attention pooling.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.06049)
Significantly WORSE than the tuned baseline (+1.7%, p=0.0163, wins 1 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny_pooling_mvp
- Paired seeds
- 8
- Baseline mean
- 1.0108
- Idea mean
- 1.0283
- Effect (Δ)
- 0.0175 (+1.7%; negative = idea better)
- Wins
- 1 / 8 paired seeds
- p-value
- 0.0163 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±1.0%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered sequence forecasting track was tested. The implementation used scalar sequence values, fixed equal weights, no learned router, no explicit pairwise Lipschitz penalty, and no vision, dynamics, retrieval, point-cloud, or removal-robustness evaluation.
How to run: python3 stage2_bench.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
Implemented end-to-end Lipschitz-Free Metric Pooling on the registered sequence forecasting track with a bounded finite probe bank and matched mean-pooling baseline. Across 8 paired seeds, the idea was significantly worse: MSE 1.028305 versus 1.010802, paired delta +0.017503, permutation p=0.0163. The trained-model signature found zero Lipschitz violations but did not confirm positive geometric association.
- Baseline
- Mean pooling, best lr=0.003: MSE 1.010802±0.092335 across 8 seeds.
- Idea
- K=8 bounded probe pooling, best lr=0.006: MSE 1.028305±0.091821 across 8 seeds; 1/8 paired wins.
Benchmark result
Significantly WORSE than the tuned baseline (+1.7%, p=0.0163, wins 1 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny_pooling_mvp
- Paired seeds
- 8
- Baseline mean
- 1.0108
- Idea mean
- 1.0283
- Effect (Δ)
- 0.0175 (+1.7%; negative = idea better)
- Wins
- 1 / 8 paired seeds
- p-value
- 0.0163 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±1.0%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered sequence forecasting track was tested. The implementation used scalar sequence values, fixed equal weights, no learned router, no explicit pairwise Lipschitz penalty, and no vision, dynamics, retrieval, point-cloud, or removal-robustness evaluation.
How to run: python3 stage2_bench.py
Artifacts
- bench_report.json 3.9 KB View Raw
- experiment.py 6.2 KB View Raw
- report.md 1.3 KB View
- report_bench_2026-09-04T134741.md 5.5 KB View
- stage2_bench.py 5.1 KB View Raw