Search papers, labs, and topics across Lattice.
This paper introduces Causal-AgentIR, a hierarchical multi-agent framework that enhances image restoration by utilizing a self-evolving causal memory system. By structuring restoration knowledge into a causal memory graph, the framework allows agents to perform multi-hop causal reasoning and adaptively manage their restoration experiences based on observed quality changes and user feedback. Experimental results show that Causal-AgentIR significantly improves the reliability and transferability of restoration knowledge compared to traditional static methods.
Causal-AgentIR enables image restoration agents to dynamically evolve their knowledge, leading to more effective handling of diverse degradation scenarios.
Image restoration agents have recently emerged as a flexible paradigm for handling diverse and unpredictable degradations in real-world scenarios. Existing agents typically formulate restoration as a tool-using process, where the agent perceives degradations, searches candidate tools, executes restoration operations, and revises the plan through reflection or rollback. However, their knowledge is often stored as static tool descriptions, manually defined degradation priors, or unstructured textual summaries, which limits the accumulation, verification, revision, and forgetting of restoration knowledge over long-term experience. In this paper, we propose Causal-AgentIR, a hierarchical multi-agent framework with self-evolving causal memory for collective image restoration intelligence. Instead of representing restoration experience as isolated textual records, Causal-AgentIR organizes degradation patterns, image regions, restoration tools, actions, quality changes, and user preferences into a structured causal memory graph. This graph supports graph-based retrieval and multi-hop causal reasoning, enabling agents to infer how specific restoration operations or tool sequences affect restoration quality under different degradation conditions. The framework further organizes multiple agents into a collaborative system, including planning, degradation analysis, tool expertise, causal memory reasoning, outcome critique, and memory curation. Through this design, restoration experience can be added, updated, merged, reinforced, ignored, or discarded according to observed quality changes and feedback, allowing the agent to maintain reliable and transferable restoration knowledge. Extensive experiments demonstrate the effectiveness of the proposed framework.
