基于ALBERT与双向GRU的中医脏腑定位模型.pdf

文档介绍：工程科学学报，第 43 卷，第 9 期：1182−1189，2021 年 9 月
Chinese Journal of Engineering, Vol. 43, No. 9: 1182−1189, September 2021
/-; 
基于ALBERT与双向GRU的中医脏腑定位模型
张德政1,2)，范欣欣1,2)，谢永红1,2) ，蒋彦钊1,2)
苣
1) 北京科技大学计算机与通信工程学院，北京 100083    2) 材料领域知识工程北京市重点实验室，北京 100083
通信作者，E-mail: ******@
苣
摘    要    脏腑定位，即明确病变所在的脏腑，是中医脏腑辨证的重要阶段. 本文旨在通过神经网络模型搭建中医脏腑定位模
型，输入症状文本信息，输出对应的病变脏腑标签，为实现中医辅助诊疗的脏腑辨证提供支持. 将中医的脏腑定位问题建模
为自然语言处理中的多标签文本分类问题，基于中医的医案数据，提出一种基于预训练模型ALBERT和双向门控循环单元
（Bi-GRU）的脏腑定位模型. 对比实验和消融实验的结果表明，本文提出的方法在中医脏腑定位的问题上相比于多层感知机
模型、决策树模型具有更高的准确性，与Word2Vec文本表示方法相比，本文使用的ALBERT预训练模型的文本表示方法有
效提升了模型的准确率. 在模型参数上，ALBERT预训练模型相比BERT模型降低了模型参数量，有效减小了模型大小. 最
终，.
关键词    多标签文本分类；ALBERT；门控循环单元；脏腑定位；中医
分类号    
Localization model of traditional Chinese medicine Zang-fu based on ALBERT and Bi-
GRU
ZHANG De-zheng1,2)，FAN Xin-xin1,2)，XIE Yong-hong1,2) ，JIANG Yan-zhao1,2)
苣
1) School of Computer and Communication Engineering, University of Science and Technology Beijing, Beijing 100083, China
2) Beijing Key Laboratory of Knowledge Engineering for Materials Science, Beijing 100083, China
Corresponding author, E-mail: ******@
苣
ABSTRACT    The ra