期刊
SWARM AND EVOLUTIONARY COMPUTATION
卷 49, 期 -, 页码 62-74出版社
ELSEVIER
DOI: 10.1016/j.swevo.2019.05.010
关键词
Particle swarm optimization; Deep neural networks; Convolutional neural networks; Image classification
资金
- National Council for Scientific and Technological Development (CNPq, Brazil) [203076/2015-0]
- National Science Foundation (NSF, USA) [ACI-1548562, OCI-1126330]
- Bridges system - National Science Foundation (NSF, USA) at the Pittsburgh Supercomputing Center (PSC) [ACI-1445606]
Deep neural networks have been shown to outperform classical machine learning algorithms in solving real-world problems. However, the most successful deep neural networks were handcrafted from scratch taking the problem domain knowledge into consideration. This approach often consumes very significant time and computational resources. In this work, we propose a novel algorithm based on particle swarm optimization (PSO), capable of fast convergence when compared with others evolutionary approaches, to automatically search for meaningful deep convolutional neural networks (CNNs) architectures for image classification tasks, named psoCNN. A novel directly encoding strategy and a velocity operator were devised allowing the optimization use of PSO with CNNs. Our experimental results show that psoCNN can quickly find good CNN architectures that achieve quality performance comparable to the state-of-the-art designs.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据