Joint Optimization of Reasoning and Dual-Memory for Self-Learning Diagnostic Agent
TLDR
SEA is a self-learning diagnostic agent that uses a dual-memory module and reinforcement learning to improve clinical reasoning and continually adapt.
Key contributions
- Introduces SEA, a self-learning diagnostic agent with a novel dual-memory module for experience reuse.
- Uses a reinforcement training framework to jointly optimize diagnostic reasoning and memory management.
- Achieves 92.46% accuracy on MedCaseReasoning (+19.6% over baseline) and strong long-horizon learning.
- Expert evaluation confirms clinical correctness and usefulness of SEA's induced diagnostic rules.
Why it matters
This paper introduces a self-learning diagnostic agent that overcomes limitations of independent case processing in LLMs. By jointly optimizing reasoning and memory, it significantly improves diagnostic accuracy and long-term adaptation. Expert-validated rules enhance trust and practical applicability in clinical settings.
Original Abstract
Clinical expertise improves not only by acquiring medical knowledge, but by accumulating experience that yields reusable diagnostic patterns. Recent LLMs-based diagnostic agents have shown promising progress in clinical reasoning for decision support. However, most approaches treat cases independently, limiting experience reuse and continual adaptation. We propose SEA, a self-learning diagnostic agent with cognitively inspired dual-memory module. We design a reinforcement training framework tailored to our designed agent for joint optimization of reasoning and memory management. We evaluate SEA in two complementary settings. On standard evaluation with MedCaseReasoning dataset, SEA achieves 92.46% accuracy, outperforming the strongest baseline by +19.6%, demonstrating the benefit of jointly optimizing reasoning and memory. On the long-horizon with ER-Reason dataset, SEA attains the best final accuracy (0.7214) and the largest improvement (+0.35 Acc@100), while baseline methods show limited or unstable gains. Expert evaluation further indicates that rules consolidated from SEA show strong clinical correctness, usefulness and trust, suggesting that the induced rules in dual-memory module are reliable and practically meaningful. Overall, SEA improves both diagnostic reasoning ability and continual learning by effectively transforming experience into reusable knowledge.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.