Discovering Regularities from Traditional Chinese Medicine Prescriptions via Bipartite Embedding Model

Discovering Regularities from Traditional Chinese Medicine Prescriptions via Bipartite Embedding Model

Chunyang Ruan, Jiangang Ma, Ye Wang, Yanchun Zhang, Yun Yang

Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence
Main track. Pages 3346-3352. https://doi.org/10.24963/ijcai.2019/464

Regularities analysis for prescriptions is a significant task for traditional Chinese medicine (TCM), both in inheritance of clinical experience and in improvement of clinical quality. Recently, many methods have been proposed for regularities discovery, but this task is challenging due to the quantity, sparsity and free-style of prescriptions. In this paper, we address the specific problem of regularities discovery and propose a graph embedding based framework for regularities discovery for massive prescriptions. We model this task as a relation prediction in which the correlation of two herbs or of herb and symptom are incorporated to characterize the different relationships. Specifically, we first establish a heterogeneous network with herbs and symptoms as its nodes. We develop a bipartite embedding model termed HS2Vec to detect regularities, which explores multiple relations of herbherb, and herb-symptom based on the heterogeneous network. Experiments on four real-world datasets demonstrate that the proposed framework is very effective for regularities discovery.
Keywords:
Machine Learning: Data Mining
Machine Learning: Unsupervised Learning
Machine Learning Applications: Bio;Medicine