Files
Test-1-3000/README.md

508 lines
11 KiB
Markdown
Raw Permalink Normal View History

---
license: mit
datasets:
- roneneldan/TinyStories
language:
- en
pipeline_tag: text-generation
tags:
- text-generation-inference
new_version: GODELEV/Test-1-4000
---
# Test-1-3000 — A 190M Parameter Narrative Intelligence Engine
<p align="center">
![Architecture](https://img.shields.io/badge/Architecture-Llama-blue)
![Parameters](https://img.shields.io/badge/Parameters-190M-green)
![Context](https://img.shields.io/badge/Context-2048-orange)
![Framework](https://img.shields.io/badge/Framework-PyTorch-red)
![Training](https://img.shields.io/badge/Training-Step_3000-purple)
</p>
---
# Overview
**Test-1-3000** is a compact yet remarkably capable decoder-only Transformer language model built upon the modern **Llama architecture**.
The project explores an important question in language model research:
> *How much narrative reasoning, coherence, and world understanding can emerge inside a small model when trained correctly?*
Despite containing only **190.55 million parameters**, Test-1-3000 demonstrates surprisingly advanced:
- Narrative continuity
- Character persistence
- Long-range memory consistency
- Emotional progression
- Logical event sequencing
- Contextual storytelling stability
The model was trained specifically for **short-form narrative intelligence**, focusing on coherent storytelling rather than broad internet-scale memorization.
Unlike many small models that generate fragmented or repetitive text, Test-1-3000 learns to maintain:
- causal relationships,
- stable story worlds,
- emotional trajectories,
- and meaningful resolutions across long contexts.
---
# Key Highlights
| Feature | Description |
|---|---|
| Architecture | Llama-based Decoder-only Transformer |
| Parameters | 190.55 Million |
| Context Length | 2048 Tokens |
| Final Training Step | 3000 |
| Final Training Loss | **0.8516** |
| Attention Optimization | Flash Attention 2 |
| Compilation | `torch.compile` |
| Precision | bfloat16 Mixed Precision |
| Positional Encoding | Rotary Positional Embeddings (RoPE) |
---
#What Makes Test-1-3000 Special?
Most compact language models struggle with:
- maintaining consistency,
- remembering earlier events,
- resolving story arcs,
- and avoiding repetition.
Test-1-3000 was trained with a different objective philosophy:
## Narrative Intelligence First
Instead of optimizing for broad factual memorization, the model focuses on:
- temporal continuity,
- event causality,
- emotional logic,
- and narrative closure.
This creates a surprisingly stable storytelling engine capable of generating coherent multi-paragraph narratives with strong thematic flow.
---
# Model Architecture
Test-1-3000 follows a modern efficient Transformer design optimized for both:
- training stability,
- and inference throughput.
The architecture borrows heavily from the proven Llama design philosophy while remaining lightweight enough for experimentation and rapid iteration.
---
# Technical Specifications
| Feature | Specification |
|---|---|
| Model Type | Decoder-only Transformer |
| Hidden Dimension | 768 |
| Layers (Depth) | 12 |
| Attention Heads | 12 |
| Intermediate Size | 3072 |
| Activation Function | SwiGLU |
| Normalization | RMSNorm |
| Vocabulary Size | 50,257 |
| Tokenizer | GPT-2 Tokenizer |
| Context Window | 2048 Tokens |
| Precision | bfloat16 |
| Attention Backend | Flash Attention 2 |
---
# Positional Understanding with RoPE
Test-1-3000 uses **Rotary Positional Embeddings (RoPE)** to maintain precise token relationship awareness throughout long contexts.
This allows the model to:
- track entities across paragraphs,
- preserve story continuity,
- maintain dialogue references,
- and understand long-range dependencies efficiently.
For a model of this scale, the 2048-token context window provides unusually strong narrative memory.
---
#The Evolution of Learning
Training Test-1-3000 revealed clear emergent phases of cognitive development.
The model did not merely memorize text patterns — it progressively developed increasingly sophisticated representations of narrative structure and world dynamics.
---
#The Lexical Phase
## *(Steps 0 → 250)*
At the beginning of training, the model learned the statistical foundations of language.
It discovered:
- common sentence structures,
- punctuation behavior,
- frequent vocabulary patterns,
- and story-opening syntax.
During this phase, phrases such as:
> "Once upon a time"
became strong narrative anchors.
The model began constructing basic grammatical fluency but still lacked deeper logical understanding.
### Characteristics
- High repetition
- Weak memory
- Poor event continuity
- Basic syntax acquisition
---
# The Relational Phase
## *(Steps 250 → 1000)*
The model started connecting concepts together into meaningful relationships.
It learned:
- object interactions,
- spatial reasoning,
- basic causality,
- and action consistency.
For example:
- parks imply trees and playing,
- rain implies umbrellas or wetness,
- sadness often precedes comfort or resolution.
The training loss rapidly decreased below **1.5**, signaling major improvements in structural reasoning.
### Emergent Behaviors
- Scene consistency
- Character-action alignment
- Basic emotional logic
- Improved descriptive continuity
---
# The Coherence Phase
## *(Steps 1000 → 2000)*
This phase marked the emergence of true narrative stabilization.
The model learned:
- story pacing,
- setup/payoff relationships,
- conflict resolution,
- and multi-sentence thematic continuity.
Stories no longer collapsed into unrelated fragments.
Instead, the model began maintaining:
- stable goals,
- emotional arcs,
- and logical conclusions.
If a story introduced a problem:
> "Lily was lonely."
the model increasingly learned to produce meaningful emotional resolutions later in the text.
### Major Improvements
- Long-range memory
- Reduced contradiction
- Better endings
- Stronger narrative flow
- Lower hallucination frequency
Final loss at this stage:
| Step | Loss |
|---|---|
| 2000 | **1.27** |
---
# The Emergent Narrative Intelligence Phase
## *(Steps 2000 → 3000)*
This final stage represented a major leap in generative sophistication.
Rather than simply maintaining coherence, the model began exhibiting signs of:
- implicit world modeling,
- narrative anticipation,
- emotional persistence,
- and latent planning behavior.
The model increasingly understood that stories possess:
- momentum,
- consequences,
- emotional gravity,
- and thematic closure.
Characters began behaving more consistently across long contexts.
Events earlier in stories influenced future generations more reliably.
The model also became significantly better at:
- avoiding repetitive loops,
- maintaining tone,
- preserving narrative identity,
- and generating cleaner transitions between scenes.
### Emergent Capabilities
- Multi-event causal chaining
- Persistent emotional tone
- Improved dialogue continuity
- Better conflict resolution
- Reduced topic drift
- More natural pacing
- Stronger thematic stability
Most importantly:
> The model began generating stories that feel intentionally written rather than statistically assembled.
---
#Final Training Statistics
| Metric | Value |
|---|---|
| Final Step | 3000 |
| Final Loss | **0.8516** |
| Training Stability | Excellent |
| Gradient Behavior | Stable |
| Divergence Events | None Observed |
---
# Training Configuration
## Hyperparameters
| Parameter | Value |
|---|---|
| Optimizer | AdamW |
| Betas | β₁=0.9, β₂=0.95 |
| Learning Rate | 5e-4 |
| Scheduler | OneCycleLR |
| Weight Decay | 0.01 |
| Precision | bfloat16 |
| Compilation | torch.compile |
| Attention Optimization | Flash Attention 2 |
| Effective Batch Size | ~262,144 Tokens / Step |
---
# Dataset
## TinyStories (2M)
Test-1-3000 was trained on the **TinyStories** dataset.
TinyStories is uniquely valuable because it isolates:
- narrative structure,
- reasoning,
- consistency,
- and causality
without the overwhelming informational noise of the open web.
The stories use:
- child-level vocabulary,
- but professionally structured narrative composition.
This creates an ideal environment for studying emergent reasoning inside small language models.
---
# Training Philosophy
The project intentionally prioritizes:
- coherence over memorization,
- reasoning over factual retrieval,
- and narrative intelligence over benchmark chasing.
The goal is not merely to create a chatbot.
The goal is to study:
> how structured cognition emerges inside compact neural systems.
---
#Usage — Quick Start
Install dependencies:
```bash
pip install transformers torch accelerate
```
---
## Inference Example
```python
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
model_path = "GODELEV/Test-1-3000"
# Load Tokenizer and Model
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(
model_path,
torch_dtype=torch.bfloat16,
device_map="auto"
)
# Prompt
prompt = "Once upon a time, Tom found a blue car."
inputs = tokenizer(
prompt,
return_tensors="pt"
).to(model.device)
# Generate
output = model.generate(
**inputs,
max_new_tokens=200,
temperature=0.7,
top_p=0.9,
repetition_penalty=1.1,
do_sample=True,
eos_token_id=tokenizer.eos_token_id,
pad_token_id=tokenizer.pad_token_id
)
print(tokenizer.decode(output[0], skip_special_tokens=True))
```
---
# Recommended Generation Settings
| Parameter | Recommended |
|---|---|
| Temperature | 0.7 |
| Top-p | 0.9 |
| Repetition Penalty | 1.1 |
| Max Tokens | 128512 |
| Sampling | Enabled |
---
# Observed Emergent Behaviors
During evaluation, the model demonstrated:
- Character persistence
- Goal-oriented progression
- Emotional continuity
- Environmental consistency
- Contextual callbacks
- Story resolution awareness
These behaviors are especially notable given the model's relatively small parameter count.
---
# Limitations
Although highly capable for its size, Test-1-3000 still has limitations:
- Limited factual world knowledge
- Occasional repetition in very long generations
- Reduced reasoning performance outside storytelling domains
- Less stable beyond trained narrative styles
The model is optimized specifically for:
> coherent short-form storytelling.
---
``
---
# 📜 Citation
```bibtex
@misc{test13000,
title={Test-1-3000: A 190M Parameter Narrative Intelligence Engine},
author={GODELEV},
year={2026},
note={Compact narrative-focused language model trained on TinyStories}
}
```
---
# License
This project is intended for:
- research,
- experimentation,
- educational use,
- and open exploration of compact language models.
---
# Final Thoughts
Test-1-3000 demonstrates that meaningful narrative intelligence can emerge inside surprisingly small neural systems when training is focused, clean, and structurally optimized.
At only **190M parameters**, the model exhibits behaviors often associated with significantly larger systems:
- narrative planning,
- emotional continuity,
- causal consistency,
- and coherent resolution generation.
The project serves as both:
- a practical storytelling model,
- and an experiment in emergent cognition within compact architectures.
---
<p align="center">
### “Small models are not weak models.
### They are compressed intelligence waiting to emerge.”
</p>
````