TemporalMesh-Transformer / CHANGELOG.md
vigneshwar234's picture
Add CHANGELOG.md
7078228 verified
|
Raw
History Blame
1.69 kB

Changelog

[1.0.0] β€” 2026-05-19

Added

  • Initial public release of TemporalMesh Transformer (TMT)
  • TMTConfig β€” full hyperparameter dataclass with 5 model scale presets
  • MeshBuilder β€” dynamic kNN graph rebuilt every forward pass from cosine similarity
  • MeshAttention β€” multi-head attention over sparse graph edges, O(SΒ·k) cost
  • TemporalPositionEncoder β€” RoPE + per-token learned decay scalars
  • ExitGate β€” per-token confidence scoring with freeze-on-threshold logic
  • DualStreamFFN β€” parallel syntax + semantic streams with learned gated fusion
  • MemoryAnchorCross β€” 16 persistent EMA key-value anchor vectors
  • TMTLayer β€” unified layer assembling all five components
  • TMTModel β€” full autoregressive model with tied output projection
  • TMTOutput β€” structured output dataclass (logits, exit_masks, confidences, graph_edges, memory_state, decay_scalars)
  • TMTTrainer β€” training loop with wandb logging, cosine warmup, checkpoint saving
  • CosineWarmupScheduler β€” learning rate schedule
  • TMTLoss β€” cross-entropy + 0.1 Γ— gate auxiliary loss
  • Dataset loader for WikiText-2 and TinyStories
  • HuggingFace tokenizer wrapper
  • Full ablation notebooks (01–04)
  • 15-test pytest suite (shapes + forward pass)
  • 20-page publication-quality PDF with 7 figures and 18 equations
  • 5-subset HuggingFace benchmark dataset
  • Zenodo DOI registration
  • GitHub Pages documentation site

Architecture Details

  • Default: d_model=512, n_heads=8, n_layers=12, graph_k=8, exit_threshold=0.85
  • ~120M parameters (TMT-Base)
  • WikiText-2 val perplexity: 29.4 (vs 42.1 vanilla baseline)
  • Average compute per token: ~48% of full-depth baseline