Computer Science ›› 2018, Vol. 45 ›› Issue (12): 255-261.doi: 10.11896/j.issn.1002-137X.2018.12.042

Special Issue: Medical Imaging

• Graphics, Image & Pattern Recognition • Previous Articles     Next Articles

Classification of Tongue Image Based on Multi-task Deep Convolutional Neural Network

TANG Yi-ping, WANG Li-ran, HE Xia, CHEN Peng, YUAN Gong-ping   

  1. (School of Information Engineering,Zhejiang University of Technology,Hangzhou 310023,China)
  • Received:2017-11-08 Online:2018-12-15 Published:2019-02-25

Abstract: It is difficult to exploit the existing methods to achieve efficient classification and identification of tongue ima-ge’ labels in parallel,and it is also difficult to utilize the correlation between labels for comprehensive analysis.Aiming at the problems above,this paper proposed a classification method of tongue image based on multi-task deep convolutional neural network and constructed a multi-task joint learning model based on deep convolutional neural network to realize the simultaneous identification of tongue color,moss color,tongue crack and tooth marks in tongue diagnosis of Chinese medicine.First,the shared network layer is used to learn all labels,and the correlation between the tags is extracted and utilized automatically from the perspective of feature extraction.Then,the learning tasks of specific labels are completed in different sub-network layers to eliminate the ambiguity in the multi-label classification problem.Finally,multiple Softmax classifiers are trained to achieve parallel prediction of all labels.Experimental results suggest that the proposed method can simultaneous extract multiple features of tongue image and implement classification by means of end to end.The lowest value is about 0.96 in several evaluation indexes and the multi-task recognition rate is about 34ms.Therefore,this algorithm has obvious advantages in accuracy and speed.

Key words: Correlation, Multi-label, Multi-task network, Tongue classification, Transfer learning

CLC Number: 

  • TP391.41
[1]LEE J,KIM H,KIM N R,et al.An approach for multi-labelclassification by directed acyclic graph with label correlation maximization[J].Information Sciences,2016,351(C):101-114.
[2]WU J,YE C,SHENG V S,et al.Active learning with label correlation exploration for multi-label image classification[J].Iet Computer Vision,2017,11(7):577-584.
[3]LI S L,LIU R,LIU H.Multi-Label Learning for Improved RBF Neural Networks[J].Computer Science,2015,42(4):316-320.(in Chinese)
李书玲,刘蓉,刘红.改进型RBF神经网络的多标签算法研究[J].计算机科学,2015,42(4):316-320.
[4]XU X D,YAO M H,LIU H W,et al.Pre-processing Method of Multi-label Classification Based on kNN[J].Computer Science,2015,42(5):106-108,131.(in Chinese)
徐晓丹,姚明海,刘华文,等.基于kNN的多标签分类预处理方法[J].计算机科学,2015,42(5):106-108,131.
[5]ZHANG Y F,HU G Q,ZHANG X F,et al.An algorithm study on tongue color recognition of patients[J].Beijing Biomedical Engineering,2016,35(1):7-11.(in Chinese)
张艺凡,胡广芹,张新峰,等.基于支持向量机的痤疮患者舌色苔色识别算法研究[J].北京生物医学工程,2016,35(1):7-11.
[6]WANG Y G,YANG J,ZHOU Y,et al.Tongue Image Color Recognition in Traditional Chinese Medicine[J].Journal of Biomedical Engineering,2005,22(6):1116-1120.(in Chinese)
王永刚,杨杰,周越,等.中医舌象颜色识别的研究[J].生物医学工程学杂志,2005,22(6):1116-1120.
[7]YANG Z H,ZHANG D P,LI N M.Kernel False-Colour Transformation and Line Extraction for Fissured Tongue Image[J].Journal of Computer-Aided Design and Computer Graphics,2010,22(5):771-776.(in Chinese)
杨朝辉,张大鹏,李乃民.裂纹舌图像的核假彩色变换及其纹线提取[J].计算机辅助设计与图形学学报,2010,22(5):771-776.
[8]ZHUMU L M,LU P,XIA C M,et al.Research on Douglas-Peucker Method in Feature Extraction from 55 Cases of Tooth-Marked Tongue Image[J].Chinese Archives of Traditional Chinese Medicine,2014,32(9):2138-2140.(in Chinese)
朱穆朗玛,陆萍,夏春明,等.基于道格拉斯-普克法提取55例齿痕舌图像特征研究[J].中华中医药学刊,2014,32(9):2138-2140.
[9]LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single shotmultibox detector[C]∥Proceedings of European Conference on Computer Vision.Amsterdam:Springer International Publi-shing,2016:21-37.
[10]JONATHAN L,EVAN S,TREVOR D.Fully convolutional networks for semantic segmentation[C]∥Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Boston:IEEE,2015:3431-3440.
[11]SUN Y,WANG X G,TANG X O.Deep Learning Face Representation from Predicting 10,000 Classes[C]∥Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Columbus:IEEE,2014:1891-1898.
[12]GIRSHICK R,DONAHUE J,DARRELL T,et al.Rich feature hierarchies for accurate object detection and semantic segmentation[C]∥Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Columbus:IEEE,2014:580-587.
[13]GIRSHICK R.Fast R-CNN[C]∥Proceedings of the IEEE Conference on International Conference on Computer Vision.Boston:IEEE,2015:1440-1448.
[14]REN S Q,HE K M,GIRSHICK R,et al.Faster R-CNN:Towards Real-Time Object Detection with Region Proposal Networks[J].IEEE Transactions on Pattern Analysis & Machine Intelligence,2017,39(6):1137-1149.
[15]HOOCHANG S,ROTH H R,GAO M,et al.Deep convolutional neural networks for computer-aided detection:CNN architectures,dataset characteristics and transfer learning[J].IEEE Transactions on Medical Imaging,2016,35(5):1285-1298.
[16]DONAHUE J,JIA Y Q,VINYALS O,et al.DeCAF:A Deep Convolutional Activation Feature for Generic Visual Recognition[C]∥Proceedings of the 31st International Conference on Machine Learning.2014:I-647-I-655.
[17]RAZAVIAN A S,AZIZPOUR H,SULLIVAN J,et al.CNN features off-the-shelf:An Astounding Baseline for Recognition[C]∥2014 IEEE Conference on Computer Vision and Pattern Recognition Workshops(CV PRW).2014:512-519.
[18]YOSINSKI J,CLUNE J,BENGIO Y,et al.How transferableare features in deep neural networks?[C]∥Proceedings of the IEEE Conference on Neural Information Processing Systems.Montreal:MIT Press,2014:3320-3328.
[19]DENG J,DONG W,SOCHER R,et al.ImageNet:A large-scale hierarchical image database[C]∥Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Miami:IEEE,2009:248-255.
[20]KAN H X,ZHANG L Y,DONG C W.A Tongue Image Recognition Method Based on Type 2 Diabetes Traditional Chinese Medicine Syndrome Classification[J].Chinese Journal of Biomedical Engineering,2016,35(6):658-664.(in Chinese)
阚红星,张璐瑶,董昌武.一种2型糖尿病中医证型的舌图像识别方法[J].中国生物医学工程学报,2016,35(6):658-664.
[1] LIU Jie-ling, LING Xiao-bo, ZHANG Lei, WANG Bo, WANG Zhi-liang, LI Zi-mu, ZHANG Hui, YANG Jia-hai, WU Cheng-nan. Network Security Risk Assessment Framework Based on Tactical Correlation [J]. Computer Science, 2022, 49(9): 306-311.
[2] WU Hong-xin, HAN Meng, CHEN Zhi-qiang, ZHANG Xi-long, LI Mu-hang. Survey of Multi-label Classification Based on Supervised and Semi-supervised Learning [J]. Computer Science, 2022, 49(8): 12-25.
[3] FANG Yi-qiu, ZHANG Zhen-kun, GE Jun-wei. Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning [J]. Computer Science, 2022, 49(8): 70-77.
[4] CHEN Ying, HAO Ying-guang, WANG Hong-yu, WANG Kun. Dynamic Programming Track-Before-Detect Algorithm Based on Local Gradient and Intensity Map [J]. Computer Science, 2022, 49(8): 150-156.
[5] SHEN Xiang-pei, DING Yan-rui. Multi-detector Fusion-based Depth Correlation Filtering Video Multi-target Tracking Algorithm [J]. Computer Science, 2022, 49(8): 184-190.
[6] YANG Xiao, WANG Xiang-kun, HU Hao, ZHU Min. Survey on Visualization Technology for Equipment Condition Monitoring [J]. Computer Science, 2022, 49(7): 89-99.
[7] WU Su-jie, ZHOU Jie, WANG Xue-ying, LYU Zhi-kang, SHAO Gen-fu. Study on Characteristics of Millimeter-wave MIMO Channel in Rainfall Environment [J]. Computer Science, 2022, 49(7): 297-303.
[8] WANG Jun-feng, LIU Fan, YANG Sai, LYU Tan-yue, CHEN Zhi-yu, XU Feng. Dam Crack Detection Based on Multi-source Transfer Learning [J]. Computer Science, 2022, 49(6A): 319-324.
[9] ZHU Xu-dong, XIONG Yun. Study on Multi-label Image Classification Based on Sample Distribution Loss [J]. Computer Science, 2022, 49(6): 210-216.
[10] PENG Yun-cong, QIN Xiao-lin, ZHANG Li-ge, GU Yong-xiang. Survey on Few-shot Learning Algorithms for Image Classification [J]. Computer Science, 2022, 49(5): 1-9.
[11] HAN Hong-qi, RAN Ya-xin, ZHANG Yun-liang, GUI Jie, GAO Xiong, YI Meng-lin. Study on Cross-media Information Retrieval Based on Common Subspace Classification Learning [J]. Computer Science, 2022, 49(5): 33-42.
[12] ZHAO Geng, WANG Chao, MA Ying-jie. Study on PAPR Reduction Based on Correlation of Chaotic Sequences [J]. Computer Science, 2022, 49(5): 250-255.
[13] TAN Zhen-qiong, JIANG Wen-Jun, YUM Yen-na-cherry, ZHANG Ji, YUM Peter-tak-shing, LI Xiao-hong. Personalized Learning Task Assignment Based on Bipartite Graph [J]. Computer Science, 2022, 49(4): 269-281.
[14] SUN Lin, HUANG Miao-miao, XU Jiu-cheng. Weak Label Feature Selection Method Based on Neighborhood Rough Sets and Relief [J]. Computer Science, 2022, 49(4): 152-160.
[15] ZUO Jie-ge, LIU Xiao-ming, CAI Bing. Outdoor Image Weather Recognition Based on Image Blocks and Feature Fusion [J]. Computer Science, 2022, 49(3): 197-203.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!