Neural network models have been extensively applied in various domains, such as autonomous driving, facial recognition, and machine translation. However, deploying neural networks in scenarios with limited computing power faces challenges such as inefficient training and slow convergence. Therefore, we propose a Lightweight Data Selection Network (LDSN) tailored for constrained computational environments for image classification. Specifically, LDSN employs a scoring model to swiftly identify samples suitable for training the lightweight classification network. It initializes a lightweight scoring network to score and rank samples from the original dataset, then utilizes high-scoring samples to train the classification network, and updates the scoring network according to validation accuracy and model loss. Besides, to address the discrete nature of sample selection, we introduce the policy gradient algorithm for training the scoring model. Finally, we employ a Limited-memory Broyden-Fletcher-Goldfarb-Shanno (LBFGS) optimization algorithm to reduce training epochs of the classification network, thereby enhancing training velocity. Experimental results on standard image datasets demonstrate the effectiveness and efficiency of LDSN.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Lightweight Data Selection Network for Image Classification via Policy Gradient

  • Yang Yu,
  • Xinqi Du,
  • Zhejian Yang,
  • Sinuo Zhang,
  • Hechang Chen,
  • Bohao Qu,
  • Jifeng Hu

摘要

Neural network models have been extensively applied in various domains, such as autonomous driving, facial recognition, and machine translation. However, deploying neural networks in scenarios with limited computing power faces challenges such as inefficient training and slow convergence. Therefore, we propose a Lightweight Data Selection Network (LDSN) tailored for constrained computational environments for image classification. Specifically, LDSN employs a scoring model to swiftly identify samples suitable for training the lightweight classification network. It initializes a lightweight scoring network to score and rank samples from the original dataset, then utilizes high-scoring samples to train the classification network, and updates the scoring network according to validation accuracy and model loss. Besides, to address the discrete nature of sample selection, we introduce the policy gradient algorithm for training the scoring model. Finally, we employ a Limited-memory Broyden-Fletcher-Goldfarb-Shanno (LBFGS) optimization algorithm to reduce training epochs of the classification network, thereby enhancing training velocity. Experimental results on standard image datasets demonstrate the effectiveness and efficiency of LDSN.