[1]逯鹏,王汉章,毛晓波,等.基于卷积自编码器网络的脉搏波分类模型[J].郑州大学学报(工学版),2021,42(05):56-61.
点击复制

基于卷积自编码器网络的脉搏波分类模型()
分享到:

《郑州大学学报(工学版)》[ISSN:1671-6833/CN:41-1339/T]

卷:
42
期数:
2021年05期
页码:
56-61
栏目:
出版日期:
2021-09-10

文章信息/Info

Title:
Pulse Wave Classification Model ba<x>sed on Convolutional Auto-Encoder
作者:
逯鹏王汉章毛晓波赵宇平刘超尚莉伽孙智霞
文献标志码:
A
摘要:
基于深度学习的脉搏波分类依赖大量标记数据,临床数据有限、标注成本高影响了脉搏波分类识别效果。设计一种基于卷积自编码器网络(Convolutional Autoencoder Networks,CAE-Net)的脉搏波分类模型,首先,利用卷积神经网络(Convolutional Neural Network,CNN)局部特征提取能力和自编码器(Autoencoder,AE)的压缩重构及降维特性构建卷积自编码器(CAE),结合脉搏波波形特点,在平均绝对误差损失函数中引入脉搏波时域特征约束,提升CAE对脉搏波低维特征自学习能力;其次,重用预训练CAE编码层网络和权重,构建卷积自编码器网络(CAE-Net),并利用有标记脉搏波数据进行网络微调。通过在心血管疾病数据集上实验表明,CAE-Net的分类准确率为98.00%,F1值达到94.40%,相较于其它脉搏波分类模型,设计网络所提取特征的区分度较高,同时减少了对标注脉搏波数据依赖,在小样本脉搏波数据具有较好分类效果。
Abstract:
Classification of pulse wave ba<x>sed on deep learning relies on a large number of labeled data, however, limited clinical data and expensive labeling costs hinder the pulse wave classification and recognition. A pulse wave classification model ba<x>sed on convolutional autoencoder networks (CAE-net) is designed in this paper, firstly, the convolutional autoencoder (CAE) is constructed, which combines the local feature extraction ability of convolutional neural network (CNN) and the compression reconstruction and dimension reduction characteristics of autoencoder (AE), and considering the characteristics of pulse wave, the time domain feature constraint of pulse wave is introduced into the mean absolute error loss function to improve the self-learning ability of CAE for low dimensional features Secondly, the CAE-Net is constructed by reusing the coding la<x>yer network and weights of the pre-training CAE, then the network is fine tuned by using labeled pulse waves. Experiments on cardiovascular disease dataset show that the classification accuracy of CAE-Net is 98%, and the F1 value is 94.40%. Compared with other classification models, the designed network can extract features with high discrimination, reduces the dependence on the labeled pulse waves, and performs well in the classification of small sample pulse wave data.
更新日期/Last Update: 2021-10-11