A medical text classification approach with ZEN and capsule network

Text classification is an important topic in natural language processing, with the development of social network, many question-and-answer pairs regarding health-care and medicine flood social platforms. It is of great social value to mine and classify medical text and provide targeted medical services for patients. The existing algorithms of text classification can deal with simple semantic text, especially in the field of Chinese medical text, the text structure is complex and includes a large number of medical nomenclature and professional terms, which are difficult for patients to understand. We propose a Chinese medical text classification model using a BERT-based Chinese text encoder by N-gram representations (ZEN) and capsule network, which represent feature uses the ZEN model and extract the features by capsule network, we also design a N-gram medical dictionary to enhance medical text representation and feature extraction. The experimental results show that the precision, recall and F1-score of our model are improved by 10.25%, 11.13% and 12.29%, respectively, compared with the baseline models in average, which proves that our model has better performance.

Automated summary

This paper introduces a Chinese medical text classification model using a BERT-based Chinese text encoder, N-gram representations, and a capsule network. The model extracts features using the capsule network and enhances medical text representation and feature extraction through the design of an N-gram medical dictionary. The experimental results demonstrate that the model outperforms the baseline models in terms of precision, recall, and F1-score.