Expose five anchored Softmax-as-linear-attention claims
Browse files- logbook.json +9 -9
- pages/executive-summary/page.md +28 -0
- pages/executive-summary/poster_embed.html +1 -0
- pages/index.md +2 -2
- pages/softmaxlin-repro/page.md +0 -10
logbook.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"schema_version": 1,
|
| 3 |
-
"title": "
|
| 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": "
|
| 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 |
-
#
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|