| Literature DB >> 34976109 |
Ke Liang1,2, Sifan Wu3, Jiayi Gu4.
Abstract
Using natural language processing (NLP) technologies to develop medical chatbots makes the diagnosis of the patient more convenient and efficient, which is a typical application in healthcare AI. Because of its importance, lots of researches have come out. Recently, the neural generative models have shown their impressive ability as the core of chatbot, while it cannot scale well when directly applied to medical conversation due to the lack of medical-specific knowledge. To address the limitation, a scalable medical knowledge-assisted mechanism (MKA) is proposed in this paper. The mechanism is aimed at assisting general neural generative models to achieve better performance on the medical conversation task. The medical-specific knowledge graph is designed within the mechanism, which contains 6 types of medical-related information, including department, drug, check, symptom, disease, and food. Besides, the specific token concatenation policy is defined to effectively inject medical information into the input data. Evaluation of our method is carried out on two typical medical datasets, MedDG and MedDialog-CN. The evaluation results demonstrate that models combined with our mechanism outperform original methods in multiple automatic evaluation metrics. Besides, MKA-BERT-GPT achieves state-of-the-art performance.Entities:
Mesh:
Year: 2021 PMID: 34976109 PMCID: PMC8718312 DOI: 10.1155/2021/5294627
Source DB: PubMed Journal: Comput Math Methods Med ISSN: 1748-670X Impact factor: 2.238
Figure 1Framework of our scalable medical knowledge-assisted generative model, MKA. In the figure, the ellipsoids represent the modules inside of our method. The orange ellipsoids show the automatic and scalable medical knowledge generation module. The figure legend inside of medical knowledge subgraph is corresponding to the entity types and relation types shown in Tables 1 and 2.
Definition of entity types in medical knowledge graphs.
| Entity type | Description |
|---|---|
|
| Entities for the clinical departments |
|
| Entities for the diseases |
|
| Entities for the symptoms |
|
| Entities for the food |
|
| Entities for the drugs |
|
| Entities for the checks |
Definition of relation types in medical knowledge graphs.
| Relation type | Description |
|---|---|
|
| Relations between |
|
| Relations between |
|
| Relations between |
|
| Relations between |
|
| Relations between |
|
| Relations between |
Algorithm 1Pseudocode of the medical knowledge subgraph generator.
Algorithm 2Pseudocode of the topic detector.
Algorithm 3Pseudocode of the medical knowledge extractor.
Comparison of the models on MedDialog-CN, where the best results are in bold.
| Model | Dialog-GPT | Transformer | BERT-GPT | MKA-Transformer | MKA-BERT-GPT |
|---|---|---|---|---|---|
| Perplexity | 9.71 | 9.52 | 8.23 | 8.81 |
|
| BLEU-2 | 5.21% | 4.92% | 4.88% | 5.02% |
|
| BLEU-4 | 1.83% | 0.90% | 0.97% | 0.99% |
|
| NIST-2 | 0.36 | 0.42 | 0.40 | 0.43 |
|
| NIST-4 | 0.32 | 0.40 | 0.39 | 0.40 |
|
| METEOR | 12.32% | 13.11% | 12.83% | 13.4% |
|
| Entropy-4 | 13.73 | 13.51 | 13.8 | 13.72 |
|
| Dist-1 | 0.02% | 0.03% | 0.03% | 0.03% |
|
| Dist-2 | 2.01% | 2.02% | 2.14% | 2.11% |
|
Comparison of the models on MedDG, where the best results are in bold.
| Model | Dialog-GPT | Transformer | BERT-GPT | MKA-Transformer | MKA-BERT-GPT |
|---|---|---|---|---|---|
| Perplexity | 8.53 | 8.52 | 5.98 | 8.41 |
|
| BLEU-2 | 6.41% | 6.30% | 7.62% | 6.62% |
|
| BLEU-4 | 2.12% | 2.08% | 2.57% | 2.40% |
|
| NIST-2 | 0.38 | 0.37 | 0.42 | 0.39 |
|
| NIST-4 | 0.35 | 0.35 | 0.39 | 0.38 |
|
| METEOR | 13.78% | 14.32% | 16.25% | 14.88% |
|
| Entropy-4 | 10.56 | 10.17 |
| 10.28 | 13.37 |
| Dist-1 | 0.01% | 0.01% |
| 0.01% |
|
| Dist-2 | 1.72% | 1.67% |
| 1.69% |
|
Improvements of the models with MKA compared to the baseline model on MedDialog-CN and MedDG test sets.
| Dataset | Model | Perplexity | BLEU-2,4 | NIST-2,4 | METEOR | Entropy-4 | Dist-1,2 |
|---|---|---|---|---|---|---|---|
| MedDialog-CN | MKA-Transformer | -0.71 | 0.10%, 0.09% | 0.01, 0 | 0.29% | 0.21 | 0.00%, 0.09% |
| MKA-BERT-GPT | -0.19 | 0.83%, 0.38% | 0.04, 0.04 | 1.11% | 0.3 | 0.01%, 0.08% | |
|
| |||||||
| MedDG | MKA-Transformer | -0.11 | 0.32%, 0.32% | 0.02, 0.03 | 0.56% | 0.11 | 0.00%, 0.02% |
| MKA-BERT-GPT | -0.03 | 0.47%, 0.30% | 0.01, 0.02 | 0.38% | -0.01 | 0.00%, 0.00% | |
An example of generated responses on the MedDialog-CN test set.
| Patient self-report | 科室 department: 男科 andrology | |
|---|---|---|
| Patient question 1 | 我很怕是高位结扎手术引起的附睾炎, 希望大夫能给我说说是怎么回事, 该怎么处理? | |
|
| ||
| Doctor response 1 | Ground truth | 中医药调理。 |
| Dialog-GPT | 手术部位不会引发附睾炎。 | |
| Transformer | 可能是近期没有充分休息, 免疫力下降。建议早点休息。 | |
| BERT-GPT | 不要趴着睡觉, 影响伤口恢复。建议多休息。 | |
| MKA-Transformer | 可以吃营养的健康食品, 比如西兰花, 鸡蛋等, 同时注重卫生清洁, 每晚可以热敷缓解疼痛。 | |
| MKA-BERT-GPT | 不可以吃冷饮, 少吃辛辣刺激的食物, 影响伤口恢复。建议多吃蔬果类, 比如苹果。不要久坐不动, 少穿紧身裤。 | |
|
| ||
| Patient question 2 | 请问我这是附睾炎吗?或是什么?恳请大夫您为我指点!感谢。 | |
|
| ||
| Doctor response 2 | Ground truth | 局部气血不畅。 |
| Dialog-GPT | 放宽心, 只是心理作用。 | |
| Transformer | 只是术后伤口疼痛。 | |
| BERT-GPT | 常见血流不通问题。 | |
| MKA-Transformer | 可能是运动过量导致提睾肌过度收缩, 导致疼痛。 | |
| MKA-BERT-GPT | 血流不通, 建议过几天复查伤口恢复情况, 减轻心理负担。 | |
An example of generated responses on the MedDG test set.
| Patient self-report | 科室 department: 肛肠科 anorectal branch | |
|---|---|---|
| Patient question 1 | 医生你好, 最近我大便总是很干燥, 而且有好几次大便之后, 发现有血, 我这个是什么情况? | |
|
| ||
| Doctor response 1 | Ground truth | 你好, 可能存在痔疮出血!但还是建议您做个肠镜! |
| Dialog-GPT | 你好, 这种情况容易出现内痔 | |
| Transformer | 你好, 考虑是痔疮。 | |
| BERT-GPT | 你好, 可能存在痔疮, 大便干结多久了? | |
| MKA-Transformer | 你好, 可能是痔疮, 建议您做个肠镜。 | |
| MKA-BERT-GPT | 你好, 可能是痔疮出血。此外, 建议您做个肠镜。 | |
|
| ||
| Patient question 2 | 为什么我的大便是一粒一粒的, 很干燥, 像羊屎那样的? | |
|
| ||
| Doctor response 2 | Ground truth | 大便干结是功能性问题, 建议您多吃水果蔬菜, 养成排便习惯! |
| Dialog-GPT | 你好, 可能存在痔疮。 | |
| Transformer | 注意饮食, 多吃蔬菜水果。 | |
| BERT-GPT | 建议您多吃蔬菜水果。 | |
| MKA-Transformer | 考虑是肠胃功能性紊乱, 建议您多吃水果蔬菜。 | |
| MKA-BERT-GPT | 考虑是肠胃功能问题, 建议您多吃水果蔬菜, 比如梨, 香蕉。若还是这样的话建议您做个肠镜。 | |