English

Radiomics-Guided Vision Transformers for Survival Analysis

Medical Physics 2026-04-24 v1

Abstract

Vision Transformers (ViTs) have shown strong empirical performance on high-dimensional medical imaging data, yet their behavior under survival objectives and the interpretability of their attention mechanisms remain poorly understood. Under shallow ViTs, we design controlled experiments showing that token-level attention dynamics can recover outcome-relevant regions and that attention-based thresholding enables effective token pruning, improving both interpretability and predictive performance. We also study pretrained deep ViTs for survival analysis and propose a radiomics-guided hybrid model that integrates pixel-based embeddings with interpretable radiomic features through a multimodal Cox framework and contrastive alignment. Applied to a COVID-19 chest X-ray cohort with a composite ICU admission or mortality endpoint, the proposed approach achieves competitive discrimination while providing clinically meaningful attention maps and feature-group importance.

Keywords

Cite

@article{arxiv.2604.21056,
  title  = {Radiomics-Guided Vision Transformers for Survival Analysis},
  author = {Qiyuan Shi and Yi Li},
  journal= {arXiv preprint arXiv:2604.21056},
  year   = {2026}
}
R2 v1 2026-07-01T12:31:24.862Z