Feature Extraction
Transformers
TensorBoard
Safetensors
English
captionbert_v2
sentence-similarity
consensus-distillation
geometric-deep-learning
amoe
custom_code
Instructions to use AbstractPhil/captionbert-8192-v2-b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use AbstractPhil/captionbert-8192-v2-b with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="AbstractPhil/captionbert-8192-v2-b", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("AbstractPhil/captionbert-8192-v2-b", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
references+attribution section, inline citations
Browse files- deep-arm/TECHNICAL.md +55 -8
deep-arm/TECHNICAL.md
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
# Deep Arm β Depth Extension for captionbert-8192-v2-B
|
| 2 |
|
| 3 |
**A technical companion to the week-in-review article. Ships as `deep-arm/TECHNICAL.md` inside the
|
| 4 |
-
`AbstractPhil/captionbert-8192-v2-B` repository.**
|
| 5 |
|
| 6 |
This document describes **deep-1**: a ~5M-parameter depth-extension module ("deep arm") that
|
| 7 |
extends the encoder's usable conditioning depth from 256 tokens to 2048+ while leaving the base
|
|
@@ -26,7 +26,8 @@ The deep arm is a **switchable long-prompt mode** layered over that frozen base:
|
|
| 26 |
|
| 27 |
The arm adds ~5M trainable parameters (~4.06M re-initialized deep position rows + ~0.92M gated
|
| 28 |
relay adapters) against a frozen trunk, trained in 8k steps overnight on a single consumer GPU by
|
| 29 |
-
distilling token-state behavior from
|
|
|
|
| 30 |
bridge. Its preregistered depth gate passed, and a labeled harder secondary passed as well.
|
| 31 |
|
| 32 |
---
|
|
@@ -44,8 +45,8 @@ models), the encoder's state-similarity curve looks healthy at depth: a degraded
|
|
| 44 |
plateau around **.77** with no cliff. Token recoverability also holds up, at **.61β.66 through
|
| 45 |
depth 1024** β the strongest of the models tested. But the semantic gauge tells a different
|
| 46 |
story: **attribute binding collapses to roughly chance (~1.0) near token depth 480 and never
|
| 47 |
-
returns.** For reference on the same instrument, t5-base holds binding 1.9β3.0 through 2048
|
| 48 |
-
(relative position encodings are the real long-context contribution there), and bert-base reads
|
| 49 |
3.6 within its native 512.
|
| 50 |
|
| 51 |
*The binding gauge, briefly:* a contrast score read from the token states at the two noun
|
|
@@ -93,7 +94,7 @@ Seven load-bearing elements:
|
|
| 93 |
around each frozen block, sigma-gated.
|
| 94 |
4. **Longformer targets.** Teacher token states from Longformer-base-4096 on long documents,
|
| 95 |
span-resampled onto the encoder's token grid, then mapped 768β512.
|
| 96 |
-
5. **Cross-dimensional Procrustes bridge.** A whitened-Procrustes map into the encoder's real
|
| 97 |
512-d token frame, fit at **cos .5013 / retrieval R@1 .849**, with a paired shuffled-null
|
| 98 |
control at floor. This exceeds the earlier 768-d proxy-frame fit (.35). Stage 0 aborts by
|
| 99 |
construction if the fit lands below cos .20.
|
|
@@ -176,7 +177,8 @@ a single scoreboard number would erase the mechanism:
|
|
| 176 |
its way from .72 β 3.80; depth 2288 crosses 1.5 only in the final 500 of 8k steps.
|
| 177 |
- **Relay adapters (~0.92M params):** retroactive-binding amplification wherever gradient
|
| 178 |
reached β bind1 moved first at every depth. The adapter gates finished **.35 β .51, opening
|
| 179 |
-
monotonically with depth** β the third independent architecture in
|
|
|
|
| 180 |
the depth-gradient regularity (its third substrate).
|
| 181 |
- **Longformer supervision through the .5013-cos bridge:** shallow behavior **improved** under
|
| 182 |
it β depth-10 binding 1.59/2.01 β 2.69/2.27. Depth was bought without taxing the trained
|
|
@@ -210,8 +212,8 @@ it off for short ones. The capability trade is a mode switch, not a regression.
|
|
| 210 |
| V2 | Joint-objective revision | Named, unbuilt | Distillation plus a shallow-capability anchor, to close the .5655 engaged-mode gap; the refinement path recorded on the model card |
|
| 211 |
| V3 | Token-fidelity masked-span arm | Named, unbuilt | A span-reconstruction objective on the existing trunks β the targeted fix for the encoder's binding gap and the conditioning line's shelf upgrade |
|
| 212 |
| V4 | Interpolation-init deep rows | Named, unbuilt | Interpolation initialization as an ablation against tiling (which is free and strong) |
|
| 213 |
-
| V5 | Anchored-trunk deep arm | Named, unbuilt β queue head | The V1 recipe applied to the AlephLM-0 anchored trunks β the variant that extends that family's inference depth. Two free reads: whether signed dispatch shapes depth repair differently than a dense trunk, and whether the a3 trunk's early-grid diversity advantage survives depth extension. Requires normalized-gauge grid reads and context layer-norm in any consumer |
|
| 214 |
-
| V6 | RoPE-native next trunk | Design-level | A related hybrid with a RoPE trunk measures ~.98 positional integrity, flat to depth 8000 β evidence that rotary positions remove the position-row problem at the root for the *next* pretrain, rather than retrofitting this one |
|
| 215 |
|
| 216 |
**Queue by leverage:** V5 β V2 β V3 β V6. All of V2/V3/V5 are single-consumer-GPU overnight runs
|
| 217 |
at the deep arm's ~5M-parameter scale.
|
|
@@ -259,3 +261,48 @@ engaged-mode benchmark (`deep_arm_bench.json`), alongside the model-card section
|
|
| 259 |
construction, the before/after battery, the bridge fit, the honest cost, toggle semantics, and
|
| 260 |
usage. Every number in this document is inherited from the registered measurement record for
|
| 261 |
that window.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Deep Arm β Depth Extension for captionbert-8192-v2-B
|
| 2 |
|
| 3 |
**A technical companion to the week-in-review article. Ships as `deep-arm/TECHNICAL.md` inside the
|
| 4 |
+
[`AbstractPhil/captionbert-8192-v2-B`](https://huggingface.co/AbstractPhil/captionbert-8192-v2-B) repository.**
|
| 5 |
|
| 6 |
This document describes **deep-1**: a ~5M-parameter depth-extension module ("deep arm") that
|
| 7 |
extends the encoder's usable conditioning depth from 256 tokens to 2048+ while leaving the base
|
|
|
|
| 26 |
|
| 27 |
The arm adds ~5M trainable parameters (~4.06M re-initialized deep position rows + ~0.92M gated
|
| 28 |
relay adapters) against a frozen trunk, trained in 8k steps overnight on a single consumer GPU by
|
| 29 |
+
[distilling](https://arxiv.org/abs/1503.02531) token-state behavior from
|
| 30 |
+
[Longformer-base-4096](https://arxiv.org/abs/2004.05150) across a learned cross-dimensional
|
| 31 |
bridge. Its preregistered depth gate passed, and a labeled harder secondary passed as well.
|
| 32 |
|
| 33 |
---
|
|
|
|
| 45 |
plateau around **.77** with no cliff. Token recoverability also holds up, at **.61β.66 through
|
| 46 |
depth 1024** β the strongest of the models tested. But the semantic gauge tells a different
|
| 47 |
story: **attribute binding collapses to roughly chance (~1.0) near token depth 480 and never
|
| 48 |
+
returns.** For reference on the same instrument, [t5-base](https://arxiv.org/abs/1910.10683) holds binding 1.9β3.0 through 2048
|
| 49 |
+
(relative position encodings are the real long-context contribution there), and [bert-base](https://arxiv.org/abs/1810.04805) reads
|
| 50 |
3.6 within its native 512.
|
| 51 |
|
| 52 |
*The binding gauge, briefly:* a contrast score read from the token states at the two noun
|
|
|
|
| 94 |
around each frozen block, sigma-gated.
|
| 95 |
4. **Longformer targets.** Teacher token states from Longformer-base-4096 on long documents,
|
| 96 |
span-resampled onto the encoder's token grid, then mapped 768β512.
|
| 97 |
+
5. **Cross-dimensional [Procrustes](https://doi.org/10.1007/BF02291478) bridge.** A whitened-Procrustes map into the encoder's real
|
| 98 |
512-d token frame, fit at **cos .5013 / retrieval R@1 .849**, with a paired shuffled-null
|
| 99 |
control at floor. This exceeds the earlier 768-d proxy-frame fit (.35). Stage 0 aborts by
|
| 100 |
construction if the fit lands below cos .20.
|
|
|
|
| 177 |
its way from .72 β 3.80; depth 2288 crosses 1.5 only in the final 500 of 8k steps.
|
| 178 |
- **Relay adapters (~0.92M params):** retroactive-binding amplification wherever gradient
|
| 179 |
reached β bind1 moved first at every depth. The adapter gates finished **.35 β .51, opening
|
| 180 |
+
monotonically with depth** β the third independent architecture in
|
| 181 |
+
[this research line](https://huggingface.co/blog/AbstractPhil/geometric-memory-ft4) to show
|
| 182 |
the depth-gradient regularity (its third substrate).
|
| 183 |
- **Longformer supervision through the .5013-cos bridge:** shallow behavior **improved** under
|
| 184 |
it β depth-10 binding 1.59/2.01 β 2.69/2.27. Depth was bought without taxing the trained
|
|
|
|
| 212 |
| V2 | Joint-objective revision | Named, unbuilt | Distillation plus a shallow-capability anchor, to close the .5655 engaged-mode gap; the refinement path recorded on the model card |
|
| 213 |
| V3 | Token-fidelity masked-span arm | Named, unbuilt | A span-reconstruction objective on the existing trunks β the targeted fix for the encoder's binding gap and the conditioning line's shelf upgrade |
|
| 214 |
| V4 | Interpolation-init deep rows | Named, unbuilt | Interpolation initialization as an ablation against tiling (which is free and strong) |
|
| 215 |
+
| V5 | Anchored-trunk deep arm | Named, unbuilt β queue head | The V1 recipe applied to the [AlephLM-0](https://huggingface.co/AbstractPhil/alephlm-0) anchored trunks β the variant that extends that family's inference depth. Two free reads: whether signed dispatch shapes depth repair differently than a dense trunk, and whether the a3 trunk's early-grid diversity advantage survives depth extension. Requires normalized-gauge grid reads and context layer-norm in any consumer |
|
| 216 |
+
| V6 | [RoPE](https://arxiv.org/abs/2104.09864)-native next trunk | Design-level | A related hybrid with a RoPE trunk measures ~.98 positional integrity, flat to depth 8000 β evidence that rotary positions remove the position-row problem at the root for the *next* pretrain, rather than retrofitting this one |
|
| 217 |
|
| 218 |
**Queue by leverage:** V5 β V2 β V3 β V6. All of V2/V3/V5 are single-consumer-GPU overnight runs
|
| 219 |
at the deep arm's ~5M-parameter scale.
|
|
|
|
| 261 |
construction, the before/after battery, the bridge fit, the honest cost, toggle semantics, and
|
| 262 |
usage. Every number in this document is inherited from the registered measurement record for
|
| 263 |
that window.
|
| 264 |
+
|
| 265 |
+
---
|
| 266 |
+
|
| 267 |
+
## References and attribution
|
| 268 |
+
|
| 269 |
+
**Models**
|
| 270 |
+
|
| 271 |
+
- Longformer β Beltagy et al., [arXiv:2004.05150](https://arxiv.org/abs/2004.05150). The
|
| 272 |
+
distillation teacher (longformer-base-4096), source of the deep token-state targets.
|
| 273 |
+
- T5 β Raffel et al., [arXiv:1910.10683](https://arxiv.org/abs/1910.10683). Depth-battery
|
| 274 |
+
comparator (t5-base), whose relative position encodings hold binding through 2048.
|
| 275 |
+
- BERT β Devlin et al., [arXiv:1810.04805](https://arxiv.org/abs/1810.04805). Depth-battery
|
| 276 |
+
comparator (bert-base), read within its native 512.
|
| 277 |
+
- CLIP β Radford et al., [arXiv:2103.00020](https://arxiv.org/abs/2103.00020). Lineage of the
|
| 278 |
+
conditioning-consumer framing: downstream consumers read per-token encoder states through
|
| 279 |
+
cross-attention rather than a pooled vector.
|
| 280 |
+
|
| 281 |
+
**Data**
|
| 282 |
+
|
| 283 |
+
- MTEB β [arXiv:2210.07316](https://arxiv.org/abs/2210.07316). Source of the suites behind the
|
| 284 |
+
certified 8-task short-input benchmark used for the engaged-mode cost.
|
| 285 |
+
|
| 286 |
+
**Methods**
|
| 287 |
+
|
| 288 |
+
- Procrustes analysis β SchΓΆnemann (1966), [doi:10.1007/BF02291478](https://doi.org/10.1007/BF02291478).
|
| 289 |
+
The cross-dimensional bridge: a whitened-Procrustes map from the teacher's 768-d frame into the
|
| 290 |
+
encoder's 512-d token frame.
|
| 291 |
+
- Knowledge distillation β Hinton et al., [arXiv:1503.02531](https://arxiv.org/abs/1503.02531).
|
| 292 |
+
The training scheme's lineage; here, per-token state distillation from the teacher across the
|
| 293 |
+
bridge.
|
| 294 |
+
- LoRA β Hu et al., [arXiv:2106.09685](https://arxiv.org/abs/2106.09685). Lineage of the small
|
| 295 |
+
gated relay adapters trained over the frozen trunk.
|
| 296 |
+
- RoPE β Su et al., [arXiv:2104.09864](https://arxiv.org/abs/2104.09864). Rotary position
|
| 297 |
+
embedding; the V6 design direction that removes the position-row problem at the root.
|
| 298 |
+
|
| 299 |
+
**Infrastructure**
|
| 300 |
+
|
| 301 |
+
- PyTorch β [pytorch.org](https://pytorch.org). Training and evaluation stack for the arm and
|
| 302 |
+
its batteries.
|
| 303 |
+
|
| 304 |
+
**Related artifacts:** the trunk this arm extends,
|
| 305 |
+
[AbstractPhil/captionbert-8192-v2-B](https://huggingface.co/AbstractPhil/captionbert-8192-v2-B)
|
| 306 |
+
(this repository; the arm ships under `deep-arm/`) Β· the anchored trunks named as the V5 target,
|
| 307 |
+
[AbstractPhil/alephlm-0](https://huggingface.co/AbstractPhil/alephlm-0) Β· the prior installment
|
| 308 |
+
of this series, [geometric-memory-ft4](https://huggingface.co/blog/AbstractPhil/geometric-memory-ft4).
|