Competitive and Collaborative Learning Accelerates the Convergence of Deep Convolutional Neural Networks

Yanbin Dang, Yuliang Yang*, Yueyun Chen, Mengyu Zhu, Dehui Yin

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In the training of convolutional neural networks (CNNs), the layer-by-layer learning based on the backpropagation (BP) algorithm causes that in each round of weights update, the learning of the latter layer determines the learning of the former layer, while the former layer cannot directly affect the latter layer. This means that the flow of error information is unidirectional, causing non-cooperative learning between layers, thereby reducing the convergence speed of the networks. In this work, we propose a network structure that evaluates the relative contribution of each layer in the CNNs to the final output error. During training, it indirectly realizes the bidirectional flow of information between layers, achieving the purpose of cross-layer collaborative learning. Our algorithm also fuses features at different scales on the detection networks, which we call the flexible feature fusion network(FFN). On public datasets, we have conducted rich experiments. With the help of FFN, the convergence speed of the object detection model is greatly improved. Without pre-training weight initialization, the convergence speed of the model is approximately doubled.

Original languageEnglish
Title of host publication2022 7th International Conference on Cloud Computing and Big Data Analytics, ICCCBDA 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages431-438
Number of pages8
ISBN (Electronic)9781665487115
DOIs
Publication statusPublished - 2022
Externally publishedYes
Event7th International Conference on Cloud Computing and Big Data Analytics, ICCCBDA 2022 - Chengdu, China
Duration: 22 Apr 202224 Apr 2022

Publication series

Name2022 7th International Conference on Cloud Computing and Big Data Analytics, ICCCBDA 2022

Conference

Conference7th International Conference on Cloud Computing and Big Data Analytics, ICCCBDA 2022
Country/TerritoryChina
CityChengdu
Period22/04/2224/04/22

Keywords

  • collaborative learning
  • multi-scale feature fusion
  • object detection
  • training efficiency

Fingerprint

Dive into the research topics of 'Competitive and Collaborative Learning Accelerates the Convergence of Deep Convolutional Neural Networks'. Together they form a unique fingerprint.

Cite this