基于迁移学习的全连接神经网络舌象分类方法
CSTR:
作者:
作者单位:

上海理工大学光电信息与计算机工程学院,上海理工大学光电信息与计算机工程学院

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(61374039),上海市自然科学基金(15ZR1429100),沪江基金(C14002).


Tongue image classification method based on transfer learning and fully connected neural network
Author:
Affiliation:

University of Shanghai for Science and Technology,University of Shanghai for Science and Technology

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    目的 针对深度学习在舌象分类中训练数据量大、训练设备要求高、训练时间长等问题,提出一种基于迁移学习的全连接神经网络小样本舌象分类方法。方法 应用经ImageNet海量数据集训练后的卷积Inception_v3网络提取舌象点、线等有效特征,再使用全连接神经网络对特征进行训练分类,将深度学习网络学习到的图像知识迁移到舌象识别任务中。利用舌象数据集进行训练、测试。结果 与典型舌象分类方法K最近邻(KNN)算法、支持向量机(SVM)算法和卷积神经网络(CNN)深度学习方法相比,本实验使用的两种方法(Inception_v3+2NN和Inception_v3+3NN)具有较高的舌象分类识别率,准确率分别达90.30%和93.98%,且样本训练时间明显缩短。结论 与KNN算法、SVM算法和CNN深度学习方法相比,基于迁移学习的全连接神经网络舌象分类方法可有效提高舌象分类的准确率、缩短网络的训练时间。

    Abstract:

    Objective To propose a classification method for small sample tongue images based on transfer learning and fully connected neural network, so as to solve the problems of large amount of data, high requirement of training equipment and long training time of deep learning in the classification of tongue images.Methods Effective features such as tongue points and lines of tongue images were extracted by the convolution Inception_v3 network after training on the massive data set of ImageNet. The above features were classified by the fully connected neural network, and the image knowledge acquired by the deep learning network was transferred to the tongue image recognition task, and then the tongue data set were used to train and test the efficiency of the network.Results Compared with the typical tongue image classification method such as K-nearest neighbor (KNN) algorithm, support vector machine (SVM) algorithm and convolutional neural network (CNN) deep learning method, the two methods (Inception_v3+2NN and Inception_v3+3NN) in our experiment had higher classification rates for tongue images, with the accuracy rates being 90.30% and 93.98%, respectively, and had shorter training time for the sample.Conclusion Compared with KNN algorithm, SVM algorithm and CNN deep learning method, the tongue image classification method based on transfer learning and fully connected neural network can effectively improve the accuracy rate of tongue image classification and shorten the training time.

    参考文献
    相似文献
    引证文献
相关视频

分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2018-06-29
  • 最后修改日期:2018-08-02
  • 录用日期:2018-09-07
  • 在线发布日期: 2018-09-07
  • 出版日期:
文章二维码
重要通知
友情提醒: 近日发现论文正式见刊或网络首发后,有人冒充我刊编辑部名义给作者发邮件,要求添加微信,此系诈骗行为!可致电编辑部核实:021-81870792。
            《海军军医大学学报》编辑部
关闭