Remember When It Matters: Proactive Memory Agent for Long-Horizon Agents
This paper introduces a proactive memory agent that actively manages and injects decision-relevant information to prevent "behavioral state decay" in long-horizon tasks, significantly improving performance across various benchmarks through selective intervention and open-weight policy training.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
Imagine you're playing a super-complex video game that lasts for hours. You start with a clear mission: "Build a castle, but don't use red bricks, and make sure the moat is filled with water." You spend the first hour gathering blue bricks and digging the ditch. Then, you get distracted by a cool dragon that shows up. You spend the next hour fighting it, forgetting about the castle. Suddenly, you remember the castle, but your brain is so full of dragon-slaying details that you accidentally grab a red brick and start building. You've forgotten the rules you knew perfectly well just an hour ago.
This is exactly what happens to AI agents when they try to solve long, complicated tasks. The researchers at Meta AI call this "behavioral state decay." It's not that the AI forgets the information entirely; the data is still there in its "memory" (the chat history), but it stops mattering to the decisions it makes next. The AI gets lost in the noise of its own long conversation.
The Solution: The "Proactive Sidekick"
To fix this, the authors didn't just give the AI a bigger notebook. Instead, they introduced a Proactive Memory Agent. Think of this as a hyper-organized, slightly bossy sidekick who stands next to the main AI player.
Here's how this sidekick works:
- The Watcher: While the main AI is busy playing the game (or writing code, or debugging), the sidekick is quietly watching the whole scene.
- The Organizer: Every few steps, the sidekick stops and organizes a "cheat sheet." It doesn't just copy-paste everything; it sorts facts into three buckets:
- Status: "We are currently trying to fix the engine." (Private info, never shown to the main player).
- Knowledge: "The castle cannot have red bricks." (Hard rules).
- Procedures: "We tried using a hammer, and it broke the wall. Don't do that again." (What worked and what failed).
- The Decider: This is the magic part. The sidekick has to decide: Should I interrupt the main player right now?
- If the main player is on a roll and doing fine, the sidekick stays silent.
- If the main player is about to make a mistake (like grabbing a red brick), the sidekick shouts, "Hey! Remember the no-red-brick rule!" and injects that reminder into the main player's mind.
What They Ruled Out (The "Don'ts")
The researchers tested a few other ideas and found they didn't work as well:
- Just dumping the whole notebook: They tried showing the main AI the entire cheat sheet every single time. This didn't work because it was too much information, like trying to read a whole encyclopedia while driving. The AI got distracted.
- Always shouting reminders: They tried making the sidekick interrupt the main player at every step, even when it wasn't necessary. This slowed things down and annoyed the main player, leading to worse results.
- Just a "wise advisor": They tried a system where a second AI just gave general advice without keeping a structured list of facts. This was less reliable because the advice wasn't grounded in the specific history of what actually happened.
The paper suggests that the secret sauce is selective intervention. The sidekick needs to be smart enough to know when to speak up and what to say, rather than just being a passive storage unit or a constant chatterbox.
The Results: Does It Actually Help?
The team tested this "sidekick" on two very tough test sets: Terminal-Bench 2.0 (where AI has to fix computer code and use command lines) and τ 2-Bench (where AI has to talk to users and use tools like an airline agent or a retail helper).
The results were pretty clear:
- For the "weaker" AI (Claude Sonnet 4.5): The sidekick helped it solve 8.3% more tasks on Terminal-Bench and 6.8% more tasks on the average of the τ 2-Bench tests. That's a huge jump!
- For the "stronger" AI (Claude Opus 4.6): Even the super-smart AI got better, solving 2.4% more tasks on Terminal-Bench and 2.5% more on τ 2-Bench. This proves the sidekick isn't just a crutch for weak AIs; it helps even the pros avoid getting distracted.
Can We Teach a Cheaper AI to Be the Sidekick?
The researchers also wanted to see if they could train a smaller, open-source AI (Qwen3.5-27B) to be the sidekick instead of using a giant, expensive one.
- At first, the untrained small AI actually made things worse.
- But after they taught it using a method called SFT (Supervised Fine-Tuning) and then GRPO (a type of reinforcement learning), it started getting it right.
- On a test set it had never seen before, this trained small sidekick helped the main AI improve its success rate from 37.6% to 41.1%.
The Bottom Line
The paper suggests that for AI to get really good at long, complex tasks, it can't just rely on its own memory. It needs a separate, proactive system that acts like a vigilant coach. This coach keeps the rules and past mistakes organized and only speaks up at the exact moment the player is about to forget them. It's not about remembering more; it's about remembering at the right time.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.