Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MemOCR: Layout-Aware Visual Memory for Efficient Long-Horizon Reasoning

About

Long-horizon agentic reasoning necessitates effectively compressing growing interaction histories into a limited context window. Most existing memory systems serialize history as text, where token-level cost is uniform and scales linearly with length, often spending scarce budget on low-value details. To this end, we introduce MemOCR, a multimodal memory agent that improves long-horizon reasoning under tight context budgets by allocating memory space with adaptive information density through visual layout. Concretely, MemOCR maintains a structured rich-text memory (e.g., headings, highlights) and renders it into an image that the agent consults for memory access, visually prioritizing crucial evidence while aggressively compressing auxiliary details. To ensure robustness across varying memory budgets, we train MemOCR with reinforcement learning under budget-aware objectives that expose the agent to diverse compression levels. Across long-context multi-hop and single-hop question-answering benchmarks, MemOCR outperforms strong text-based baselines and achieves more effective context utilization under extreme budgets.

Yaorui Shi, Shugui Liu, Yu Yang, Wenyu Mao, Yuxin Chen, Qi GU, Hui Su, Xunliang Cai, Xiang Wang, An Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Long-term DialogueLoCoMo (test)
Multi-Hop Accuracy65.6
26
Question Answering2Wiki 100K context
Accuracy65.5
25
Question AnsweringHotpotQA 10K context
Accuracy84.8
19
Question Answering2Wiki 10K context
Accuracy72.2
19
Question Answering2Wiki 30K context
Accuracy73.7
19
Question AnsweringNQ 10K context
Accuracy61.8
19
Question AnsweringAverage 10K context
Accuracy74.6
19
Question AnsweringTriviaQA 10K context
Accuracy80.8
19
Question AnsweringNQ 100K context
Accuracy0.583
18
Question AnsweringTriviaQA 30K context
Accuracy0.813
18
Showing 10 of 18 rows

Other info

GitHub

Follow for update