SabaPivot commited on
Commit
4f601f4
·
verified ·
1 Parent(s): d040a50

Expose five anchored Softmax-as-linear-attention claims

Browse files
logbook.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "schema_version": 1,
3
- "title": "Repro - Softmax as Linear Attention in the Large-Prompt Regime",
4
  "emoji": "🎯",
5
  "space_id": "SabaPivot/repro-softmax-linear-attention",
6
  "paper": {
@@ -13,9 +13,15 @@
13
  "updated_at": "2026-07-16T14:49:03+00:00",
14
  "root": {
15
  "slug": "index",
16
- "title": "Repro - Softmax as Linear Attention in the Large-Prompt Regime",
17
  "file": "pages/index.md",
18
  "children": [
 
 
 
 
 
 
19
  {
20
  "slug": "claim-1-finite-prompt-concentration-prop-3-1",
21
  "title": "Claim 1: Finite-prompt concentration (Prop 3.1)",
@@ -51,15 +57,9 @@
51
  "title": "Conclusion",
52
  "file": "pages/conclusion/page.md",
53
  "children": []
54
- },
55
- {
56
- "slug": "softmaxlin-repro",
57
- "title": "softmaxlin-repro",
58
- "file": "pages/softmaxlin-repro/page.md",
59
- "children": []
60
  }
61
  ]
62
  },
63
  "agent_view_tokens": 8996,
64
  "revision": "1784213343170298004"
65
- }
 
1
  {
2
  "schema_version": 1,
3
+ "title": "Reproduction: Softmax as Linear Attention in the Large-Prompt Regime: a Measure-based Perspective",
4
  "emoji": "🎯",
5
  "space_id": "SabaPivot/repro-softmax-linear-attention",
6
  "paper": {
 
13
  "updated_at": "2026-07-16T14:49:03+00:00",
14
  "root": {
15
  "slug": "index",
16
+ "title": "Reproduction: Softmax as Linear Attention in the Large-Prompt Regime: a Measure-based Perspective",
17
  "file": "pages/index.md",
18
  "children": [
19
+ {
20
+ "slug": "executive-summary",
21
+ "title": "Executive summary",
22
+ "file": "pages/executive-summary/page.md",
23
+ "children": []
24
+ },
25
  {
26
  "slug": "claim-1-finite-prompt-concentration-prop-3-1",
27
  "title": "Claim 1: Finite-prompt concentration (Prop 3.1)",
 
57
  "title": "Conclusion",
58
  "file": "pages/conclusion/page.md",
59
  "children": []
 
 
 
 
 
 
60
  }
61
  ]
62
  },
63
  "agent_view_tokens": 8996,
64
  "revision": "1784213343170298004"
65
+ }
pages/executive-summary/page.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Executive summary
2
+
3
+ ---
4
+ <!-- trackio-cell
5
+ {"type":"markdown","id":"softmax_exec_v2","created_at":"2026-07-22T04:05:00+00:00","title":"Executive summary","pinned":true,"pinned_at":"2026-07-22T04:05:00+00:00"}
6
+ -->
7
+ All five supplied claims for [Softmax as Linear Attention in the Large-Prompt Regime](https://openreview.net/forum?id=MvuCgK0Qns) were independently tested. The Gaussian infinite-prompt identity matches direct Monte Carlo at relative error 4e-4–1e-3 for L=10^7, including anisotropic and rank-degenerate covariances. Output and gradient errors exhibit the predicted prompt-length and variance signatures. In the authors' released training code, softmax/linear risk gap shrinks from 0.21 to 0.014 and same-seed parameter distance from 4.20 to 0.023 as L grows 10→1000. Under anisotropic KMS covariates, the learned matrix aligns with Σ⁻¹ at cosine 0.9999 and approaches the Bayes-risk level. The reproduction also identifies and fixes an upstream anisotropic data-generator bug before Claim 5 is evaluated.
8
+
9
+ ## Scope & cost
10
+
11
+ | Item | This reproduction | Full paper run |
12
+ | --- | --- | --- |
13
+ | Scope | All 5 claims; concentration, gradients, affine limit, risk transfer, anisotropic optimum | Full 500k-step grid and additional seeds |
14
+ | Scale | 400 prompts × 500 queries; L≤30,000 for rates; direct MC to 10,000,000; 12 training runs × 100k steps | Authors' 500k-step training schedule |
15
+ | Hardware | 1× H100 plus CPU | One GPU plus CPU |
16
+ | Compute/cost | About 1.5 GPU-hours plus CPU; approximately $5 | Approximately 5× training budget |
17
+ | Limitation | Training steps reduced from 500k to 100k; all reductions disclosed | Full training schedule |
18
+
19
+ Evidence links: [authors' repository](https://github.com/eboursier/softmax_as_linear), [artifact Bucket](https://huggingface.co/buckets/SabaPivot/repro-softmax-linear-attention-artifacts), and [reproduction bundle](https://huggingface.co/buckets/SabaPivot/repro-softmax-linear-attention-artifacts#softmaxlin-repro/repro-bundle:v1).
20
+
21
+ ---
22
+ <!-- trackio-cell
23
+ {"type":"figure","id":"softmax_poster_v2","created_at":"2026-07-22T04:05:00+00:00","title":"Reproduction poster (poster_embed.html)","source_path":"poster_embed.html","pinned":true,"pinned_at":"2026-07-22T04:05:00+00:00"}
24
+ -->
25
+ ````html
26
+ <!-- poster_embed.html -->
27
+ <iframe src="https://chenruishuo-posterly.hf.space" title="Posterly reproduction poster" style="width:100%;height:720px;border:0" loading="lazy"></iframe>
28
+ ````
pages/executive-summary/poster_embed.html ADDED
@@ -0,0 +1 @@
 
 
1
+ <iframe src="https://chenruishuo-posterly.hf.space" title="Posterly reproduction poster" style="width:100%;height:720px;border:0" loading="lazy"></iframe>
pages/index.md CHANGED
@@ -1,13 +1,13 @@
1
- # Repro - Softmax as Linear Attention in the Large-Prompt Regime
2
 
3
  ## Pages
4
 
5
  | Page |
6
  | --- |
 
7
  | [Claim 1: Finite-prompt concentration (Prop 3.1)](#/claim-1-finite-prompt-concentration-prop-3-1) |
8
  | [Claim 2: Gradient concentration (Prop 3.4)](#/claim-2-gradient-concentration-prop-3-4) |
9
  | [Claim 3: Infinite-prompt softmax is affine (Lemma 2.1)](#/claim-3-infinite-prompt-softmax-is-affine-lemma-2-1) |
10
  | [Claim 4: Risk transfer under gradient flow (Thm 4.3)](#/claim-4-risk-transfer-under-gradient-flow-thm-4-3) |
11
  | [Claim 5: Bayes-optimal anisotropic ICL (Thm 5.1)](#/claim-5-bayes-optimal-anisotropic-icl-thm-5-1) |
12
  | [Conclusion](#/conclusion) |
13
- | [softmaxlin-repro](#/softmaxlin-repro) |
 
1
+ # Reproduction: Softmax as Linear Attention in the Large-Prompt Regime: a Measure-based Perspective
2
 
3
  ## Pages
4
 
5
  | Page |
6
  | --- |
7
+ | [Executive summary](#/executive-summary) |
8
  | [Claim 1: Finite-prompt concentration (Prop 3.1)](#/claim-1-finite-prompt-concentration-prop-3-1) |
9
  | [Claim 2: Gradient concentration (Prop 3.4)](#/claim-2-gradient-concentration-prop-3-4) |
10
  | [Claim 3: Infinite-prompt softmax is affine (Lemma 2.1)](#/claim-3-infinite-prompt-softmax-is-affine-lemma-2-1) |
11
  | [Claim 4: Risk transfer under gradient flow (Thm 4.3)](#/claim-4-risk-transfer-under-gradient-flow-thm-4-3) |
12
  | [Claim 5: Bayes-optimal anisotropic ICL (Thm 5.1)](#/claim-5-bayes-optimal-anisotropic-icl-thm-5-1) |
13
  | [Conclusion](#/conclusion) |
 
pages/softmaxlin-repro/page.md DELETED
@@ -1,10 +0,0 @@
1
- # softmaxlin-repro
2
-
3
-
4
- ---
5
- <!-- trackio-cell
6
- {"type": "artifact", "id": "cell_f31339a2c059", "created_at": "2026-07-16T14:48:48+00:00", "title": "Artifact: softmaxlin-repro/repro-bundle:v0", "artifact": "softmaxlin-repro/repro-bundle:v0", "artifact_type": "dataset"}
7
- -->
8
- **📦 Artifact** `softmaxlin-repro/repro-bundle:v0` · dataset · 3.7 MB
9
-
10
- https://huggingface.co/buckets/SabaPivot/repro-softmax-linear-attention-artifacts#softmaxlin-repro/repro-bundle:v0