Complexity-Deep: Token-Routed MLP with Mu-Guided Dynamics for Efficient Transformer Architectures
Description
We present Complexity-Deep, a transformer architecture introducing two key innovations: (1) Token-Routed MLP, a deterministic expert routing mechanism based on token identity that achieves perfect load balancing without auxiliary losses, and (2) Mu-Guided Dynamics, a simplified PID-inspired system that accumulates context across layers and influences both attention and expert routing. Our Token-Routed MLP uses modulo-based routing (expert_id = token_id mod N) ensuring uniform expert utilization regardless of token frequency distribution. We further introduce CGGR (Contiguous Group GEMM Routing), a Triton-based kernel optimization achieving 5-6x speedup over batched matrix multiplication. The architecture incorporates modern techniques including Grouped Query Attention (GQA), RoPE positional embeddings, QK Normalization, and Flash Attention via SDPA. Our 1.5B parameter model demonstrates the viability of deterministic routing as an alternative to learned routing in Mixture-of-Experts architectures.
Files
complexity_paper.pdf
Files
(356.1 kB)
| Name | Size | Download all |
|---|---|---|
|
md5:5511096819fd124742daf18b1536a70c
|
356.1 kB | Preview Download |
Additional details
Related works
- Is supplemented by
- Software: https://github.com/Complexity-ML/complexity-deep (URL)