计算机科学与探索2024，Vol.18Issue(8)：2130-2139,10.DOI:10.3778/j.issn.1673-9418.2309100
基于对话结构与图注意力网络的药物推荐算法
Medication Recommendation Algorithm Based on Dialogue Structure and Graph Attention Network
摘要
Abstract
Existing studies on medication recommendation are mostly based on the electronic health records.How-ever,these data are difficult to reflect the current health status of patients and learn the current health needs of pa-tients.Thus,this paper fuses the dialogue and disease information,and proposes a dialogue-based structure and graph attention network recommendation algorithm.Firstly,the grey relational analysis and graph attention network are integrated and the relations of nodes are measured using grey relational analysis to provide a novel relation-aware graph structure.It can improve the traditional graph networks to learn the node relations.Secondly,a dialogue hierarchical encoder is established to encode the representations of utterance and dialogue via the new graph struc-ture.Meanwhile,two graph structures are designed to learn the node correlation for generating the contextual dia-logue representation.Finally,the representations of disease by knowledge graph and graph network are incorporated with dialogue to predict and recommend.Experimental results show that the proposed algorithm is superior to base-lines in terms of all metrics.Compared with the best baseline DNN,the performance of the proposed algorithm im-proves 1.8%and 3.5%in terms of F1 and Jaccard,which shows that the proposed algorithm can improve the recom-mendation performance.关键词
药物推荐/对话结构/图注意力网络/灰关联分析Key words
medication recommendation/dialogue structure/graph attention network/grey relational analysis分类
信息技术与安全科学引用本文复制引用
陈江美,张文德,谭睿璞..基于对话结构与图注意力网络的药物推荐算法[J].计算机科学与探索,2024,18(8):2130-2139,10.基金项目
国家自然科学基金青年基金(61300104) （61300104）
福建省社会科学基金(FJ2023B041) （FJ2023B041）
福建省高校信息化学会科研课题(FJGX22007). This work was supported by the Youth Fund of the National Natural Science Foundation of China(61300104),the Social Science Foun-dation of Fujian Province(FJ2023B041),and the Research Project of College Education Informatization Institute of Fujian Province(FJGX22007). （FJGX22007）
