Traditional Chinese Medicine (TCM) Diagnosis Model Building Based on Multi-label Classification

¹ School of Traditional Chinese Medicine, Beijing University of Chinese Medicine, 100029, Beijing China
² School of Computer Science, Beijing University of Posts and Telecommunications, 100876, Beijing China

^a Corresponding author: yanzheng@bupt.edu.cn
^a Corresponding author: liyuhang@bucm.edu.cn

Abstract

In the study, we propose a TCM diagnosis model that can be used for multi-label classification and give clear diagnosis, as well as the basis for diagnosis and differentiation when the symptoms correspond to multiple diseases or syndromes. The implementation of the model is divided into three steps. Firstly, choose the machine learning algorithm to train the TCM diagnosis model. The features of the training data are symptoms and the labels are diseases or syndromes. Secondly, give the number α (α>1, α∈Z⁺), the model will output the diagnoses with the top α highest probability according to the input symptoms as candidate diagnoses. Finally, the rules of differential diagnosis are designed to determine which candidate diagnoses should be reserved, thereby complete the multi-label classification. In our test dataset, by 10-fold cross-validation, the average accuracy of the single label classification was 0.882; the average precision was 0.974; the average recall was 1.000; the average f1 score was 0.967; the average accuracy of the multi-label classification was 0.706; the average micro precision was 0.934; the average micro recall was 0.941 and the average hamming loss was 0.060. Through the test we can know that this model had a good potential for auxiliary decision making in clinical diagnosis and treatment.