Multimodal Survival Analysis with Locally Deployable Large Language Models
Organizations: University of Oxford
Abstract
We study multimodal survival analysis integrating clinical text, tabular covariates, and genomic profiles using locally deployable large language models (LLMs). As many institutions face tight computational and privacy constraints, this setting motivates the use of lightweight, on-premises models. Our approach jointly estimates calibrated survival probabilities and generates concise, evidence-grounded prognosis text via teacher-student distillation and principled multimodal fusion. On a TCGA cohort, it outperforms standard baselines, avoids reliance on cloud services and associated privacy concerns, and reduces the risk of hallucinated or miscalibrated estimates that can be observed in base LLMs.
Figures & tables
| Model/Configuration | Hidden-state | Verbalized | Combined | ||||||||||
| Name | Text ∗ | Cov | GE | Survival | Fusion | Pretrain | CC | IBS | IBS | IBS | |||
| BERTSurv | CoxPH | Early | — | — | 0.691 | 0.150 | — | — | — | — | |||
| Cov-only | Discrete | — | — | — | 0.665 | 0.149 | — | — | — | — | |||
| CoxPH | — | — | — | 0.668 | 0.149 | — | — | — | — | ||||
| GE-only | Discrete | — | — | — | 0.734 | 0.139 | — | — | — | — | |||
| CoxPH | — | — | — | 0.751 | 0.135 | — | — | — | — | ||||
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Component | Parameter | Value |
| \Block [draw=gray, fill=gray!15]28-1 Student Model | LLM | LLM Name | DeepSeek-R1 Distill Qwen-1.5B |
| Freezing | All but last 18 transformer layers | ||
| Precision | bfloat16 | ||
| Attention dropout | 0.1 | ||
| Learning rate | 5e-5 | ||
| Autoencoder | Latent dim | 128 |
Multimodal Survival Analysis
Multimodal survival analysis combines imaging, molecular, and clinical data to predict when health events occur and how risk changes over time.
See all 46 papers