Uddamvathanak, Rom, Feng, Yang, Xulei, Yang, Das, Ankit Kumar, Shen, Yan, Salahuddin, Mohamed, Hussain, Shaista, and Chawla, Shailey (2018) Two-stage ensemble of Deep Convolutional Neural Networks for object recognition. In: Proceedings of the IEEE International Conference on Intelligent Rail Transportation. From: ICIRT 2018: IEEE International Conference on Intelligent Rail Transportation, 12-14 December 2018, Singapore.

PDF (Published Version) - Published Version Restricted to Repository staff only

Abstract

Deep Convolutional Neural Networks (CNNs) have been widely used for object recognition in images. However, due to the complexity in setting the structures and related parameters for deep CNNs, performance of individual CNN models may vary largely and lack robustness. In realizing this problem, this paper proposed a two-stage ensemble of deep CNNs for object recognition. To increase model diversity for ensemble, multiple basic CNN models/structures are first pre-defined. For each basic CNN model, multiple rounds of training are conducted based on sub-sampling the training dataset. In the first stage of ensemble, multiple outputs from each basic CNN model is integrated using the MinMax median. This is followed by the second stage of ensemble to combine the outputs from all basic CNN models. The proposed method was implemented and experiments were carried out on Kaggle's 'Statoil/C-CORE Iceberg Classifier Challenge' image data for iceberg and ship recognition, as well as on 'Northeastern University surface defect dataset' for surface defect classification problem. The experimental results showed that the proposed ensemble method outperformed the individual CNN models, and achieved the-state-of-the-art performance as compared to the best submission to the Kaggle's challenge using CNNs.

Item ID:	56205
Item Type:	Conference Item (Research - E1)
ISBN:	978-1-5386-7528-1
Keywords:	Convolutional Neural Networks (CNNs); two-stage ensemble; object recognition; image classification
Date Deposited:	26 Mar 2019 00:52
FoR Codes:	46 INFORMATION AND COMPUTING SCIENCES > 4603 Computer vision and multimedia computation > 460306 Image processing @ 50% 46 INFORMATION AND COMPUTING SCIENCES > 4611 Machine learning > 461104 Neural networks @ 50%
SEO Codes:	89 INFORMATION AND COMMUNICATION SERVICES > 8902 Computer Software and Services > 890202 Application Tools and System Utilities @ 100%
Downloads:	Total: 6
	More Statistics