Search papers, labs, and topics across Lattice.
This paper introduces TaskMem, a reinforcement learning framework that trains a memorization policy for multimodal agents to selectively retain task-relevant information from continuous observation streams. TaskMem uses a two-phase training approach: first optimizing memory quality based on fidelity, and then tuning an adapter on a base MLLM using environment tasks to guide the memorization policy. Experiments on reformulated streaming versions of VideoMME, EgoLife, and EgoTempo show that TaskMem improves VQA accuracy by up to 7% compared to baselines.
Forget everything you thought you knew about multimodal agent memory: TaskMem learns what to remember on the fly, boosting VQA accuracy by up to 7% without even looking at the raw video.
Long-term memory is essential for multimodal agents to build coherent experience, accumulate world knowledge, and achieve continual learning. However, constructing effective memory goes beyond memory module design and basic requirements such as accuracy and fidelity; the key challenge lies in determining what to memorize. Multimodal agents, such as embodied agents, continuously perceive, reason, and act in real or virtual environments, receiving an unbounded stream of multimodal observations. From this combinatorial explosion of information, an agent must selectively retain content that is relevant to its role in the environment and valuable for future tasks. To bridge this gap, we frame memory generation as a learnable memorization policy and introduce TaskMem (Task-focused Memorization Policy Learning), a reinforcement-learning-based framework that enables the policy to dynamically adjust its focus to the demands of real tasks encountered in the environment. TaskMem adopts a two-phase training paradigm: Phase One learns how to memorize by optimizing memory quality under fundamental fidelity requirements; Phase Two occurs after deployment, where the agent learns what to memorize by tuning an adapter on its base MLLM, using recent environment tasks to define a reward model that guides the memorization policy toward task-relevant content. To evaluate our approach, we reformulate VideoMME, EgoLife, and EgoTempo into streaming benchmarks that simulate a realistic setting in which an agent processes streaming observations and handles tasks arriving online. To isolate memory assessment, the questions must be answered using only the agent's memory, without access to raw video. Built on Qwen3-VL-30B-A3B, TaskMem improves VQA accuracy by 6.3%, 7.0%, and 5.3% on these benchmarks, respectively.
