Medication recommender is to suggest appropriate medication combinations based on a patient’s health history, e.g., diagnoses and procedures. Existing works represent different diagnoses/procedures well separated by one-hot encodings. However, they ignore the latent hierarchical structures of these medical terms, undermining the generalization performance of the model. For example, “Respiratory Diseases”, “Chronic Respiratory Diseases” and “Chronic Bronchiti” have a hierarchical relationship, progressing from general to specific. To address this issue, we propose a novel hierarchical encoder named HIER to hierarchically represent diagnoses and procedures, which is based on standard medical codes and compatible with any existing methods. Specifically, the proposed method learns relation embedding with a self-supervised objective for incorporating the neighbor hierarchical structure. Additionally, we develop the position encoding to explicitly introduce global hierarchical position. Extensive experiments demonstrate significant and consistent improvements in recommendation accuracy across four baselines and two real-world clinical datasets. The code is available at https://github.com/yuliang-liang/HEIR .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Self-supervised Hierarchical Representation for Medication Recommendation

  • Yuliang Liang,
  • Yuting Liu,
  • Yizhou Dang,
  • Enneng Yang,
  • Guibing Guo,
  • Wei Cai,
  • Jianzhe Zhao,
  • Xingwei Wang

摘要

Medication recommender is to suggest appropriate medication combinations based on a patient’s health history, e.g., diagnoses and procedures. Existing works represent different diagnoses/procedures well separated by one-hot encodings. However, they ignore the latent hierarchical structures of these medical terms, undermining the generalization performance of the model. For example, “Respiratory Diseases”, “Chronic Respiratory Diseases” and “Chronic Bronchiti” have a hierarchical relationship, progressing from general to specific. To address this issue, we propose a novel hierarchical encoder named HIER to hierarchically represent diagnoses and procedures, which is based on standard medical codes and compatible with any existing methods. Specifically, the proposed method learns relation embedding with a self-supervised objective for incorporating the neighbor hierarchical structure. Additionally, we develop the position encoding to explicitly introduce global hierarchical position. Extensive experiments demonstrate significant and consistent improvements in recommendation accuracy across four baselines and two real-world clinical datasets. The code is available at https://github.com/yuliang-liang/HEIR .