基于持续学习的中医舌色苔色协同分类方法
Collaborative Classification Method of TCM Tongue Color and Coating Color Based on Continual Learning
-
摘要: 中医(traditional Chinese medicine, TCM)舌诊客观化研究中需要分析的舌象特征很多, 不同的舌象特征往往采用单独的方法进行分析, 导致分析系统的整体实现复杂度大幅增加。为此, 基于持续学习的思想, 提出一种中医舌色苔色协同分类方法, 该方法将舌色分类作为旧任务, 将苔色分类作为新任务, 充分利用2个任务的相似性和相关性, 仅通过一个网络结构就同时实现舌色和苔色的准确分类。首先, 设计一种基于全局-局部混合注意力机制(global local hybrid attention, GLHA)的双分支网络结构, 将网络高层语义特征与低层特征相融合, 提升特征的表达能力; 然后, 提出基于正则化和回放相结合的持续学习策略, 使得该网络在学习新任务知识的同时有效防止对旧任务知识的遗忘。在2个自建的中医舌象特征分析数据集上的实验结果表明, 提出的协同分类方法可以获得与单个任务相当的分类性能, 同时可以将2个分类任务的整体复杂度降低一半左右。其中, 舌色分类准确率分别达到93.92%和92.97%, 精确率分别达到93.69%和92.87%, 召回率分别达到93.96%和93.16%;苔色分类准确率分别达到90.17%和90.26%, 精确率分别达到90.05%和90.17%, 召回率分别达到90.24%和90.29%。Abstract: There are many characteristics of tongue that need to be analyzed in traditional Chinese medicine (TCM). Different characteristics are often analyzed by individual methods, which significantly increases the overall implementation complexity of the analysis system. Therefore, this paper proposes a collaborative classification method of tongue color and coating color in TCM based on continual learning. This method takes tongue color classification as an old task and coating color classification as a new task, which makes full use of the similarity and relevance of the two tasks to realize the accurate classification of tongue color and coating color simultaneously under a single network framework. First, a dual branch network structure with global local hybrid attention (GLHA) mechanism was designed, which aggregates high-level semantic features with low-level features to improve the representative capability of features. Second, a continual learning strategy based on the combination of regularization and rehearsal was proposed, which made the network effectively prevent forgetting the knowledge learned from old task while learning new task. The experimental results on two self-established TCM tongue datasets show that, the proposed collaborative classification method can achieve a comparable classification performance with a single task, and simultaneously, reduce the overall complexity of the two classification tasks by almost half. Among them, the accuracy of tongue color classification reaches 93.92% and 92.97%, the precision reaches 93.69% and 92.87%, the recall reaches 93.96% and 93.16%, respectively. While that of the coating color classification reaches 90.17% and 90.26%, the precision reaches 90.05% and 90.17%, the recall reaches 90.24% and 90.29%, respectively.