Frontiers of Information Technology & Electronic Engineering
>> 2016,
Volume 17,
Issue 9
doi:
10.1631/FITEE.1500346
Article
Two-level hierarchical feature learning for image classification
. College of Computer Science and Technology, Zhejiang University, Hangzhou 310027, China.. Ningbo Institute of Technology, Zhejiang University, Ningbo 315100, China.. College of Science and Technology, Ningbo University, Ningbo 315100, China
Available online: 2016-10-08
Next
Previous
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.
京公网安备 11010502051620号
