alphaFeedback

WorksComputer Science

Going Deeper With Memories

Publication details
License
arXiv nonexclusive license
Status
Published
AI contributions
Not recorded for this paper
View details
Ancillary data
1 code link
View
Abstract · v1

Byte level language models avoid tokenizer dependence and provide universal coverage, but processing every byte at full resolution is expensive. We introduce DeepMem, a causal multiscale architecture that separates semantic compression from exact byte preservation. A learned invertible subband transform constructs a hierarchy of shorter sequences, while a lossless carrier and explicit detail sidebands preserve byte identity. Fast weight memory blocks operate along the encoder, with iterative shared state computation at the compressed bottleneck, while an attention based super resolution decoder restores byte resolution representations. DeepMem achieves stronger language modeling performance than a matched byte level Transformer on FineWeb-Edu and PG19, and obtains the best macro average performance across the evaluated FLORES languages. It also improves PIQA accuracy while remaining competitive on code modeling, multiple choice reasoning, and byte level corruption robustness. Bottleneck refinement requires approximately 7,600×7{,}600\times fewer counted FLOPs than equivalent full resolution reasoning, while maintaining competitive evaluation throughput.

Review conversation

Reviews are not available for this work yet.