Inverse-Square Fractional Attention
Implementation & benchmark of arXiv:2607.09585 — Sharp and Endpoint Two-Weight Fractional Integral Estimates for Schr"odinger Operators with Inverse-Square Potentials
Source paper: Sharp and Endpoint Two-Weight Fractional Integral Estimates for Schr"odinger Operators with Inverse-Square Potentials arXiv:2607.09585 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace or augment geometric attention on spatial or point-cloud tokens with a positive fractional kernel containing the paper's inverse-square origin factor. This gives tokens near a designated singular center a controlled increase in receptive-field influence while preserving a scale-invariant distance decay, which may help models represent cusp-like fields, radial singularities, and multiscale spatial interactions.
Formulas
Mathematical statement
The paper studies the inverse-square Schrödinger operator H_a=-Delta+a|x|^{-2} on R^d, with d>=3 and -(d-2)^2/4<=a<0. Define the Hardy singularity exponent sigma=(d-2-sqrt((d-2)^2+4a))/2, so 0<sigma<=(d-2)/2. For 0<s<d-2sigma, the positive kernel of the fractional inverse operator satisfies H_a^{-s/2}(x,y) approximately proportional to |x-y|^{s-d} times (min(|x|/|x-y|, |y|/|x-y|, 1))^{-sigma}, where x,y are points in R^d and min denotes the minimum of the three arguments. The factor |x-y|^{s-d} is fractional-integral decay, while the minimum factor amplifies interactions when either endpoint is close to the singular origin relative to their separation. The restriction s<d-2sigma keeps the origin singularity integrable. We use this positive kernel as a geometry-dependent attention prior while retaining content-dependent query-key logits.
Implementation notes
Integrate this at the attention-logit stage of a vision transformer, point-cloud transformer, or coordinate-aware graph neural network whose tokens have coordinates x_i in R^d. Let c be the singularity center; initially use the image center, point-cloud centroid, or a fixed physical origin. For each head, compute ordinary queries q_i, keys k_j, and values v_j. In parallel compute r_ij=||x_i-x_j||_2+epsilon, u_i=||x_i-c||_2+epsilon, and the kernel K_ij from the displayed formula. Add lambda log(K_ij+epsilon) to the standard scaled dot-product logits, mask invalid pairs, and apply softmax. Normalize coordinates to a bounded domain and use epsilon=10^-4. Initialize lambda=0 to isolate the effect, then learn a nonnegative lambda per head. Optionally learn sigma using sigma=(d-2)/2 sigmoid(b), and parameterize s=(d-2sigma) sigmoid(r) so the paper's admissibility condition is always satisfied. For large token sets, use the k nearest neighbors together with a few radial bins rather than dense pairwise attention. The paper determines the kernel form and admissible exponent range; c, lambda, and optional exponents are estimated by backpropagation. First test on a 32x32 synthetic coordinate-regression task with clipped singular targets such as f(x)=||x-c||^{-gamma}, followed by a point-cloud classification benchmark. Compare vanilla attention and relative-position bias at equal parameters and FLOPs. Success is lower error on singular or multiscale targets, faster loss descent, or better accuracy under token subsampling. Permuting coordinates is a negative control and should remove any gain.
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 7/10
Built inverse-square fractional attention with the paper-derived positive kernel added to attention logits, plus numerical verification and a matched synthetic singular-field regression. The kernel was scale-homogeneous to relative error below 3e-8, matched the predicted 51.99x near-origin amplification, and passed the d=3 admissibility check. Across three fixed seeds, overall test MSE improved from mean 0.005827 to 0.005710 (~2.0%), but near-origin MSE slightly worsened (0.006125 to 0.006261), so the gain is modest and not specifically demonstrated as improved cusp accuracy.
- Agent confidence
- 7/10
- Baseline
- Three-seed mean overall test MSE 0.005827; mean near-origin MSE 0.006125; mean far-region MSE 0.005602
- Idea
- Three-seed mean overall test MSE 0.005710; mean near-origin MSE 0.006261; mean far-region MSE 0.005465; overall relative improvement ~2.0%
Limitations: Only a small synthetic 2D coordinate-token regression was tested; the kernel sanity check used d=3 while the toy attention model used 2D coordinates. No image or point-cloud benchmark, learned lambda or sigma, relative-position baseline, FLOP-normalized comparison, subsampling test, or permutation negative control was run.
How to run: python3 experiment.py && /home/maxwelhelp/main/bin/python3 repeat.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.09585)
Significantly WORSE than the tuned baseline (+4.2%, p=0.0386, wins 2 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.1321
- Idea mean
- 0.1377
- Effect (Δ)
- 0.0056 (+4.2%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.0386 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±4.0%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the structurally matched sequence track was tested. Fixed s, sigma, lambda, and center were not swept; no vision or point-cloud benchmark was available. The run used 18 epochs and 2000/500 train/test samples. The temporal coordinate embedding makes the self-distance comparison numerically extreme, so the signature was conservatively marked unconfirmed.
How to run: python3 bench_run.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 inverse-square fractional attention in a matched transformer on the sequence forecast benchmark using train_model, a shared learning-rate grid, 8 paired seeds, and the benchmark permutation test. The math check passed, but the idea significantly worsened test MSE: 0.137701 versus 0.132134, paired delta +0.005567, p=0.0386. The trained-model mechanism signature was not confirmed because the observed attention odds ratio did not quantitatively match the kernel prediction.
- Baseline
- Best lr=0.0015; 8-seed sequence test MSE mean 0.132134, std 0.012442
- Comparison
- paired delta +0.005567; idea wins 2/8 seeds; permutation p=0.0386; verdict idea worse (significant)
- Idea
- Best shared-grid lr=0.006; 8-seed sequence test MSE mean 0.137701, std 0.012268
- Math check
- Scale-homogeneity max relative error 3.40e-05; near-origin amplification 40.8057 predicted and observed; admissibility passed
- Mechanism signature
- Predicted kernel ratio 2.5716e9; observed trained attention odds ratio 4.5553e5; confirmed=false
Benchmark result
Significantly WORSE than the tuned baseline (+4.2%, p=0.0386, wins 2 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.1321
- Idea mean
- 0.1377
- Effect (Δ)
- 0.0056 (+4.2%; negative = idea better)
- Wins
- 2 / 8 paired seeds
- p-value
- 0.0386 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±4.0%
- Mechanism
- Not confirmed ✗
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the structurally matched sequence track was tested. Fixed s, sigma, lambda, and center were not swept; no vision or point-cloud benchmark was available. The run used 18 epochs and 2000/500 train/test samples. The temporal coordinate embedding makes the self-distance comparison numerically extreme, so the signature was conservatively marked unconfirmed.
How to run: python3 bench_run.py
Artifacts
- bench_report.json 3.0 KB View Raw
- bench_run.py 6.4 KB View Raw
- experiment.py 4.2 KB View Raw
- repeat.py 1.3 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-09-04T211721.md 4.9 KB View
- results.json 0.8 KB View Raw