remg1997 commited on
Commit
5586743
·
verified ·
1 Parent(s): 4bd23a4

Add model card (license + description) for leaderboard submission

Browse files
Files changed (1) hide show
  1. README.md +68 -0
README.md ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - babylm
5
+ - babylm-2026
6
+ - modernbert
7
+ - masked-language-model
8
+ - strict-small
9
+ language:
10
+ - en
11
+ library_name: transformers
12
+ pipeline_tag: fill-mask
13
+ ---
14
+
15
+ # ModernBERT-Small (Factorized Linear Embeddings) — BabyLM 2026 Strict-Small
16
+
17
+ This model is a `ModernBERT-Small` masked language model (RoPE, GeGLU, alternating local/global
18
+ attention, 384 hidden size, 16 layers, 6 attention heads) trained from scratch on the
19
+ **BabyLM 2026 Strict-Small** 10M-word corpus, as part of an ablation study on
20
+ parameter-efficient token embedding layers for developmentally-plausible pretraining under the
21
+ BabyLM Challenge's strict-small compute and data budget.
22
+
23
+ ## Embedding design
24
+
25
+ Standard transformer token embedding tables scale as `vocab_size x hidden_size`, which for this
26
+ model's 30,522-token vocabulary and 384 hidden size would be an 11.7M-parameter dense lookup
27
+ table -- a large fraction of the model's total parameter budget under the strict-small
28
+ constraint. This checkpoint instead uses an **ALBERT-style linear factorization**: tokens are
29
+ first embedded into a smaller 128-dimensional bottleneck space, then linearly projected up to the
30
+ model's 384-dimensional hidden size, replacing one large `vocab_size x hidden_size` matrix with
31
+ two much smaller ones (`vocab_size x 128` and `128 x 384`). The output projection (`tie_word_
32
+ embeddings=false`) uses its own separate decoder rather than sharing the input embedding weights.
33
+
34
+ This is one variant in a broader comparison of embedding-layer parameterizations (dense, linear
35
+ and MLP-style factorization, tensor-train decomposition, deterministic Fourier expansion, and
36
+ compositional/frequency-adaptive variants) evaluated under identical data, tokenizer, optimizer,
37
+ and training budget, to isolate the effect of the embedding layer's parameterization on
38
+ downstream BabyLM evaluation performance.
39
+
40
+ ## Training data
41
+
42
+ BabyLM 2026 Strict-Small corpus (~10M words), tokenized with a byte-level BPE tokenizer trained
43
+ on the same corpus (vocab size 30,522). No external data, synthetic augmentation, or human
44
+ annotation beyond the corpus as officially released.
45
+
46
+ ## Usage
47
+
48
+ ```python
49
+ from transformers import AutoModelForMaskedLM, AutoTokenizer
50
+
51
+ model = AutoModelForMaskedLM.from_pretrained(
52
+ "remg1997/modernbert-small-modernbert-small-factorized-linear-babylm2026",
53
+ trust_remote_code=True,
54
+ )
55
+ tokenizer = AutoTokenizer.from_pretrained(
56
+ "remg1997/modernbert-small-modernbert-small-factorized-linear-babylm2026"
57
+ )
58
+ ```
59
+
60
+ Each `chck_{N}M` branch of this repository corresponds to a BabyLM-Challenge compliance
61
+ checkpoint (one per `N` million words of training data seen); `main` points at the final,
62
+ fully-trained checkpoint.
63
+
64
+ ## Evaluation
65
+
66
+ Evaluated with the official [`babylm-eval`](https://github.com/babylm-org/babylm-eval) harness
67
+ (BLiMP, EWoK, entity tracking, COMPS, Global PIQA, reading-time correlation, GLUE/SuperGLUE
68
+ fine-tuning, and Age-of-Acquisition word-surprisal correlation) under the strict-small track.