基于TCMTF的中草药推荐方法研究

摘要

【目的】以中草药推荐任务作为切入点，针对现有中草药推荐模型忽略中医相关理论知识，导致效果不佳、推荐处方偏离实际等问题，提出了一种基于改进Transformer的中草药推荐模型。【方法】在症状序列信息挖掘模块加入卷积神经网络，拓展模型多分支结构，并以训练好的症状文本向量作为嵌入向量来挖掘相关信息，提升模型深层次特征结合能力，提高推荐质量。提出一种多特征融合注意力机制，降低注意力累加过程中被过高关注的特征维度，使得累加结果可以关注到未被关注区域，降低推荐中草药重复概率。提出一种熵平滑损失，通过减弱严格顺序对结果造成的影响进一步改善推荐结果。【结果】模型在一个公共临床中医处方数据集和一个合作医院私有数据集上进行实验，实验结果表明与Herb-Know、TCM Translator等其他中草药推荐模型相比，该模型可有效提高推荐中草药质量，改善该过程中重复问题。消融实验也证明了各个模块的有效性。

关键词

中草药推荐 / Transformer / 熵平滑损失 / 多特征融合注意力机制

基金资助

国家自然科学基金（U21A20469,61872261,61972274）