Skip to main content
  1. PaperReading/
  2. EMNLP/

Text2Tree: Aligning Text Representation to the Label Tree Hierarchy for Imbalanced Medical Classification

EMNLP ICD Tree 2023

有监督对比学习将同一类别视为正样本,然而罕见病缺乏同一类别正样本

通过ICD树重新定义ICD标签间的相似度,从而将相似度高的常见病作为正样本,以缓解长尾病的数据缺失问题

同一类目的疾病的相似度>同一章节不同类目疾病的相似度>不同章节疾病间的相似度

img

通过注意力机制,利用父节点和相邻节点,来完善罕见病的表达

img