Qibo: A Large Language Model for Traditional Chinese Medicine

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Heyi, Wang, Xin, Meng, Zhaopeng, Chen, Zhe, Zhuang, Pengwei, Jia, Yongzhe, Xu, Dawei, Guo, Wenbin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917701218205696
author Zhang, Heyi
Wang, Xin
Meng, Zhaopeng
Chen, Zhe
Zhuang, Pengwei
Jia, Yongzhe
Xu, Dawei
Guo, Wenbin
author_facet Zhang, Heyi
Wang, Xin
Meng, Zhaopeng
Chen, Zhe
Zhuang, Pengwei
Jia, Yongzhe
Xu, Dawei
Guo, Wenbin
contents Large Language Models (LLMs) has made significant progress in a number of professional fields, including medicine, law, and finance. However, in traditional Chinese medicine (TCM), there are challenges such as the essential differences between theory and modern medicine, the lack of specialized corpus resources, and the fact that relying only on supervised fine-tuning may lead to overconfident predictions. To address these challenges, we propose a two-stage training approach that combines continuous pre-training and supervised fine-tuning. A notable contribution of our study is the processing of a 2GB corpus dedicated to TCM, constructing pre-training and instruction fine-tuning datasets for TCM, respectively. In addition, we have developed Qibo-Benchmark, a tool that evaluates the performance of LLM in the TCM on multiple dimensions, including subjective, objective, and three TCM NLP tasks. The medical LLM trained with our pipeline, named $\textbf{Qibo}$, exhibits significant performance boosts. Compared to the baselines, the average subjective win rate is 63%, the average objective accuracy improved by 23% to 58%, and the Rouge-L scores for the three TCM NLP tasks are 0.72, 0.61, and 0.55. Finally, we propose a pipline to apply Qibo to TCM consultation and demonstrate the model performance through the case study.
format Preprint
id arxiv_https___arxiv_org_abs_2403_16056
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Qibo: A Large Language Model for Traditional Chinese Medicine
Zhang, Heyi
Wang, Xin
Meng, Zhaopeng
Chen, Zhe
Zhuang, Pengwei
Jia, Yongzhe
Xu, Dawei
Guo, Wenbin
Computation and Language
Artificial Intelligence
Large Language Models (LLMs) has made significant progress in a number of professional fields, including medicine, law, and finance. However, in traditional Chinese medicine (TCM), there are challenges such as the essential differences between theory and modern medicine, the lack of specialized corpus resources, and the fact that relying only on supervised fine-tuning may lead to overconfident predictions. To address these challenges, we propose a two-stage training approach that combines continuous pre-training and supervised fine-tuning. A notable contribution of our study is the processing of a 2GB corpus dedicated to TCM, constructing pre-training and instruction fine-tuning datasets for TCM, respectively. In addition, we have developed Qibo-Benchmark, a tool that evaluates the performance of LLM in the TCM on multiple dimensions, including subjective, objective, and three TCM NLP tasks. The medical LLM trained with our pipeline, named $\textbf{Qibo}$, exhibits significant performance boosts. Compared to the baselines, the average subjective win rate is 63%, the average objective accuracy improved by 23% to 58%, and the Rouge-L scores for the three TCM NLP tasks are 0.72, 0.61, and 0.55. Finally, we propose a pipline to apply Qibo to TCM consultation and demonstrate the model performance through the case study.
title Qibo: A Large Language Model for Traditional Chinese Medicine
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2403.16056