File(s) not publicly available

Particle swarm optimization based approach for finding optimal values of convolutional neural network parameters

conference contribution
posted on 27.03.2019, 00:00 by Toshi SinhaToshi Sinha, Ali HaidarAli Haidar, Brijesh VermaBrijesh Verma
Convolutional Neural Networks (CNNs) have demonstrated great potential in complex image classification problems in past few years. CNNs have a large number of parameters and the system accuracy depends directly on the selection of these parameters. With diverse parameters, selection of optimal parameter remains a trial and error, ad hoc or expert's mercy. In practice, optimal parameter selection remains the biggest obstacle in designing a real-world application using CNN. Convolutional neural network's performance is highly affected by its parameters. A novel approach is proposed in this paper to select convolutional neural network parameters in an image classification task. The proposed approach incorporated particle swarm optimization to select the parameters of the convolutional network. Two datasets, one benchmark CIFAR-IO and one real world application dataset, road-side vegetation dataset, were selected to evaluate the proposed approach. It is demonstrated that proposed approach efficiently explores the solution space, and determines the best combination of parameters. Extensive experiments, along with the statistical tests, revealed that proposed approach is an effective technique for automatically optimizing CNN's parameters. © 2018 IEEE.

History

Start Page

1

End Page

6

Number of Pages

6

Start Date

08/07/2018

Finish Date

13/07/2018

ISBN-13

9781509060177

Location

Rio de Janeiro, Brazil

Publisher

IEEE

Place of Publication

Piscataway, NJ

Peer Reviewed

Yes

Open Access

No

Author Research Institute

Centre for Intelligent Systems

Era Eligible

Yes

Name of Conference

2018 IEEE Congress on Evolutionary Computation (CEC 2018)