arXiv cs.AI· Ziyang Song, Lia Shen, Yixuan Li, Qincheng Lu, Weihao Li, Haoran Xu, He Zhu, Xinyu Wang, Yue Li·· 9 小时前AI 评分22
SMI:基于本体的语义 ID 实现层级感知的医学概念表示
SMI: Ontology-Grounded Semantic IDs for Hierarchy-Aware Medical Concept Representation
AI 导读
研究提出免训练框架 SMI,通过专家定义的医学本体为诊断、手术、药物概念构建由粗到细的语义 ID,并聚合冻结的生物医学 LM 嵌入生成层级感知的医学概念表示。在 MIMIC-IV 上,SMI 提升了死亡率预测、再入院预测、药物推荐和罕见药物预测;在 eICU 上,SMI 在跨医院分布偏移下生成更稳健的患者表示。该工作已被 NeurIPS 2026 主赛道接收。
来源:arXiv cs.AI · arxiv.org