使用随机层冻结和特征细化策略的快速迁移学习方法

Fast Transfer Learning Method Using Random Layer Freezing and Feature Refinement Strategy

IEEE Transactions on Cybernetics · 2024
被引 8
ABS 3

中文导读

提出一种基于随机层冻结和Moore-Penrose逆的快速重训练策略,在预训练深度卷积神经网络上实现高效迁移学习,收敛速度比传统方法快约1.5倍。

Abstract

Recently, Moore-Penrose inverse (MPI)-based parameter fine-tuning of fully connected (FC) layers in pretrained deep convolutional neural networks (DCNNs) has emerged within the inductive transfer learning (ITL) paradigm. However, this approach has not gained significant traction in practical applications due to its stringent computational requirements. This work addresses this issue through a novel fast retraining strategy that enhances applicability of the MPI-based ITL. Specifically, during each retraining epoch, a random layer freezing protocol is utilized to manage the number of layers undergoing feature refinement. Additionally, this work incorporates an MPI-based approach for refining the trainable parameters of FC layers under batch processing, contributing to expedited convergence. Extensive experiments on several ImageNet pretrained benchmark DCNNs demonstrate that the proposed ITL achieves competitive performance with excellent convergence speed compared to conventional ITL methods. For instance, the proposed strategy converges nearly 1.5 times faster than retraining the ImageNet pretrained ResNet-50 using stochastic gradient descent with momentum (SGDM).

迁移学习深度学习卷积神经网络计算机视觉