论文标题
综合征意识到的草药推荐与多掌卷卷积网络
Syndrome-aware Herb Recommendation with Multi-Graph Convolution Network
论文作者
论文摘要
草药建议在中药(TCM)的治疗过程中起着至关重要的作用,该过程旨在推荐一组草药来治疗患者的症状。尽管已经开发了用于药草建议的几种机器学习方法,但它们仅在建模草药和症状之间的相互作用方面受到限制,而忽略了综合征诱导的中间过程。进行TCM诊断时,经验丰富的医生通常会诱导患者症状的综合征,然后根据诱导综合征建议草药。因此,我们认为诱导综合征是症状的总体描述,对于草药推荐很重要,应适当处理。但是,由于综合征诱导的歧义和复杂性,大多数处方都缺乏综合症的明确基础真理。在本文中,我们提出了一种新方法,该方法将隐式综合征诱导过程考虑到Herb建议。考虑到一系列需要治疗的症状,我们旨在通过有效融合集合中所有症状的嵌入来产生整体综合征代表,以模仿医生如何诱导综合症。为了嵌入学习症状,我们还从输入处方中构建了症状伴随图形,以捕获症状之间的关系;然后,我们在症状 - 症状和症状 - 草皮图上构建图形卷积网络(GCN),以学习症状嵌入。同样,我们在草药草和症状 - 草皮图上构建了草药图形,并在学习草药嵌入中构建GCN,最终与综合征表示相互作用以预测草药的分数。这样,可以获得更全面的表示。我们在公共TCM数据集上进行了广泛的实验,显示出对最先进的草药推荐方法的显着改善。
Herb recommendation plays a crucial role in the therapeutic process of Traditional Chinese Medicine(TCM), which aims to recommend a set of herbs to treat the symptoms of a patient. While several machine learning methods have been developed for herb recommendation, they are limited in modeling only the interactions between herbs and symptoms, and ignoring the intermediate process of syndrome induction. When performing TCM diagnostics, an experienced doctor typically induces syndromes from the patient's symptoms and then suggests herbs based on the induced syndromes. As such, we believe the induction of syndromes, an overall description of the symptoms, is important for herb recommendation and should be properly handled. However, due to the ambiguity and complexity of syndrome induction, most prescriptions lack the explicit ground truth of syndromes. In this paper, we propose a new method that takes the implicit syndrome induction process into account for herb recommendation. Given a set of symptoms to treat, we aim to generate an overall syndrome representation by effectively fusing the embeddings of all the symptoms in the set, to mimic how a doctor induces the syndromes. Towards symptom embedding learning, we additionally construct a symptom-symptom graph from the input prescriptions for capturing the relations between symptoms; we then build graph convolution networks(GCNs) on both symptom-symptom and symptom-herb graphs to learn symptom embedding. Similarly, we construct a herb-herb graph and build GCNs on both herb-herb and symptom-herb graphs to learn herb embedding, which is finally interacted with the syndrome representation to predict the scores of herbs. In this way, more comprehensive representations can be obtained. We conduct extensive experiments on a public TCM dataset, showing significant improvements over state-of-the-art herb recommendation methods.