SAFER: A Calibrated Risk-Aware Multimodal Recommendation Model for Dynamic Treatment Regimes
Yishan Shen, Yuyang Ye, Hui Xiong, Yong Chen
Abstract
Dynamic treatment regimes (DTRs) are critical to precision medicine, optimizing long-term outcomes through personalized, real-time decisionmaking in evolving clinical contexts, but require careful supervision for unsafe treatment risks. Existing efforts rely primarily on clinicianprescribed gold standards despite the absence of a known optimal strategy, and predominantly using structured EHR data without extracting valuable insights from clinical notes, limiting their reliability for treatment recommendations. In this work, we introduce SAFER, a calibrated risk-aware tabular-language recommendation framework for DTR that integrates both structured EHR and clinical notes, enabling them to learn from each other, and addresses inherent label uncertainty by assuming ambiguous optimal treatment solution for deceased patients. Moreover, SAFER employs conformal prediction to provide statistical guarantees, ensuring safe treatment recommendations while filtering out uncertain predictions. Experiments on two publicly available sepsis datasets demonstrate that SAFER outperforms state-ofthe-art baselines across multiple recommendation metrics and counterfactual mortality rate, while offering robust formal assurances. These findings underscore SAFER's potential as a trustworthy and theoretically grounded solution for highstakes DTR applications.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a9a40d49-032a-415c-b2b3-7bfde38b4831Cited by top-tier papers1
Ask how each one uses itBuilds on7
- Estimating counterfactual treatment outcomes over time through adversarially balanced representationsIoana Bica, Ahmed M. Alaa, James Jordon, Mihaela van der SchaarICLR 2020 · 224 citations
- Causal Transformer for Estimating Counterfactual OutcomesValentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICML 2022 · 146 citations
- Conditional Generation Net for Medication RecommendationRui Wu, Zhaopeng Qiu, Jiacheng Jiang, Guilin Qi et al.WWW 2022 · 135 citations
- Harnessing Multimodal Large Language Models for Multimodal Sequential RecommendationYuyang Ye, Zhi Zheng, Yishan Shen, Tianshu Wang et al.AAAI 2025 · 68 citations
- Conformal Alignment: Knowing When to Trust Foundation Models with GuaranteesYu Gui, Ying Jin, Zhimei RenNeurIPS 2024 · 63 citations
Related papers
- Trajectory-Aware Clinical Risk Prediction via Severity-Grounded Knowledge Graphs and Retrieval-Augmented GenerationKyunghoon Jeon, Youmin Ko, Woohwan Jung, Hyunjoon KimKDD 2026
- CURA: Clinical Uncertainty Risk Alignment for Language Model-Based Risk PredictionSizhe Wang, Ziqi Xu, Claire Najjuuko, Charles Alba et al.ACL 2026
- Offline Guarded Safe Reinforcement Learning for Medical Treatment Optimization StrategiesRunze Yan, Xun Shen, Akifumi Wachi, Sebastien Gros et al.NeurIPS 2025 · 7 citations
- Evaluating and Learning Optimal Dynamic Treatment Regimes under Truncation by DeathSihyung Park, Wenbin Lu, Shu YangNeurIPS 2025 · 1 citation
- Conformalized Survival Analysis for General Right-Censored DataHen Davidov, Shai Feldman, Gil Shamai, Ron Kimmel et al.ICLR 2025
