TR-HASH MoE 200M β€” Interrupted 160B-Token Refinement Checkpoint

Phase 2 stopped at step 8,156 / 17,802. The root model.safetensors is this intermediate checkpoint (45.8%, approximately 32.07B refinement tokens), not a completed 70B-token refinement. Together with the 130B-token base run, it represents roughly 162B trained token exposures and is reported as the 160B-token source checkpoint.

Plan

Phase 2 takes the finished, fully-decayed TR-HASH 200M base checkpoint and runs a full-parameter refinement pass over the same 70B unique tokens used for base pretraining -- no replay, no augmentation, a fresh optimizer and learning-rate schedule, initialized from the base checkpoint's weights only.

Setting Value
Base checkpoint TR-HASH-200M-130B (post-decay)
Training tokens 69,997,690,880 (70B unique, exact match required)
Training progress represented by current weights Step 8,156 / 17,802 (45.8%, ~32.07B tokens)
Replay / augmentation None
Optimizer Fresh at phase-2 start (not resumed from base pretraining state)
Parameters refined Full model (not LoRA)

Interim PIQA result

The step-8,156 checkpoint already exceeds the released base checkpoint on PIQA. This is an interim measurement, not the final phase-2 result.

Checkpoint PIQA acc PIQA acc_norm Correct (acc) Examples
TR-HASH-200M-130B base final 65.45% 65.61% β€” 1,838
Phase 2, step 8,156 68.66% 68.39% 1,262 / 1,838 1,838

Evaluation protocol: full PIQA validation split, zero-shot causal continuation log-likelihood, no chat template, maximum sequence length 2,048. The checkpoint was evaluated in FP16 through MLX using the same model weights. acc_norm selects the continuation by length-normalized log-likelihood. Final results will be measured again after step 17,802.

Model summary

Same architecture as the base checkpoint: 201.2M parameters, 16 transformer layers, GQA (14 query heads / 2 KV heads), deterministic token-ID routed TR-Hash MoE (4 stored experts, top-2 active), tied embeddings. See TR-HASH-MoE-200M-130B for the full architecture and base pretraining details.

Status

  • Base pretraining run complete
  • Phase-2 refinement launched
  • Intermediate step-8,156 weights, optimizer state, curves, and PIQA evaluation uploaded
  • Phase-2 run stopped at step 8,156
  • Full 70B-token refinement completed (not reached)

No active process is expected to finish this repository. The root weights remain the evaluated step-8,156 intermediate checkpoint.

Instruction post-training

The earlier experimental LoRA repository has been retired. The next planned stage is a full-parameter instruction SFT, documented in AETHORIA-AI/TR-HASH-MoE-200M-160B-SFT.

Project links

License

The model release is CC BY-NC 4.0. Source datasets and individual documents retain their own licenses and terms.

Downloads last month
33
Safetensors
Model size
0.2B params
Tensor type
I64
Β·
F32
Β·
I32
Β·
U8
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for AETHORIA-AI/TR-HASH-MoE-200M-160B-Refinement

Finetunes
2 models

Collection including AETHORIA-AI/TR-HASH-MoE-200M-160B-Refinement