Swarm Intelligence Based Ensemble Learning of Deep Neural Networks

被引:0
|
作者
Li, Tao
Ma, Jinwen [1 ]
机构
[1] Peking Univ, Sch Math Sci, Dept Informat Sci, Beijing 100871, Peoples R China
关键词
Deep learning; Ensemble learning; Swarm intelligence;
D O I
10.1007/978-3-030-36808-1_28
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Ensemble learning is a powerful tool in machine learning and it is very significant to utilize the mechanism of ensemble learning to improve the performance of deep learning for recognition and classification tasks. In this paper, we propose a general ensemble learning framework of deep neural networks based on swarm intelligence algorithms for solving the combination coefficients to the outputs of these component deep neural networks. We formulate the weights assigning problem for the deep neural networks as an optimization problem whose objective function is highly complicated and use swarm intelligence algorithms to solve it. We evaluate this ensemble learning framework on two real-world datasets, Market-1501 for person re-identification and CIFAR for image classification, and show that it outperforms a single deep neural network remarkably.
引用
收藏
页码:256 / 264
页数:9
相关论文
共 50 条