TinyCeNN-LM: Quality-Gated Conversion of Pretrained Attention with CeNN-Inspired Cellular-Recurrent Layers
Organizations: Institute for Smart System Technologies University of Klagenfurt Klagenfurt, Austria · University of Klagenfurt / Inst. f. Smart Systems Technologies, Austria & Faculte Polytechnique Universite de Kinshasa, DR-Congo
Abstract
Replacing attention in a pretrained language model is a compatibility problem: a plausible substitute may alter representations expected by later layers. TinyCeNN-LM introduces a quality-gated post-training conversion framework using CeNN-inspired cellular-recurrent layers with bounded local processing, compact recurrent memory, routing, fusion, and accept-or-rollback validation. Three implementations are studied: Integrated Memory, MemoryFusion, and PDelta3-GDN2-CLVR+Local32. Strict PDelta3 conversion accepts a layer only when representation and NLL criteria pass fixed thresholds. On SmolLM2-135M, layers 0-2 are accepted with cumulative , while layer 3 is rejected despite acceptable NLL because representation fidelity fails. On Qwen3.5-0.8B, full-attention layers 3, 7, and 11 are accepted with final . Integrated Memory keeps perplexity within to while reducing total cache by up to . A sampled 200-item downstream sanity check gives -- overall accuracy for converted Qwen releases. The results support conservative, quality-gated structural conversion rather than universal attention replacement or speedup.
Figures & tables
| Model | Extra params | CeNN state | Native KV@256 | Native KV@2048 |
|---|---|---|---|---|
| SmolLM2 | 0.153M | 65.6 KiB | 192 KiB | 1.50 MiB |
| Qwen3.5 | 0.753M | 178 KiB | 512 KiB | 4.00 MiB |
| Model | Layer | NMSE | Cosine | NLL inc | NLL cum | Local gate | Decision |
|---|---|---|---|---|---|---|---|
| SmolLM2-135M | 0 | 0.0069 | 0.9974 | +0.00239 | +0.00239 | 0.749 | Accept |
| 1 | 0.0108 | 0.9960 | +0.00972 | +0.01211 | 0.766 | Accept | |
| 2 | 0.0856 | 0.9587 | -0.00002 | +0.01209 | 0.766 | Accept | |
| 3 | 0.2265 | 0.8949 | +0.00278 | +0.01487 | 0.441 | Reject | |
| Qwen3.5-0.8B | 3 | 0.0621 | 0.9745 | -0.00325 | -0.00325 | 0.721 | Accept |
| 7 | 0.0389 | 0.9705 | +0.01327 | +0.01003 | 0.721 | Accept |
| Model / release | Custom | MMLU | PIQA | MMMLU | GPQA-D | Overall |
|---|---|---|---|---|---|---|
| layers | Pro | DE | ||||
| Qwen3.5-0.8B (native) | – | 12.0 | 48.0 | 34.0 | 26.0 | 30.0 |
| PDelta3-CLVR+Local32 | 3,7,11 | 14.0 | 48.0 | 30.0 | 22.0 | 28.5 |
| MemoryFusion | 3,7 | 10.0 | 54.0 | 28.0 | 30.0 | 30.5 |
| Integrated Memory | 3,23 | 12.0 | 42.0 | 44.0 | 30.0 | 32.0 |
| Context | PPL/orig. | Cache/orig. | Save | Prefill / Decode |
|---|---|---|---|---|
| 256 | 0.9993 | 0.9857 | 1.43% | 0.830 / 0.893 |
| 512 | 1.0021 | 0.9595 | 4.05% | 0.849 / 0.910 |
| 1024 | 1.0034 | 0.9464 | 5.36% | 0.829 / 0.892 |
| 2048 | 1.0093 | 0.9399 | 6.01% | 0.830 / 0.891 |
Language Model Post-Training
Language model post-training adapts pretrained models to follow instructions, reason, use tools, and meet task-specific or human-defined goals.
See all 193 papers