Domain Specific Pre-training Methods for Traditional Chinese Medicine Prescription Recommendation.

Wei Li,Zheng Yang,Yanqiu Shao
DOI: https://doi.org/10.1007/978-981-99-9119-8_12
2024-01-01
Abstract:Traditional Chinese Medicine (TCM) is an important constituent of medical treatment. During the development history of TCM, there have been a large number of medical records accumulated, which embody the experiential judgement of the TCM practitioners. There are usually the symptoms observed by the practitioner and the according treatment methods within the records. In the treatment procedure, TCM practitioners often refer to the classical records and the prescriptions within them, which makes recommending prescriptions from the records based on the observation of the symptoms valuable in practice. Based on these observations, we propose to model this problem as a matching based recommendation task. To precisely model the relation between symptoms and prescriptions, inspired by the success of pre-trained language models, we propose a TCM domain specific hybrid input construction method and multi-grained negative sampling methods and training objectives. To verify the effectiveness of the proposed method, we conduct extensive experiments on the symptom-prescription dataset. The experiment results show that our proposed method can accurately recommend suitable prescriptions with more abundant candidates for the reference of TCM practitioners, making it more valuable in practice.
What problem does this paper attempt to address?