KitsuVp commited on
Commit
f61e925
·
verified ·
1 Parent(s): c172d52

Model save

Browse files
Files changed (4) hide show
  1. README.md +72 -306
  2. config.json +11 -1
  3. model.safetensors +2 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -1,312 +1,78 @@
1
  ---
2
- language: en
3
- license: apache-2.0
4
  tags:
5
- - causal-lm
6
- - research
7
- - fp8
8
- - attention
9
- - normalization
10
- - neollm
11
- - pace
12
- datasets:
13
- - HuggingFaceFW/fineweb-edu
14
  ---
15
 
16
- # NeoLLM
17
-
18
- NeoLLM is a **135 M parameter** decoder-only language model trained from scratch on
19
- [FineWeb-Edu](https://huggingface.co/datasets/HuggingFaceFW/fineweb-edu) in **FP8**
20
- precision, completing training in approximately **6 hours** on a single NVIDIA RTX 5090.
21
- It integrates a collection of recently published attention and normalization techniques
22
- into a single architecture, with the goal of studying how they interact during
23
- pretraining. The model is actively being developed and the current checkpoint represents
24
- an intermediate training state.
25
-
26
- > **Author / contact:** [@Kyokopom](https://x.com/Kyokopom) on X
27
- > **Repository:** [KitsuVp/NeoLLM](https://huggingface.co/KitsuVp/NeoLLM)
28
-
29
- ---
30
-
31
- ## Architecture
32
-
33
- NeoLLM is a decoder-only transformer with the following configuration:
34
-
35
- | Parameter | Value |
36
- |---|---|
37
- | Hidden size | 512 |
38
- | Layers | 12 |
39
- | Attention heads | 8 |
40
- | KV heads (GQA) | 4 |
41
- | Head dim | 64 |
42
- | Intermediate size | 1536 |
43
- | Vocabulary | Qwen3 tokenizer (64,402 tokens) |
44
- | Context length | 512 tokens |
45
-
46
- ### Parameter breakdown
47
-
48
- | Parameter bucket | Count |
49
- |---|---|
50
- | **Total parameters** | 118.31M (118,313,336) |
51
- | **Embedding parameters** (tied) | 32.97M (32,973,824) |
52
- | **Non-embedding parameters** | 85.34M (85,339,512) |
53
- | **Effective trainable parameters** | 118.31M (118,313,336) |
54
-
55
- > Weight tying is **enabled**: the input embedding matrix and the language-model head
56
- > share the same parameters, so the effective trainable budget is
57
- > `total − embed = 85.34M`.
58
-
59
- ### Integrated techniques
60
-
61
- NeoLLM combines architecture modules, optional auxiliary objectives, and
62
- training-time optimizer/stability components from the following papers.
63
-
64
- **Embedding and token representation**
65
-
66
- - **Learnable Multipliers** ([arXiv:2601.04890](https://arxiv.org/abs/2601.04890)) — Adds
67
- per-row and per-column learnable scalar parameters to selected matrix layers and, when
68
- enabled, embeddings.
69
- - **Leviathan** ([arXiv:2601.22040](https://arxiv.org/abs/2601.22040)) — Optional
70
- continuous token embedding generator that can replace the discrete input lookup table.
71
- - **KHRONOS** ([arXiv:2505.13315](https://arxiv.org/abs/2505.13315)) — Kernel/basis
72
- reference used by the Leviathan continuous token generator implementation.
73
- - **JTok / JTok-M** ([arXiv:2602.00800](https://arxiv.org/abs/2602.00800)) — Optional
74
- token-indexed self-modulation surfaces over Leviathan coordinates.
75
- - **Spelling Bee Embeddings** ([arXiv:2601.18030](https://arxiv.org/abs/2601.18030)) —
76
- Augments token embeddings with character-level spelling information.
77
- - **Token Embedding Manifold analysis** ([arXiv:2504.01002](https://arxiv.org/abs/2504.01002)) —
78
- Reference motivation for treating token embeddings as structured objects rather than
79
- unconstrained lookup rows.
80
-
81
- **Attention, positions, and output projection**
82
-
83
- - **FAN** ([arXiv:2502.21309](https://arxiv.org/abs/2502.21309)) — Fourier Analysis Networks.
84
- A portion of the projection channels are dedicated to periodic cosine/sine features.
85
- - **MEA** ([arXiv:2601.19611](https://arxiv.org/abs/2601.19611)) — Explicit Multi-head
86
- Attention. Adds small learnable interaction matrices between attention heads for K and V.
87
- - **LUCID** ([arXiv:2602.10410](https://arxiv.org/abs/2602.10410)) — Applies a learned
88
- lower-triangular preconditioner to V before attention, decorrelating value representations
89
- across positions.
90
- - **Affine-Scaled Attention** ([arXiv:2602.23057](https://arxiv.org/abs/2602.23057)) — Adds
91
- two learnable per-head scalars (α and β) to the softmax weights:
92
- `[α·softmax(QKᵀ) + β]·V`.
93
- - **XSA** ([arXiv:2603.09078](https://arxiv.org/abs/2603.09078)) — Exclusive Self Attention.
94
- After computing attention, removes the component of the output aligned with the token's
95
- own value vector.
96
- - **Directional Routing** ([arXiv:2603.14923](https://arxiv.org/abs/2603.14923)) — Each head
97
- learns K=4 directions in the output space; a learned router suppresses the attention output
98
- along each direction per input.
99
- - **Gated Attention** ([arXiv:2505.06708](https://arxiv.org/abs/2505.06708)) — A sigmoid gate
100
- is applied to the attention output before the output projection, introducing non-linearity
101
- and preventing attention sinks.
102
- - **Momentum Attention** ([arXiv:2411.03884](https://arxiv.org/abs/2411.03884)) — Modifies Q
103
- and K by subtracting a fraction of the previous position's Q and K values (causal
104
- first-difference).
105
- - **Interleaved Head Attention / IHA** ([arXiv:2602.21371](https://arxiv.org/abs/2602.21371)) —
106
- Builds pseudo-heads from learned cross-head mixtures to create multiple attention patterns
107
- per original head.
108
- - **REPO** ([arXiv:2512.14391](https://arxiv.org/abs/2512.14391)) — Context re-positioning
109
- module that learns contextual position coordinates above a configurable start layer.
110
- - **GRAPE** ([arXiv:2512.07805](https://arxiv.org/abs/2512.07805)) — Group representational
111
- position encoding used by the REPO-GRAPE positional path.
112
- - **GOAT priors** ([arXiv:2601.15380](https://arxiv.org/abs/2601.15380)) — Optional
113
- factorized attention log-prior channels inspired by trainable attention priors.
114
- - **Hadamard output projection** ([arXiv:2603.08343](https://arxiv.org/abs/2603.08343)) —
115
- Replaces dense attention output projection with a structured Hadamard transform plus
116
- lightweight scaling.
117
-
118
- **Normalization, residual flow, and MLP**
119
-
120
- - **SeeDNorm** ([arXiv:2510.22777](https://arxiv.org/abs/2510.22777)) — Applied to Q and K
121
- projections. Dynamically rescales normalization from the input's own statistics.
122
- - **LayerNorm Scaling / LNS** ([arXiv:2502.05795](https://arxiv.org/abs/2502.05795)) — Each
123
- layer's output is scaled by 1/√ℓ where ℓ is the layer index.
124
- - **GPAS** ([arXiv:2506.22049](https://arxiv.org/abs/2506.22049)) — Gradient-Preserving
125
- Activation Scaling for residual junctions.
126
- - **PolyNorm** ([arXiv:2602.04902](https://arxiv.org/abs/2602.04902)) — Replaces the standard
127
- MLP activation with normalized linear, quadratic, and cubic branches.
128
- - **SimpleGPT** ([arXiv:2602.01212](https://arxiv.org/abs/2602.01212)) — Second-order
129
- geometry-inspired normalization strategy applied inside MLP projections.
130
- - **StackMemory / STACKTRANS** ([NeurIPS 2025](https://openreview.net/forum?id=2bbDg587uh)) —
131
- Optional differentiable hidden-state stack between decoder layers.
132
- - **Attention Residuals / AttnRes** ([arXiv:2603.15031](https://arxiv.org/abs/2603.15031)) —
133
- Optional learned depth-wise aggregation over previous layer outputs or block summaries.
134
- - **LAUREL** ([arXiv:2411.07501](https://arxiv.org/abs/2411.07501)) — Optional learned
135
- augmented residual layer with residual-weight and low-rank variants.
136
-
137
- **Training objectives and training-time regularizers**
138
-
139
- - **TWEO** ([arXiv:2511.23225](https://arxiv.org/abs/2511.23225)) — Optional
140
- Transformers Without Extreme Outliers activation regularizer for FP8/low-bit-friendly
141
- training.
142
- - **NITP** ([arXiv:2605.24956](https://arxiv.org/abs/2605.24956)) — Optional Next Implicit
143
- Token Prediction auxiliary objective using shallow-layer implicit token targets and a
144
- cosine loss.
145
- - **NextLat** ([arXiv:2511.05963](https://arxiv.org/abs/2511.05963)) — Optional next-latent
146
- prediction objective using latent dynamics, Smooth L1 supervision, and frozen-head KL.
147
-
148
- **Optimizer and training stability**
149
-
150
- - **Conda** ([arXiv:2509.24218](https://arxiv.org/abs/2509.24218)) —
151
- Column-Normalized Adam optimizer path used by the training script.
152
- - **Cautious Weight Decay** ([arXiv:2510.12402](https://arxiv.org/abs/2510.12402)) —
153
- Sign-selective weight decay variant used by the custom optimizer logic.
154
- - **Correction of Decoupled Weight Decay** ([arXiv:2512.08217](https://arxiv.org/abs/2512.08217)) —
155
- Adapts decoupled weight decay during learning-rate decay.
156
- - **AdamHD** ([arXiv:2511.14721](https://arxiv.org/abs/2511.14721)) —
157
- Decoupled Huber decay regularization reference used by the optimizer.
158
- - **GradientStabilizer** ([arXiv:2502.17055](https://arxiv.org/abs/2502.17055)) —
159
- Optional threshold-free gradient magnitude stabilizer.
160
- - **PACE** ([arXiv:2606.25086](https://arxiv.org/abs/2606.25086)) —
161
- Optional iterate-average controller that trains for the EMA model returned at evaluation
162
- and final serialization. The Conda-basis adaptation and its difference from AdamW are
163
- documented below.
164
-
165
- ---
166
-
167
- ### PACE integration and AdamW-reference differences
168
-
169
- PACE follows Au and Block's returned-model objective: the live weights are pulled toward a
170
- power-law EMA with a clipped per-coordinate gain, and evaluation/final serialization use that
171
- EMA estimator.
172
-
173
- - **Reference AdamW rule:** the gain uses AdamW's original-coordinate diagonal
174
- second moment, `eta * c * (1+t)^(-kappa) / (sqrt(v_hat) + eps)`.
175
- - **NeoLLM Conda rule (`mode=conda`):** for projected 2-D tensors, both the EMA
176
- displacement and `v_hat` are represented in Conda's cached SVD basis. The control is
177
- projected back after applying the diagonal gain. This is a deliberate change from AdamW
178
- required to avoid mixing incompatible coordinate systems.
179
- - **Optional exact AdamW pullback geometry (`mode=adamw`):** an additional
180
- original-coordinate second moment is maintained for projected matrices. The live optimizer
181
- step remains Conda.
182
- - **Conda scale:** in `mode=conda`, Conda's matrix-update scale multiplies the unsaturated
183
- gain automatically because it is part of the effective Conda preconditioner. AdamW has no
184
- corresponding scale.
185
- - **Fixed algorithm internals:** the EMA is stored in FP32, the gain is clipped at `1`, and PACE
186
- reuses each Conda group’s numerical epsilon. These are not exposed as independent switches.
187
- - **Minimal modes:** `use_pace=False` is plain Conda; `use_pace=True, c=0` is Conda+EMA;
188
- `use_pace=True, c>0` is complete PACE.
189
- - **Ordering:** PACE runs only after Conda, CWD/CHD, and weight-decay correction have fully
190
- updated the live weights.
191
- - **Disabled guarantee:** with `use_pace=False`, no PACE state is allocated and no existing
192
- Conda arithmetic or parameter update is changed.
193
- - **Checkpoint policy:** resumable internal checkpoints retain live weights and complete optimizer
194
- state, while evaluation and the final returned/Hub model always use the EMA when PACE is active.
195
 
196
- Current run: **disabled; no EMA state, auxiliary moment, or pullback is allocated**.
197
-
198
- ---
199
-
200
- ## Training
201
-
202
- | Setting | Value |
203
- |---|---|
204
- | Dataset | FineWeb-Edu (sample-10BT) |
205
- | Tokens seen | ~1.54B (46,875 steps × batch 64 × length 512) |
206
- | Precision | FP8 native (E4M3 weights/activations, E5M2 gradients) + BF16 fallback |
207
- | Optimizer | Conda (PACE disabled) |
208
- | PACE | disabled; no EMA state, auxiliary moment, or pullback is allocated |
209
- | Learning rate | 6e-04 with linear warmup (10 % of steps) |
210
- | Weight decay | 0.1 |
211
- | Training time | ~4h 51m |
212
- | Hardware | NVIDIA RTX 5090 (single GPU) |
213
-
214
- ### Training curve
215
-
216
- | Step | Train Loss | Val Loss |
217
- |---|---|---|
218
- | 5,000 | 4.827 | 4.716 |
219
- | 10,000 | 4.480 | 4.369 |
220
- | 15,000 | 4.334 | 4.227 |
221
- | 20,000 | 4.251 | 4.142 |
222
- | 25,000 | 4.195 | 4.091 |
223
- | 30,000 | 4.152 | 4.049 |
224
- | 35,000 | 4.127 | 4.019 |
225
- | 40,000 | 4.029 | 3.942 |
226
- | 45,000 | 3.984 | 3.885 |
227
- | 46,875 | — | 3.878 |
228
-
229
- ---
230
-
231
- ## Limitations
232
-
233
- - **Token budget** — ~1.5 B tokens seen; below estimated optimum. Knowledge-intensive tasks
234
- will improve with more training.
235
- - **Gradient spike at step 40k** — Reorganized the attention pattern in layer 9 that
236
- previously captured long-range token correlations. A checkpoint from ~step 38k is expected
237
- to have better aggregate benchmark scores.
238
- - **PolyNorm exclusivity** — The quadratic branch has become partially redundant with the
239
- linear branch. Will be corrected in the next training run.
240
- - **Base model only** — Not instruction-tuned or aligned; purely a next-token-prediction
241
- base model.
242
-
243
- ---
244
-
245
- ## References
246
-
247
- All papers whose techniques are integrated into NeoLLM's architecture,
248
- training objective, or training stack:
249
-
250
- | Area | Technique | Paper title | Reference |
251
- |---|---|---|---|
252
- | Embeddings | Learnable Multipliers | Freeing the Scale of Language Model Matrix Layers | [arXiv:2601.04890](https://arxiv.org/abs/2601.04890) |
253
- | Embeddings | Leviathan | A Separable Architecture for Continuous Token Representation in Language Models | [arXiv:2601.22040](https://arxiv.org/abs/2601.22040) |
254
- | Embeddings | KHRONOS | KHRONOS: a Kernel-Based Neural Architecture for Rapid, Resource-Efficient Scientific Computation | [arXiv:2505.13315](https://arxiv.org/abs/2505.13315) |
255
- | Embeddings | JTok / JTok-M | JTok: On Token Embedding as Another Axis of Scaling Law via Joint Token Self-Modulation | [arXiv:2602.00800](https://arxiv.org/abs/2602.00800) |
256
- | Embeddings | Spelling Bee | Spelling Bee Embeddings for Language Modeling | [arXiv:2601.18030](https://arxiv.org/abs/2601.18030) |
257
- | Embeddings | Token embedding analysis | Token Embeddings Violate the Manifold Hypothesis | [arXiv:2504.01002](https://arxiv.org/abs/2504.01002) |
258
- | Attention / positions | FAN | Fourier Analysis Networks | [arXiv:2502.21309](https://arxiv.org/abs/2502.21309) |
259
- | Attention / positions | MEA | Explicit Multi-head Attention for Inter-head Interaction in Large Language Models | [arXiv:2601.19611](https://arxiv.org/abs/2601.19611) |
260
- | Attention / positions | LUCID | Attention with Preconditioned Representations | [arXiv:2602.10410](https://arxiv.org/abs/2602.10410) |
261
- | Attention / positions | Affine-Scaled Attention | Affine-Scaled Attention: Towards Flexible and Stable Transformer Attention | [arXiv:2602.23057](https://arxiv.org/abs/2602.23057) |
262
- | Attention / positions | XSA | Exclusive Self Attention | [arXiv:2603.09078](https://arxiv.org/abs/2603.09078) |
263
- | Attention / positions | Directional Routing | Directional Routing in Transformers | [arXiv:2603.14923](https://arxiv.org/abs/2603.14923) |
264
- | Attention / positions | Gated Attention | Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free | [arXiv:2505.06708](https://arxiv.org/abs/2505.06708) |
265
- | Attention / positions | Momentum Attention | Momentum Attention | [arXiv:2411.03884](https://arxiv.org/abs/2411.03884) |
266
- | Attention / positions | IHA | Interleaved Head Attention | [arXiv:2602.21371](https://arxiv.org/abs/2602.21371) |
267
- | Attention / positions | REPO | Language Models with Context Re-Positioning | [arXiv:2512.14391](https://arxiv.org/abs/2512.14391) |
268
- | Attention / positions | GRAPE | Group Representational Position Encoding | [arXiv:2512.07805](https://arxiv.org/abs/2512.07805) |
269
- | Attention / positions | GOAT priors | You Need Better Attention Priors | [arXiv:2601.15380](https://arxiv.org/abs/2601.15380) |
270
- | Attention / positions | Hadamard o_proj | Rethinking Attention Output Projection: Structured Hadamard Transforms for Efficient Transformers | [arXiv:2603.08343](https://arxiv.org/abs/2603.08343) |
271
- | Residual / normalization | SeeDNorm | Self-Rescaled Dynamic Normalization | [arXiv:2510.22777](https://arxiv.org/abs/2510.22777) |
272
- | Residual / normalization | LNS | The Curse of Depth in LLMs | [arXiv:2502.05795](https://arxiv.org/abs/2502.05795) |
273
- | Residual / normalization | GPAS | Gradient-Preserving Activation Scaling | [arXiv:2506.22049](https://arxiv.org/abs/2506.22049) |
274
- | Residual / normalization | PolyNorm | PolyNorm / PolyCom | [arXiv:2602.04902](https://arxiv.org/abs/2602.04902) |
275
- | Residual / normalization | SimpleGPT | SimpleGPT | [arXiv:2602.01212](https://arxiv.org/abs/2602.01212) |
276
- | Residual / normalization | StackMemory / STACKTRANS | Recursive Transformer: Boosting Reasoning Ability with State Stack | [NeurIPS 2025](https://openreview.net/forum?id=2bbDg587uh) |
277
- | Residual / normalization | Attention Residuals | Attention Residuals | [arXiv:2603.15031](https://arxiv.org/abs/2603.15031) |
278
- | Residual / normalization | LAUREL | LAUREL: Learned Augmented Residual Layer | [arXiv:2411.07501](https://arxiv.org/abs/2411.07501) |
279
- | Objectives | TWEO | Transformers Without Extreme Outliers Enables FP8 Training And Quantization For Dummies | [arXiv:2511.23225](https://arxiv.org/abs/2511.23225) |
280
- | Objectives | NITP | Next Implicit Token Prediction for LLM Pre-training | [arXiv:2605.24956](https://arxiv.org/abs/2605.24956) |
281
- | Objectives | NextLat | Next-Latent Prediction Transformers Learn Compact World Models | [arXiv:2511.05963](https://arxiv.org/abs/2511.05963) |
282
- | Optimizer / training | Conda | Column-Normalized Adam for Training Large Language Models Faster | [arXiv:2509.24218](https://arxiv.org/abs/2509.24218) |
283
- | Optimizer / training | CWD | Cautious Weight Decay | [arXiv:2510.12402](https://arxiv.org/abs/2510.12402) |
284
- | Optimizer / training | WD correction | Correction of Decoupled Weight Decay | [arXiv:2512.08217](https://arxiv.org/abs/2512.08217) |
285
- | Optimizer / training | AdamHD | AdamHD: Decoupled Huber Decay Regularization for Language Model Pre-Training | [arXiv:2511.14721](https://arxiv.org/abs/2511.14721) |
286
- | Optimizer / training | GradientStabilizer | GradientStabilizer | [arXiv:2502.17055](https://arxiv.org/abs/2502.17055) |
287
- | Optimizer / training | PACE | Training for the Model You Return: Improving Optimization for Iterate-Averaged Language Models | [arXiv:2606.25086](https://arxiv.org/abs/2606.25086) |
288
-
289
- ---
290
-
291
- ## Citation
292
-
293
- ```bibtex
294
- @misc{neollm2026,
295
- title = {NeoLLM: A Research Language Model Integrating Recent Attention and Normalization Techniques},
296
- author = {KitsuVp},
297
- year = {2026},
298
- url = {https://huggingface.co/KitsuVp/NeoLLM}
299
- }
300
- ```
301
-
302
- ---
303
-
304
- ## Author
305
-
306
- [@Kyokopom](https://x.com/Kyokopom) on X
307
-
308
- ---
309
-
310
- ## License
311
 
312
- Apache 2.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ library_name: transformers
 
3
  tags:
4
+ - generated_from_trainer
5
+ model-index:
6
+ - name: NeoLLM
7
+ results: []
 
 
 
 
 
8
  ---
9
 
10
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
+ # NeoLLM
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
 
15
+ This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Loss: 4.4086
18
+ - Ntp Loss: 3.7068
19
+ - Ntp Ce Unweighted: 2.9908
20
+ - Mile Reweighting Delta: 0.7159
21
+ - Mu Loss: 0.0001
22
+ - Tweo Loss: 0.0240
23
+ - Nitp Loss: 0.4369
24
+ - Total Model Loss: 4.1439
25
+ - Optimizer Step: 46875.0
26
+ - Optimizer Metrics Due: 0.0
27
+ - Pace Step: 0.0
28
+ - Pace Update Due: 0.0
29
+ - Pace Previous Iterate Active: 1.0
30
+
31
+ ## Model description
32
+
33
+ More information needed
34
+
35
+ ## Intended uses & limitations
36
+
37
+ More information needed
38
+
39
+ ## Training and evaluation data
40
+
41
+ More information needed
42
+
43
+ ## Training procedure
44
+
45
+ ### Training hyperparameters
46
+
47
+ The following hyperparameters were used during training:
48
+ - learning_rate: 0.0006
49
+ - train_batch_size: 64
50
+ - eval_batch_size: 64
51
+ - seed: 42
52
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
53
+ - lr_scheduler_type: linear
54
+ - lr_scheduler_warmup_steps: 0.1
55
+ - num_epochs: 1
56
+
57
+ ### Training results
58
+
59
+ | Training Loss | Epoch | Step | Validation Loss | Loss | Ce Unweighted | Reweighting Delta | Model Loss | Metrics Due | Update Due | Previous Iterate Active |
60
+ |:-------------:|:------:|:----:|:---------------:|:------:|:-------------:|:-----------------:|:----------:|:-----------:|:----------:|:-----------------------:|
61
+ | 5.0825 | 0.1067 | 0.0 | 4.9996 | 0.4963 | 3.5936 | 0.6309 | 4.7210 | 0.0 | 0.0 | 1.0 |
62
+ | 4.8755 | 0.2133 | 0.0 | 4.7932 | 0.4740 | 3.3662 | 0.6736 | 4.5143 | 0.0 | 0.0 | 1.0 |
63
+ | 4.7762 | 0.32 | 0.0 | 4.6980 | 0.4716 | 3.2756 | 0.6823 | 4.4298 | 0.0 | 0.0 | 1.0 |
64
+ | 4.7103 | 0.4267 | 0.0 | 4.6257 | 0.4661 | 3.2067 | 0.6798 | 4.3528 | 0.0 | 0.0 | 1.0 |
65
+ | 4.6705 | 0.5333 | 0.0 | 4.5969 | 0.4587 | 3.1800 | 0.6922 | 4.3312 | 0.0 | 0.0 | 1.0 |
66
+ | 4.6362 | 0.64 | 0.0 | 4.5559 | 0.4544 | 3.1354 | 0.6976 | 4.2877 | 0.0 | 0.0 | 1.0 |
67
+ | 4.6174 | 0.7467 | 0.0 | 4.5421 | 0.4549 | 3.1175 | 0.7052 | 4.2779 | 0.0 | 0.0 | 1.0 |
68
+ | 4.5314 | 0.8533 | 0.0 | 4.4681 | 0.4435 | 3.0391 | 0.7076 | 4.1905 | 0.0 | 0.0 | 1.0 |
69
+ | 4.4885 | 0.96 | 0.0 | 4.4140 | 0.4384 | 2.9989 | 0.7130 | 4.1506 | 0.0 | 0.0 | 1.0 |
70
+ | 4.4811 | 1.0 | 0.0 | 4.4086 | 0.4369 | 2.9908 | 0.7159 | 4.1439 | 0.0 | 0.0 | 1.0 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 5.14.1
76
+ - Pytorch 2.13.0+cu132
77
+ - Datasets 5.0.1
78
+ - Tokenizers 0.22.2
config.json CHANGED
@@ -51,8 +51,15 @@
51
  "max_position_embeddings": 512,
52
  "mea_component_key_value_heads": 4,
53
  "mea_groupnorm_eps": 1e-06,
 
 
 
 
 
 
54
  "model_type": "neollm",
55
  "momentum_gamma": 0.1,
 
56
  "nextlat_dynamics_proj_factor": 1.0,
57
  "nextlat_horizon": 1,
58
  "nextlat_kl_weight": 1.0,
@@ -118,10 +125,13 @@
118
  "use_lns": true,
119
  "use_lucid_attention": false,
120
  "use_mea_attention": false,
 
 
121
  "use_momentum_attention": true,
 
122
  "use_nextlat": false,
123
  "use_nitp": true,
124
- "use_nitp_temporal": true,
125
  "use_repo": true,
126
  "use_repo_goat_prior": false,
127
  "use_repo_grape": true,
 
51
  "max_position_embeddings": 512,
52
  "mea_component_key_value_heads": 4,
53
  "mea_groupnorm_eps": 1e-06,
54
+ "meap_exclude_last": true,
55
+ "meap_implementation": "triton",
56
+ "meap_mask_ratio": 0.15,
57
+ "meap_mask_token_id": 0,
58
+ "meap_seed": 42,
59
+ "mile_loss_gamma": 1.0,
60
  "model_type": "neollm",
61
  "momentum_gamma": 0.1,
62
+ "mu_loss_lambda": 0.0001,
63
  "nextlat_dynamics_proj_factor": 1.0,
64
  "nextlat_horizon": 1,
65
  "nextlat_kl_weight": 1.0,
 
125
  "use_lns": true,
126
  "use_lucid_attention": false,
127
  "use_mea_attention": false,
128
+ "use_meap": false,
129
+ "use_mile_loss": true,
130
  "use_momentum_attention": true,
131
+ "use_mu_loss": true,
132
  "use_nextlat": false,
133
  "use_nitp": true,
134
+ "use_nitp_temporal": false,
135
  "use_repo": true,
136
  "use_repo_goat_prior": false,
137
  "use_repo_grape": true,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e2d79a82ec12f9a950502bacacb5343dc6a3f16209e741fa6213d3195d9a0c81
3
- size 244921048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1533978dc5d2187d6ba0372d5c644b233b014eae9af564699be1aa621ca666f7
3
+ size 240723968
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:107b5ecb3f440d82914084c62f8c4c6a8407623c8aa0df8031505051d633725e
3
  size 5329
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c5c1ae4a7c39a3f728422d56a8d5adcf601d3d1a08340ae329fed9801482677
3
  size 5329