基于迁移学习的全连接神经网络舌象分类方法

被引:17
作者
杨晶东
张朋
机构
[1] 上海理工大学光电信息与计算机工程学院自主机器人实验室
基金
上海市自然科学基金;
关键词
人工智能; 迁移学习; 深度学习; 舌象; 卷积神经网络;
D O I
10.16781/j.0258-879x.2018.08.0897
中图分类号
R241.25 [舌诊]; TP183 [人工神经网络与计算];
学科分类号
100505 ; 081104 ; 0812 ; 0835 ; 1405 ;
摘要
目的针对深度学习在舌象分类中训练数据量大、训练设备要求高、训练时间长等问题,提出一种基于迁移学习的全连接神经网络小样本舌象分类方法。方法应用经Image Net海量数据集训练后的卷积Inceptionv3网络提取舌象点、线等有效特征,再使用全连接神经网络对特征进行训练分类,将深度学习网络学习到的图像知识迁移到舌象识别任务中。利用舌象数据集进行训练、测试。结果与典型舌象分类方法 K最近邻(KNN)算法、支持向量机(SVM)算法和卷积神经网络(CNN)深度学习方法相比,本实验使用的两种方法(Inceptionv3+2NN和Inceptionv3+3NN)具有较高的舌象分类识别率,准确率分别达90.30%和93.98%,且样本训练时间明显缩短。结论与KNN算法、SVM算法和CNN深度学习方法相比,基于迁移学习的全连接神经网络舌象分类方法可有效提高舌象分类的准确率、缩短网络的训练时间。
引用
收藏
页码:897 / 902
页数:6
相关论文
共 5 条
[1]   基于卷积神经网络迁移学习的SAR图像目标分类 [J].
刘晨 ;
曲长文 ;
周强 ;
李智 ;
李健伟 .
现代雷达, 2018, 40 (03) :38-42
[2]   舌诊图像点刺和瘀点的识别与提取 [J].
王昇 ;
刘开华 ;
王丽婷 .
计算机工程与科学, 2017, 39 (06) :1126-1132
[3]   基于神经网络集成的舌苔分类方法 [J].
刘关松 ;
徐建国 ;
高敦岳 .
计算机工程, 2003, (14) :100-102
[4]   ImageNet Large Scale Visual Recognition Challenge [J].
Russakovsky, Olga ;
Deng, Jia ;
Su, Hao ;
Krause, Jonathan ;
Satheesh, Sanjeev ;
Ma, Sean ;
Huang, Zhiheng ;
Karpathy, Andrej ;
Khosla, Aditya ;
Bernstein, Michael ;
Berg, Alexander C. ;
Fei-Fei, Li .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 115 (03) :211-252
[5]  
Deep learning in neural networks: An overview[J] . Jürgen Schmidhuber.Neural Networks . 2014