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

CoMemo: LVLMs Need Image Context with Image Memory

About

Recent advancements in Large Vision-Language Models built upon Large Language Models have established aligning visual features with LLM representations as the dominant paradigm. However, inherited LLM architectural designs introduce suboptimal characteristics for multimodal processing. First, LVLMs exhibit a bimodal distribution in attention allocation, leading to the progressive neglect of middle visual content as context expands. Second, conventional positional encoding schemes fail to preserve vital 2D structural relationships when processing dynamic high-resolution images. To address these limitations, we propose CoMemo - a dual-path architecture that combines a Context image path with an image Memory path for visual processing, effectively alleviating visual information neglect. Additionally, we introduce RoPE-DHR, a novel positional encoding mechanism that employs thumbnail-based positional aggregation to maintain 2D spatial awareness while mitigating remote decay in extended sequences. Evaluations across seven benchmarks,including long-context comprehension, multi-image reasoning, and visual question answering, demonstrate CoMemo's superior performance compared to conventional LVLM architectures. Project page is available at https://lalbj.github.io/projects/CoMemo/.

Shi Liu, Weijie Su, Xizhou Zhu, Wenhai Wang, Jifeng Dai• 2025

Related benchmarks

TaskDatasetResultRank
Mathematics ReasoningMathVision Mini
Accuracy43.4
35
Mathematical ReasoningMathVerse-V
Accuracy55
28
Multimodal UnderstandingMMStar (test)
Accuracy69.9
26
Multimodal UnderstandingMMMU (dev)
Accuracy62
25
Multimodal UnderstandingMMT emo
Accuracy60.8
20
Multimodal UnderstandingMMBench-EN lite
Accuracy88.6
20
Multimodal UnderstandingMMBench-CN lite
Accuracy88
20
Diagram UnderstandingAI2D lite
Accuracy79.6
20
Showing 8 of 8 rows

Other info

Follow for update