Girinath11 commited on
Commit
bc543bc
Β·
verified Β·
1 Parent(s): c4ab5ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -28
README.md CHANGED
@@ -27,7 +27,7 @@ pipeline_tag: text-generation
27
 
28
  ---
29
 
30
- ## πŸ”₯ Novel Architecture: Mixture of Recursion (MoR)
31
 
32
  Traditional transformers apply the same depth to every input β€” simple and complex sentences both
33
  go through all N layers. This wastes compute on easy inputs and under-processes hard ones.
@@ -101,16 +101,16 @@ GPT-2 Medium evaluated on identical samples for fair comparison:
101
  ### What the Results Actually Show
102
 
103
  ```
104
- βœ… Architecture works β€” training is stable, loss decreases
105
- βœ… Router works β€” 0 NaN batches, routing signals valid
106
- βœ… Novel contribution β€” perplexity-based self-supervised routing
107
- ⚠️ Generalization gap β€” val PPL 15.37 vs test PPL 43.21
108
- ⚠️ More data needed β€” 150K samples is small for this model size
109
  ```
110
 
111
  ---
112
 
113
- ## πŸš€ Quick Start
114
 
115
  ```bash
116
  pip install transformers torch
@@ -166,7 +166,7 @@ print(chat("Explain neural networks simply"))
166
 
167
  ---
168
 
169
- ## πŸ“š Training Details
170
 
171
  ### Dataset β€” 150K Conversational Samples
172
 
@@ -203,7 +203,7 @@ Loss:
203
 
204
  ---
205
 
206
- ## πŸ’‘ Technical Innovation
207
 
208
  ### Self-Supervised Perplexity Routing
209
 
@@ -230,21 +230,21 @@ This model is my **first from-scratch LLM** β€” I learned a lot:
230
 
231
  ```
232
  What worked:
233
- βœ… NaN-safe attention masking (-1e4)
234
- βœ… Self-supervised routing signal
235
- βœ… Stable FP16 training (0 NaN batches)
236
- βœ… Perplexity drops meaningfully across epochs
237
 
238
  What needs improvement:
239
- ⚠️ More diverse training data (150K is small)
240
- ⚠️ Longer training (2 epochs not enough)
241
- ⚠️ Stronger regularization to prevent overfitting
242
- ⚠️ Better evaluation on neutral benchmarks from day 1
243
  ```
244
 
245
  ---
246
 
247
- ## ⚠️ Limitations
248
 
249
  - **Context window**: 512 tokens max
250
  - **English only**: Not suitable for other languages
@@ -255,20 +255,20 @@ What needs improvement:
255
 
256
  ---
257
 
258
- ## 🎯 Intended Use
259
 
260
  | Use case | Suitable? |
261
  |---|---|
262
- | Research on adaptive computation | βœ… Yes |
263
- | Learning how LLMs are built from scratch | βœ… Yes |
264
- | Prototyping conversational AI | βœ… With caveats |
265
- | Production chatbots | ❌ No |
266
- | Medical / legal / financial advice | ❌ No |
267
- | Factual question answering | ❌ No |
268
 
269
  ---
270
 
271
- ## πŸ“„ Citation
272
 
273
  ```bibtex
274
  @misc{girinath2026mor,
@@ -285,7 +285,7 @@ What needs improvement:
285
 
286
  ---
287
 
288
- ## πŸ™ Acknowledgments
289
 
290
  - Anthropic β€” HH-RLHF dataset
291
  - Tsinghua University β€” UltraChat dataset
@@ -295,6 +295,6 @@ What needs improvement:
295
 
296
  ---
297
 
298
- **Model status** : βœ… Research / Educational use
299
  **Last updated** : March 2026
300
  **License** : MIT
 
27
 
28
  ---
29
 
30
+ ## Novel Architecture: Mixture of Recursion (MoR)
31
 
32
  Traditional transformers apply the same depth to every input β€” simple and complex sentences both
33
  go through all N layers. This wastes compute on easy inputs and under-processes hard ones.
 
101
  ### What the Results Actually Show
102
 
103
  ```
104
+ Architecture works β€” training is stable, loss decreases
105
+ Router works β€” 0 NaN batches, routing signals valid
106
+ Novel contribution β€” perplexity-based self-supervised routing
107
+ Generalization gap β€” val PPL 15.37 vs test PPL 43.21
108
+ More data needed β€” 150K samples is small for this model size
109
  ```
110
 
111
  ---
112
 
113
+ ## Quick Start
114
 
115
  ```bash
116
  pip install transformers torch
 
166
 
167
  ---
168
 
169
+ ## Training Details
170
 
171
  ### Dataset β€” 150K Conversational Samples
172
 
 
203
 
204
  ---
205
 
206
+ ## Technical Innovation
207
 
208
  ### Self-Supervised Perplexity Routing
209
 
 
230
 
231
  ```
232
  What worked:
233
+ NaN-safe attention masking (-1e4)
234
+ Self-supervised routing signal
235
+ Stable FP16 training (0 NaN batches)
236
+ Perplexity drops meaningfully across epochs
237
 
238
  What needs improvement:
239
+ More diverse training data (150K is small)
240
+ Longer training (2 epochs not enough)
241
+ Stronger regularization to prevent overfitting
242
+ Better evaluation on neutral benchmarks from day 1
243
  ```
244
 
245
  ---
246
 
247
+ ## Limitations
248
 
249
  - **Context window**: 512 tokens max
250
  - **English only**: Not suitable for other languages
 
255
 
256
  ---
257
 
258
+ ## Intended Use
259
 
260
  | Use case | Suitable? |
261
  |---|---|
262
+ | Research on adaptive computation | Yes |
263
+ | Learning how LLMs are built from scratch | Yes |
264
+ | Prototyping conversational AI | With caveats |
265
+ | Production chatbots | No |
266
+ | Medical / legal / financial advice | No |
267
+ | Factual question answering | No |
268
 
269
  ---
270
 
271
+ ## Citation
272
 
273
  ```bibtex
274
  @misc{girinath2026mor,
 
285
 
286
  ---
287
 
288
+ ## Acknowledgments
289
 
290
  - Anthropic β€” HH-RLHF dataset
291
  - Tsinghua University β€” UltraChat dataset
 
295
 
296
  ---
297
 
298
+ **Model status** : Research / Educational use
299
  **Last updated** : March 2026
300
  **License** : MIT