pith:6QDFC2Y4
Training Infinitely Deep and Wide Transformers
Gradient flow in the conditional Wasserstein metric converges to global minima for infinitely deep and wide transformers when the initial loss is small enough and the attention NTK is injective.
arxiv:2605.17660 v1 · 2026-05-17 · math.OC · cs.AI · cs.LG · stat.ML
Add to your LaTeX paper
\usepackage{pith}
\pithnumber{6QDFC2Y4U7PG646RJLUYHTFPOH}
Prints a linked badge after your title and injects PDF metadata. Compiles on arXiv. Learn more · Embed verified badge
Record completeness
Claims
Under the NTK injectivity assumption (linear independence of log-sum-exp functions modulo affine functions), gradient flow in the conditional Wasserstein metric converges to global minima whenever the initial loss is sufficiently small.
The mean-field limit and the NTK injectivity condition remain valid approximations for the finite but large transformers that are actually trained; this premise enters when the convergence theorem is stated after the injectivity characterization.
Develops a mean-field neural PDE model for transformer training, proves forward-pass well-posedness via function-space ODEs, derives conditional Wasserstein gradients, and shows global convergence of gradient flow under an NTK injectivity condition equivalent to linear independence of log-sum-exp fu
References
Formal links
Receipt and verification
| First computed | 2026-05-20T00:04:51.429856Z |
|---|---|
| Builder | pith-number-builder-2026-05-17-v1 |
| Signature | Pith Ed25519
(pith-v1-2026-05) · public key |
| Schema | pith-number/v1.0 |
Canonical hash
f406516b1ca7de6f73d14ae983ccaf71ffa71e1ba8407e18af9cc750415bf815
Aliases
· · · · ·Agent API
Verify this Pith Number yourself
curl -sH 'Accept: application/ld+json' https://pith.science/pith/6QDFC2Y4U7PG646RJLUYHTFPOH \
| jq -c '.canonical_record' \
| python3 -c "import sys,json,hashlib; b=json.dumps(json.loads(sys.stdin.read()), sort_keys=True, separators=(',',':'), ensure_ascii=False).encode(); print(hashlib.sha256(b).hexdigest())"
# expect: f406516b1ca7de6f73d14ae983ccaf71ffa71e1ba8407e18af9cc750415bf815
Canonical record JSON
{
"metadata": {
"abstract_canon_sha256": "62f80cc2f7e1bf01bdc441db144ff1c58a541f85b00010ed393581da69da6f36",
"cross_cats_sorted": [
"cs.AI",
"cs.LG",
"stat.ML"
],
"license": "http://creativecommons.org/licenses/by/4.0/",
"primary_cat": "math.OC",
"submitted_at": "2026-05-17T21:30:13Z",
"title_canon_sha256": "8e2ad1d8fba0ed6441fbfa822016d407e90e873fd5e608931be87cddd501d7fe"
},
"schema_version": "1.0",
"source": {
"id": "2605.17660",
"kind": "arxiv",
"version": 1
}
}