multi-head

Tag

Cards List
#multi-head

Multi-Head Recurrent Memory Agents

arXiv cs.LG · 4d ago Cached

This paper identifies memory retention as the bottleneck in recurrent memory agents for long contexts and proposes Multi-Head Recurrent Memory (MHM), a training-free framework that partitions memory into independent heads with a select-then-update strategy. The lightweight instantiation MHM-LRU significantly improves retention and end-to-end accuracy across 100K–1M token ranges, raising retention from below 30% to 73.96% on RULER-HQA at 896K tokens.

0 favorites 0 likes
← Back to home

Submit Feedback