Research on Domain Adaptive Classification Based on Gradient Weight Pursuit
DOI:
CSTR:
Author:
Affiliation:

1.School of Automation, Nanjing University of Information Science and Technology;2.Nanjing University of Information Science and Technology

Clc Number:

Fund Project:

Science and Technology Innovation 2030- "New generation of Artificial Intelligence" major project(No.2018AAA0100400),The National Natural Science Foundation of China (No.U21B2049,No.61936005)

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In this paper, we propose a pruning and optimization algorithm (GWP) based on gradient weight Pursuit to solve the overfitting problem in the unsupervised domain, that is, the accuracy of downstream tasks is much lower than that of training sets. The dense-sparse-dense strategy is applied to solve the overfitting problem for the difference and adversarial adaptive methods in the unsupervised domain. The network is trained intensively and which connections are important are learned. The second is the pruning stage. Different from the pruning process in the original dense-sparse-dense strategy, the optimization algorithm in this paper considers both weight and gradient. On the one hand, weight information (i.e. zero-order information) is used, and on the other hand, the influence of gradient information (i.e. first-order information) on the network pruning process is also considered. In the final intensive phase, the pruned connections are restored and the intensive network is retrained with a smaller learning rate. Finally, the obtained network achieves ideal results in downstream tasks. The experimental results show that the proposed GWTA can effectively improve the accuracy of downstream tasks and has a plug-and-play effect compared with the original difference-based and adversarial domain adaptive methods.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:September 27,2023
  • Revised:February 15,2024
  • Adopted:February 25,2024
  • Online:
  • Published:
Article QR Code

Address:No. 219, Ningliu Road, Nanjing, Jiangsu Province

Postcode:210044

Phone:025-58731025