p-Spin Glass Network Efficient Single-Batch Continual Learning
Organizations: Independent Researcher.
Abstract
Modern sequence models heavily rely on massive memory footprints and large-batch stochastic optimization, barriers that restrict sample efficiency and continual learning. We introduce the -Spin Glass Network, a novel architecture that overcomes these limitations, structurally manages optimization variance and yields four noticeable capabilities: 1. It enforces memory efficiency: native ternary quantization compresses internal parameters by , while exact implicit gradients strictly bound activation memory to . 2. it demonstrates sample efficiency, matching the asymptotic performance of a Transformer baseline while utilizing fewer training sequences. 3. Method enables single-batch stability and smooth, monotonic convergence at a stochastic micro-batch size of . 4. Finally, this stability proves modality-agnostic, maintaining robust temporal credit assignment across both discrete subword and long horizon uncompressed raw byte streams. Ultimately, this work removes large batch requirement for stable deep learning, establishing a foundation for continuous learning and edge AI.
Figures & tables
| Architecture | Modality | Batch Size | Seq. Seen | Time/Step | Final Loss ( ) |
|---|---|---|---|---|---|
| Transformer Base | Subword | s | |||
| -Spin (Ours) | Subword | s | |||
| -Spin (Ours) | Byte | s |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Component | Parameter Count |
|---|---|
| Total Parameters | |
| Embedding / Head (Tied) | |
| Non-Embedding Params |
| Configuration | Layer Type | Params (M) | Logical Precision | Size (MB) |
|---|---|---|---|---|
| Subword -Spin | Embeddings (Tied) | M | -bit (FP32) | MB |
| Eq. MLP (Ternary) | M | -bit (Packed 2b) | MB | |
| State-Space Attention | M | -bit (FP16/BF16) | MB | |
| Total / Footprint | 57.72 M | Uncompressed Eq: 161.09 MB | 124.40 MB | |
| Byte -Spin | Embeddings (Tied) | M | -bit (FP32) | MB |
| Eq. MLP (Ternary) | M | -bit (Packed 2b) | MB |
Memory-Efficient Training
Memory-efficient training reduces the memory needed to optimize neural networks by compressing, recomputing, or relocating training state.
See all 90 papers