Next-visit Diagnosis Prediction on MIMIC-IV
59.8Recall@5GRAIL (+Llama-3.1-8B-Instruct)
Evaluation results
| Method | Links | |||
|---|---|---|---|---|
| GRAIL (+Llama-3.1-8B-Instruct)LLM Backend=Llama-3.1-8B-Instruct2026.02 | 59.8 | 61.8 | 52.2 | |
| GRAIL (+Qwen3-8B)LLM Backend=Qwen3-8B2026.02 | 58.7 | 65.7 | 52 | |
| GRAIL (w.o. LLM backend)LLM Backend=None2026.02 | 38.3 | 41 | 31.5 | |
| Llama-3.1-70B-InstructZero-shot=True, Parameter Count=70B2026.02 | 37.6 | 47 | 28.7 | |
| Hyperbolic (no stitch)Geometry=Hyperbolic, Temporal Stitching=False2026.02 | 26.5 | 38.7 | 29.3 | |
| Euclidean GRAILGeometry=Euclidean, Framework=Graph & Retrieval2026.02 | 25.7 | 37.7 | 28.4 | |
| Euclidean RAGGeometry=Euclidean, Framework=Graph & Retrieval2026.02 | 24.1 | 35.7 | 26.6 | |
| BEHRTModel Type=Sequential Baseline2026.02 | 22.6 | 33.4 | 24.6 | |
| Transformer (Eucl.)Geometry=Euclidean2026.02 | 21.4 | 31.8 | 23.2 | |
| RETAINModel Type=Sequential Baseline2026.02 | 20.1 | 30.1 | 21.7 | |
| Qwen3-8BZero-shot=True, Parameter Count=8B2026.02 | 16 | 25 | 17.1 | |
| Llama-3.1-8B-InstructZero-shot=True, Parameter Count=8B2026.02 | 15.4 | 24.3 | 16.5 |