CARVE: Content-Aware Recurrent with Value Efficiency for Chunk-Parallel Linear Attention

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, extended

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

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

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Architect

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.