Next-visit Medication Prediction on MIMIC-IV
50.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 | 50.8 | 57.3 | 52.3 | |
| GRAIL (+Qwen3-8B)LLM Backend=Qwen3-8B2026.02 | 50.6 | 57 | 52.1 | |
| Llama-3.1-70B-InstructZero-shot=True2026.02 | 24.9 | 28.6 | 25.6 | |
| GRAIL (w.o. LLM backend)LLM Backend=None2026.02 | 20.1 | 26.2 | 21.4 | |
| Hyperbolic (no stitch)Geometry=Hyperbolic2026.02 | 9 | 14.6 | 10.1 | |
| Euclidean GRAILGeometry=Euclidean2026.02 | 8.6 | 14 | 9.7 | |
| Euclidean RAGGeometry=Euclidean2026.02 | 8 | 13.2 | 9 | |
| BEHRTModel Type=Sequential Baseline2026.02 | 7 | 11.8 | 7.8 | |
| Transformer (Eucl.)Geometry=Euclidean2026.02 | 6.6 | 11.2 | 7.4 | |
| RETAINModel Type=Sequential Baseline2026.02 | 6.2 | 10.6 | 7 | |
| Qwen3-8BZero-shot=True2026.02 | 4.3 | 7.8 | 5 | |
| Llama-3.1-8B-InstructZero-shot=True2026.02 | 4.1 | 7.5 | 4.8 |