CARVE: Content-Aware Recurrent with Value Efficiency for Chunk-Parallel Linear Attention
Summary
CARVE (Content-Aware Recurrent with Value Efficiency) is a novel architecture addressing key limitations in advanced recurrent language models like GDN-2. It resolves three intrinsic defects of prior recurrent sequence models (like GDN-2): memory-blind gating, write-gate bandwidth bottlenecks, and value-axis erase destroying chunk-parallelism. CARVE introduces content-aware erase, reusing recurrent output tensors ("m_c") for memory-conditioned selectivity at zero additional HBM cost, with a measured 0.18% gate perturbation across chunk lengths up to L=128. It also implements a scalar value write gate, reducing parameters from 589,824 to 9,216 per layer without associative capacity loss. At the 1.3B-parameter scale, trained on 100B tokens, CARVE achieves a WikiText perplexity of 15.72 (vs. GDN-2's 15.90), leads common-sense reasoning by +0.63 pp, and sets new benchmarks on RULER retrieval. Critically, it maintains throughput within 0.4% of the baseline while reducing peak memory by 13% and mixer parameters by 19%.
Key takeaway
For Machine Learning Engineers designing or optimizing recurrent neural networks, CARVE presents a compelling architecture for long-context language models. Its content-aware erase and scalar write gate deliver superior perplexity and retrieval performance while significantly reducing memory footprint by 13% and mixer parameters by 19%. You should evaluate CARVE's key-axis gating principle to achieve efficient chunk-parallel training and enhanced memory management in your next-generation recurrent models, especially for tasks demanding precise information retention.
Key insights
Recurrent models can achieve content-aware forgetting and efficiency by reusing kernel outputs and restricting gating to the key axis.
Principles
- Key-axis gating enables chunk-parallel training.
- Output reuse provides memory-awareness at zero HBM cost.
- Scalar write gates suffice for associative recall.
Method
CARVE restricts gating to the key axis, uses a chunk-mean of previous recurrent outputs ("m_c") for content-aware erase, and employs a scalar write gate per head, preserving WY-form chunk-parallel training.
In practice
- Implement key-axis-only gating for recurrent models.
- Reuse kernel outputs for memory-aware gate conditioning.
- Consider scalar write gates for parameter reduction.
Topics
- Recurrent Neural Networks
- Linear Attention
- Memory Management
- Chunk-Parallel Training
- Hardware Efficiency
- Language Modeling
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Architect
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.