Rethinking Next-Visit Diagnosis Prediction Beyond Aggregate Metrics: Retrieval--Memory Multi-Agent Adjudication
Heejoon Koo
Abstract
Next-visit diagnosis prediction forecasts the diagnoses documented at a patient's next encounter, turning retrospective electronic health records into prospective clinical prediction. However, conventional aggregate evaluation obscures its multifaceted nature: distributionally, models must handle variable-size diagnosis sets and long-tailed labels; temporally, they must distinguish persistent, newly recorded, and recurrent diagnoses; and both vary with observed history depth. We introduce an evaluation framework along these dimensions and apply it to rule-based heuristics, LLM-based prompting, retrieval, memory, and multi-agent approaches. This reveals that repeating the last visit outperforms most LLM-based approaches overall despite predicting no new or recurrent diagnoses, and that retrieval and memory exhibit distinct profiles whose relative strengths vary across temporal transitions and backbones. Motivated by this, we propose LONGINUS, a retrieval-memory multi-agent adjudication framework combining evidence retrieved from patient records with structured memory of the patient trajectory. Using cross-view agreement as a reliability heuristic, it preserves shared predictions and adjudicates disagreements without revealing their source. Across MIMIC-III and MIMIC-IV with four LLM backbones, LONGINUS achieves the best transition-specific performance among LLM-based methods in nearly all settings while remaining competitive with them on aggregate metrics.
Create a lesson
Related papers
COMPREHEND: Ultra-Fine-Grained Benchmark Understanding by Entailment Hierarchy
Sivaraaman Balakrishnan, Prakriti Shetty, Jeremiah Hong et al.
Constructing a Regional Slang Lexicon for Electricity Demands via Soft-Label Bidirectional Anchoring and Active Learning
Rong Chang, Lijing Shi, Zhaoxiang Zang
Shared Context Candidate Scoring with Jointly Adapted Readout Networks
Jingyang Du, Yiben Fu, Yang Shen
Tokenizer Myths and Where To Find Them
Rylan Schaeffer, Sanmi Koyejo
Length Is Cheap, Competition Is Not: Attention Predicts Context Rot in Retrieval
Yinuo Liu
PACT: Certifying What Is Written in Parallel Decoding of Diffusion Language Models
Yanchao Li, Yusu Zhang, Jiaqing Xie et al.