A Novel TCM Prescription Recommendation Method based on Attention Factorization Machines


  • Wenxuan Xu
  • Qingtao Wu




Attention Factorization Machines, Factorization Machines, Prescription Recommendation, Attention Network


A prescription recommendation algorithm for attention factorization machines is proposed in this study. This algorithm leverages the pair-wise interaction of factorization machines to capture the multi-category attributes of patients and prescriptions. Additionally, an attention network is incorporated into the factorization machine to assign higher weights to the effective features within the prescription. This enables the algorithm to discern the importance of different combinations of features in the prescription, thereby enhancing the recommendation performance of the model. Through extensive experimentation, it is observed that the prescription recommendation model based on attention factorization machines does not rely on manual features and exhibits commendable recommendation performance. Furthermore, it achieves a certain degree of individualized recommendation effect.


K. Chen, and X. U. Hao. The integration of traditional Chinese medicine and Western medicine. European Review. Vol. 11 (2003) No. 2, p. 225-235.

Y. Jaiswal, Z. Liang, Z. Zhao. Botanical drugs in Ayurveda and traditional Chinese medicine. Journal of ethnopharmacology. Vol. 194 (2016), p. 245-259.

L. Hua, Y. Gao, J. Zou, et al. Reflections on treatment of COVID-19 with traditional Chinese medicine. Chinese medicine Vol. 15.1 (2020) , p. 94.

S. X. Wang, Y. Wang, Y. B. Lu, et al. Diagnosis and treatment of novel coronavirus pneumonia based on the theory of traditional Chinese medicine. Journal of integrative medicine, Vol. 18 (2020) No. 4, p. 275-283.

J. C. Pan, Y. T. Tsai, J. N. Lai, et al. The traditional Chinese medicine prescription pattern of patients with primary dysmenorrhea in Taiwan: a large-scale cross sectional survey. Journal of ethnopharmacology. Elsevier. Vol. 152 (2014) No. 2, p. 314-319.

S. Wang, Y. Hu, W. Tan, et al. Compatibility art of traditional Chinese medicine: from the perspective of herb pairs. Journal of ethnopharmacology. Vol. 143 (2012) No. 2, p. 412-423.

C. Y. Ung, H. Li, Z. Cao, et al. Are herb-pairs of traditional Chinese medicine distinguishable from others? Pattern analysis and artificial intelligence classification study of traditionally defined herbal properties. Journal of Ethnopharmacology. Vol. 111 (2007) No. 2, p. 371-377.

F. Yu, T. Takahashi, J. Moriya, et al. Traditional Chinese medicine and Kampo: a review from the distant past for the future. Journal of International Medical Research. Vol. 34 (2006) No. 3, p. 231-239.

H. Chen, Y. He. Machine Learning Approaches in Traditional Chinese Medicine: A Systematic Review. The American journal of Chinese medicine. Vol. 50 (2022) No. 01, p. 91-131.

X. Chu, B. Sun, Q. Huang, S. Peng, Y. Zhou, Y. Zhang. Quantitative knowledge presentation models of traditional Chinese medicine (TCM): A review. Artificial intelligence in medicine. Elsevier. Vol. 103 (2020), p. 101810.

W. Li, Z. Yang, X. Sun. Exploration on Generating Traditional Chinese Medicine Prescription from Symptoms with an End-to-End method. In: Natural Language Processing and Chinese Computing: 8th CCF International Conference, NLPCC 2019, Dunhuang, China, October 9–14, 2019, Proceedings, Part I 8. Springer International Publishing, 2019. p. 486-498.

Y. Zhang, K. Hu, T. Yang, et al. Prescription recommendation algorithm for Traditional Chinese Medicine treatment of lung cancer based on complex networks. Journal of Traditional Chinese Medicine and National Medicine. Vol. 30 (2019) No. 5, p. 1257-1260.

J. Xiao, H. Ye, X. He, et al. Attentional factorization machines: Learning the weight of feature interactions via attention networks. arXiv preprint arXiv:1708.04617 (2017).

L. Yao, Y. Zhang, B. Wei, et al. A Topic Modeling Approach for Traditional Chinese Medicine Prescriptions. IEEE Transactions on Knowledge & Data Engineering. 2018, p. 1-1.

X. Dong, Y. Zheng, Z. Shu, et al. TCMPR: TCM Prescription recommendation based on subnetwork term mapping and deep learning. In 2021 IEEE International Conference on Bioinformatics and Biomedicine (BIBM). 2012, p. 3776-3783.

Z. Gong, N. Zhang, J. He. KGRN: Knowledge Graph Relational Path Network for Target Prediction of TCM Prescriptions. In International Conference on Intelligent Computing, 2021, p. 148-161.

T. Liang, L. Zheng, L. Chen, et al. Multi-view factorization machines for mobile app recommendation based on hierarchical attention. Knowledge-Based Systems. Elsevier. Vol. 187 (2020), p. 104821.

P. Wen, W. Yuan, Q. Qin, et al. Neural attention model for recommendation based on factorization machines. Applied Intelligence. Springer. Vol. 51(2021) No. 4, p. 1829-1844.

Z. Wang, J. Ma, Y. Zhang, et al. Attention-over-Attention Field-Aware Factorization Machine. Proceedings of the AAAI Conference on Artificial Intelligence. Vol. 34 (2020) No. 04, p. 6323-6330.







How to Cite

Xu, W., & Wu, Q. (2023). A Novel TCM Prescription Recommendation Method based on Attention Factorization Machines. Journal of Computing and Electronic Information Management, 10(3), 55-61. https://doi.org/10.54097/jceim.v10i3.8682

Similar Articles

1-10 of 72

You may also start an advanced similarity search for this article.