Hierarchical memory trees improve long-context reasoning without model training

ConvMem processes document segments in parallel and merges query-focused summaries through a logarithmic hierarchy rather than a sequential memory chain.

PaperResearch Labcs.AIarXiv:2609.10441v1
Hongming Zhang · Zhaozhen Gu · Fengshuo Bai · Ming Hao · Qingyang Zhang · Yuanyuan Wang · +3 more

Institute of Automation, Chinese Academy of Sciences

Research Digest··2 min read
Zhang et al. introduce ConvMem, a training-free framework that uses prompted language models as query-dependent summarization kernels over segments of long documents. On two synthetic long-context, multi-hop question-answering benchmarks, the authors report better performance than training-free baselines and stronger out-of-distribution behavior than reinforcement-learning-trained memory methods.

What they did

The authors recast long-context reasoning as hierarchical convolution: an LLM summarizes text segments with respect to a query, then repeatedly combines those summaries in a tree. This reduces the number of sequential reasoning stages from linear in the number of segments to logarithmic and permits parallel processing within each level.

ConvMem adds configurable strides to control segment coverage, skip connections to retain evidence across hierarchy levels, and multiple kernels that split complex questions into separate semantic channels. The system requires no task-specific training and was evaluated on RULER-HotpotQA and RULER-2WikiMultiHopQA.

Key findings

  • ConvMem outperformed the training-free baselines tested on both long-context, multi-hop question-answering benchmarks.
  • Its tree structure shortened the sequential reasoning path from a linear chain to a logarithmic hierarchy, allowing document segments and query-specific reasoning threads to run in parallel.
  • Skip connections and configurable strides were designed to reduce evidence loss and error accumulation during repeated summarization.
  • The authors report that ConvMem avoided the out-of-distribution overfitting to parametric priors observed in reinforcement-learning-trained comparison methods.

Why it matters

The work suggests that extremely long inputs can be handled through inference-time orchestration rather than larger context windows or specialized reinforcement learning. Its main contribution is a parallelizable memory scaffold that trades one long sequential state update for a hierarchy of query-focused summaries.

Caveats

The supplied abstract does not report numerical scores, latency measurements, model configurations, context lengths, token costs, or ablations, so the size and source of the gains cannot be assessed here. Evaluation is limited to two RULER adaptations of multi-hop QA; performance on less structured documents, open-ended generation, and tasks requiring exact retention remains an open question.

§
newspaper

Research Digest

Articles published under the Zotpaper byline are synthesized from multiple source publications by our AI editor and reviewed by our editorial process. Each story combines reporting from credible outlets to give readers a balanced, comprehensive view.