中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Practical Block-Wise Neural Network Architecture Generation

文献类型:会议论文

作者Zhao Zhong1; Junjie Yan2; Wei Wu2; Jing Shao2; Cheng-Lin Liu1
出版日期2018
会议日期18-23 June 2018
会议地点Salt Lake City, UT, USA
英文摘要

Convolutional neural networks have gained a remarkable success in computer vision. However, most usable network architectures are hand-crafted and usually require expertise and elaborate design. In this paper, we provide a block-wise network generation pipeline called BlockQNN which automatically builds high-performance networks using the Q-Learning paradigm with epsilon-greedy exploration strategy. The optimal network block is constructed by the learning agent which is trained sequentially to choose component layers. We stack the block to construct the whole auto-generated network. To accelerate the generation process, we also propose a distributed asynchronous framework and an early stop strategy. The block-wise generation brings unique advantages: (1) it performs competitive results in comparison to the hand-crafted state-of-the-art networks on image classification, additionally, the best network generated by BlockQNN achieves 3.54% top-1 error rate on CIFAR-10 which beats all existing auto-generate networks. (2) in the meanwhile, it offers tremendous reduction of the search space in designing networks which only spends 3 days with 32 GPUs, and (3) moreover, it has strong generalizability that the network built on CIFAR also performs well on a larger-scale ImageNet dataset.

源URL[http://ir.ia.ac.cn/handle/173211/23562]  
专题自动化研究所_模式识别国家重点实验室_模式分析与学习团队
通讯作者Cheng-Lin Liu
作者单位1.中科院自动化所
2.SenseTime
推荐引用方式
GB/T 7714
Zhao Zhong,Junjie Yan,Wei Wu,et al. Practical Block-Wise Neural Network Architecture Generation[C]. 见:. Salt Lake City, UT, USA. 18-23 June 2018.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。