一种数据增强和多模型集成的细粒度分类算法 |
| |
作者姓名: | 蒋 杰 熊昌镇 |
| |
作者单位: | 北方工业大学城市道路交通智能控制技术北京市重点实验室,北京 100144 |
| |
基金项目: | 国家重点研发计划资助项目(2016YFB1200402);北京市教委科技创新服务能力建设项目(PXM2017-014212-000033,PXM2017-014212-000031) |
| |
摘 要: | 针对解决数据缺少和单个卷积网络模型性能的限制造成细粒度分类准确率不高的问
题,提出了一种数据增强和多模型集成融合的分类算法。首先通过镜像、旋转、多尺度缩放、高
斯噪声、随机剪切和色彩增强6 种变换对CompCars 数据集进行增强处理,然后采用差异化采样
数据集的方法训练CaffeNet、VGG16 和GoogleNet 3 种差异化的网络。然后采用多重集成的方法
集成多种模型的输出结果。实验中测试网络结构在不同数据增强算法和不同模型集成下的分类结
果。模型集成的分类准确率达到94.9%,比最好的单GoogleNet 模型的分类精确率提高了9.2 个
百分点。实验结果表明该算法可以有效地提高分类的准确率。
|
关 键 词: | 细粒度分类 数据增强 卷积网络 集成学习 |
Data Augmentation with Multi-Model Ensemble for Fine-Grained Category Classification |
| |
Authors: | JIANG Jie XIONG Changzhen |
| |
Affiliation: | Beijing Key Laboratory of Urban Intelligent Control, North China University of Technology, Beijing 100144, China |
| |
Abstract: | In order to solve low classification precision caused by the lack of training data or the
classification performance constraint of single convolutional network model, a fine-grained category
classification algorithm based on data augmentation and multi-model ensemble is proposed. Firstly, the
paper designs a variety of data augmentation methods to increase the number of pictures in CompCars
dataset, including mirroring, rotation, multiscale scaling, Gaussian noise, random cropping and color
enhancement. Then 3 differentiated models, CaffeNet, VGG16 and GoogleNet, are trained using the
constructed differentiated dataset by different data sampling. A multi-layer ensemble learning method is
used to integrate multi-model’s classification results. The experimental results show the fine-grained
classification of the differentiated convolution network trained on the different datasets generated by the
different data augmentation method. The experiment also shows the classification results of multi-model
ensemble with different ensemble strategy. The final classification precision of multi-model ensemble is
94.9%. Compared with the best single model GoogleNet, the classification precision is increased 9.2%.
The results verify the effectives of proposed algorithm. |
| |
Keywords: | fine-grained category classification data augmentation convolutional network ensemble learning |
|
| 点击此处可从《》浏览原始摘要信息 |
|
点击此处可从《》下载全文 |