Power fingerprint identification based on the improved V-I trajectory with color encoding and transferred CBAM-ResNet.

Journal: PloS one
Published Date:

Abstract

In power fingerprint identification, feature information is insufficient when using a single feature to identify equipment, and small load data of specific customers, difficult to meet the refined equipment classification needs. A power fingerprint identification based on the improved voltage-current(V-I) trajectory with color encoding and transferred CBAM-ResNet34 is proposed. First, the current, instantaneous power, and trajectory momentum information are added to the original V-I trajectory image using color coding to obtain a color V-I trajectory image. Then, the ResNet34 model was pre-trained using the ImageNet dataset and a new fully-connected layer meeting the device classification goal was used to replace the fully-connected layer of ResNet34. The Convolutional Block Attention Module (CBAM) was added to each residual structure module of ResNet34. Finally, Class-Balanced (CB) loss is introduced to reweight the Softmax cross-entropy (SM-CE) loss function to solve the problem of data imbalance in V-I trajectory identification. All parameters are retrained to extract features from the color V-I trajectory images for device classification. The experimental results on the imbalanced PLAID dataset verify that the method in this paper has better classification capability in small sample imbalanced datasets. The experimental results show that the method effectively improves the identification accuracy by 4.4% and reduces the training time of the model by 14 minutes compared with the existing methods, which meets the accuracy requirements of fine-grained power fingerprint identification.

Authors

  • Lin Lin
    Central Laboratory, The First Affiliated Hospital of Xiamen University, Xiamen, China, zhibinli33@163.com, liusuhuan@xmu.edu.cn.
  • Jie Zhang
    College of Physical Education and Health, Linyi University, Linyi, Shandong, China.
  • Xu Gao
    National Research Base of Intelligent Manufacturing Service, School of Management Science and Engineering, Chongqing Technology and Business University, Chongqing, 400067, China; Chongqing Water Group Co. Ltd., Chongqing, 400042, China. Electronic address: hughgao@outlook.com.
  • Jiancheng Shi
    Department of Radiology, Peking University Third Hospital, Yanqing Hospital, Beijing, China.
  • Cheng Chen
    Key Laboratory of Precision and Intelligent Chemistry, School of Chemistry and Materials Science, University of Science and Technology of China, China.
  • Nantian Huang
    School of Electrical Engineering, Northeast Dianli University, Jilin, China.