没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
在某些图像分类任务中,不同类别之间的相似性是不同的,并且样本通常被误分类为高度相似的类别。 为了区分高度相似的类别,需要更具体的功能,以便分类器可以提高分类性能。 在本文中,我们提出了一种基于深度卷积神经网络(CNN)的新颖的两级分层特征学习框架,该框架简单有效。 首先,使用转移学习方法训练不同级别的深度特征提取器,该方法将预训练的深度CNN模型朝新的目标数据集进行微调。 其次,将从所有类别中提取的一般特征和从高度相似的类别中提取的特定特征融合到特征向量中。 然后将最终的特征表示输入线性分类器。 最后,使用Caltech-256,Oxford Flower-102和塔斯马尼亚珊瑚点数(CPC)数据集进行的实验表明,通过两级分层特征学习获得的深层特征的表达能力很强大。 与平面多重分类法相比,我们提出的方法有效地提高了分类精度。
资源推荐
资源详情
资源评论
Song et al. / Front Inform Technol Electron Eng 2016 17(9):897-906 897
Frontiers of Information Technology & Electronic Engineering
www.zju.edu.cn/jzus; engineering.cae.cn; www.springerlink.com
ISSN 2095-9184 (print); ISSN 2095-9230 (online)
E-mail: jzus@zju.edu.cn
Two-level hierarchical feature learning for
image classification
∗
Guang-hui SONG
1,2
, Xiao-gang JIN
†‡1
,Gen-langCHEN
2
,YanNIE
3
(
1
College of Computer Science and Technology, Zhejiang University, Hangzhou 310027, China)
(
2
Ningbo Institute of Technology, Zhejiang University, Ningb o 315100, China)
(
3
College of Science and Technology, Ningbo University, Ningb o 315100, China)
†
E-mail: xiaogangj@cise.zju.edu.cn
Received Oct. 20, 2015; Revision accepted Apr. 10, 2016; Crosschecked Aug. 8, 2016
Abstract: In some image classification tasks, similarities among different categories are different and the samples
are usually misclassified as highly similar categories. To distinguish highly similar categories, more specific features
are required so that the classifier can improve the classification performance. In this paper, we propose a novel
two-level hierarchical feature learning framework based on the deep convolutional neural network (CNN), which is
simple and effective. First, the deep feature extractors of different levels are trained using the transfer learning
method that fine-tunes the pre-trained deep CNN model toward the new target dataset. Second, the general feature
extracted from all the categories and the specific feature extracted from highly similar categories are fused into a
feature vector. Then the final feature representation is fed into a linear classifier. Finally, experiments using the
Caltech-256, Oxford Flower-102, and Tasmania Coral Point Count (CPC) datasets demonstrate that the expression
ability of the deep features resulting from two-level hierarchical feature learning is powerful. Our proposed method
effectively increases the classification accuracy in comparison with flat multiple classification methods.
Key words: Transfer learning, Feature learning, Deep convolutional neural network, Hierarchical classification,
Spectral clustering
http://dx.doi.org/10.1631/FITEE.1500346 CLC number: TP391.4
1 Introduction
The deep convolutional neural network (CNN)
has achieved impressive classification performance in
the ImageNet benchmark (Krizhevsky et al., 2012).
Surprisingly, transfer learning methods based on the
deep convolutional feature trained on a generic recog-
nition task are also successful in various computer
vision tasks, such as object classification, domain
adaptation, and scene recognition. They achieve
results superior to those of the previous meth-
‡
Corresponding author
*
Project supported by the National Natural Science Foundation
of China (No. 61379074) and the Zhejiang Provincial Natural Sci-
ence Foundation of China (Nos. LZ12F02003 and LY15F020035)
OR CID: Xiao-gang JIN, http://orcid.org/0000-0002-7787-7228
c
Zhejiang University and Springer-Verlag Berlin Heidelberg 2016
ods (Donahue et al., 2014; Zeiler and Fergus, 2014;
Cai et al., 2015). Therefore, the feature learning
ability of deep CNN has received considerable at-
tention. In previous studies, deep CNN models were
used as feature extractors but not as classifiers, and
they provided a way to obtain more specific visual
features (Yosinski et al., 2014).
At present, most deep CNN models serve as
flat end-to-end classifiers for image recognition tasks.
These deep models take the raw image as the network
input, extract image features using back-propagation
through layers of convolutional filters, and finally
output the categorized results using a softmax out-
put layer. However, the reality is that image datasets
have a growing sample size and image category. Simi-
larities are different among different categories, with
898 Song et al. / Front Inform Technol Electron Eng 2016 17(9):897-906
some categories being more difficult to distinguish
than others. For example, let us classify three fruit
varieties, i.e., apples, oranges, and bananas. The
first two varieties are clearly difficult to distinguish,
whereas the bananas are easy to separate from the
other two. A similar situation exists in many image
datasets. Fig. 1 shows that the same border colors
and line types represent the highly similar categories
in the Oxford Flower-102 dataset.
Camellia
Mallow
Coltsfoot
Barberton daisy
Fig. 1 The same border colors and line types rep-
resenting the highly similar categories in the Oxford
Flower-102 dataset. Distinguishing between camellia
and mallow is difficult. The situation is similar be-
tween coltsfoot and barberton daisy. References to
color refer to the online version of this figure
Inspired by the idea of transfer learning, we have
come up with two questions: first, whether the gen-
eral feature extractor trained on all fruit varieties can
be adequate for distinguishing the categories with
high similarity, and second, how to extract more
specific features using the feature learning method
to improve the classification performance. Adopt-
ing the hierarchical classification and feature learn-
ing methods to solve the above problem is natural.
Hierarchical classification methods usually organize
categories into a tree hierarchy (Deng et al., 2011;
Srivastava and Salakhutdinov, 2013). In this way
the parent node can select different hand-designed
feature descriptors according to the concrete con-
ditions of child nodes for the classifier. In trans-
fer learning with feature learning ability, the gen-
eral features are first learned on the root node, and
the specific features are further obtained from the
parent nodes for the highly similar categories. Hier-
archical classification methods have the advantages
of high efficiency and scalability, but they can only
achieve a trade-off between accuracy and efficiency
(Zhao et al., 2011; Liu et al., 2013). In particular,
too many levels produce error accumulation, which
seriously affects the classification accuracy. We in-
tend to find a way to use the hierarchical feature
learning ability while preventing the accumulation of
errors, leading to a decrease in accuracy. However, so
far, only limited studies have focused on combining
deep feature learning with hierarchical classification
to improve classification accuracy (Yan et al., 2015).
In this paper we propose a novel two-level hier-
archical feature learning framework, which involves
three steps. First, the general feature extractor is
trained by fine-tuning a pre-trained deep CNN model
on the target dataset. At the same time, the similar-
ity among different categories is obtained by process-
ing the confusion matrix. Second, the highly similar
categories are divided into the same parent node,
and the specific feature extractor is trained sepa-
rately on each parent node. Finally, the general and
specific deep features are fused into a feature vector
used for the final linear classifier. The experiments
using the Caltech-256, Oxford Flower-102, and Tas-
mania Coral Point Count (CPC) datasets demon-
strate that transfer learning methods gradually en-
hance the expression ability of deep features. Our
proposed method can effectively improve the classi-
fication accuracy based on the deep CNN models on
these three datasets.
2 Related work
2.1 Deep convolutional neural network and
feature learning
The deep CNN model has been widely used
because of its better classification performance. It
is composed of five convolutional layers (conv1–
conv5), two fully connected layers (fc6 and fc7),
and a softmax output layer (Krizhevsky et al., 2012).
This network model has strong feature learning
ability and can generate feature representations by
learning low-level features in the early convolu-
tional layers and transforming them to high-level
semantic features in the latter convolutional layers
剩余9页未读,继续阅读
资源评论
weixin_38614377
- 粉丝: 2
- 资源: 945
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 【java毕业设计】高仿小米电子商城项目模板源码(springboot+vue+mysql+说明文档).zip
- 5G建设和AI技术推动下,中证5G通信ETF的投资价值探讨
- Python项目之淘宝模拟登录.zip
- 课程设计项目:python+QT实现的小型编译器.zip
- (源码)基于AVR ATmega644的智能卡AES解密系统.zip
- (源码)基于C++插件框架的计算与打印系统.zip
- (源码)基于Spring Boot和Vue的苍穹外卖管理系统.zip
- (源码)基于wxWidgets库的QMiniIDE游戏开发环境管理系统.zip
- 通过C++实现原型模式(Prototype Pattern).rar
- 学习记录111111111111111111111111
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功