Image classification framework based on contrastive self-supervised learning

被引：0
|
作者：
Zhao H.-W. [1 ]
Zhang J.-R. [1 ]
Zhu J.-P. [2 ]
Li H. [1 ]
机构：
[1] College of Computer Science and Technology, Jilin University, Changchun
[2] Shanghai Zhengen Industrial Co., Ltd., Shanghai
来源：
Jilin Daxue Xuebao (Gongxueban)/Journal of Jilin University (Engineering and Technology Edition) | 2022年 / 52卷 / 08期
关键词：
computer application; contrastive learning; image classification; self-supervised learning;
D O I：
10.13229/j.cnki.jdxbgxb20210607
中图分类号：
学科分类号：
摘要：
In order to solve the problem that supervised learning needs a lot of time to complete data set annotation in the field of image classification, a self-supervised image classification framework, SSIC framework, is proposed. SSIC framework is a self supervised learning method based on contrastive learning, which has better performance than the existing unsupervised methods. A new framework is designed and a more effective pretext task is selected to improve the robustness of the model. In addition, a targeted loss function is proposed to improve the performance of image classification. experiments was conducted on UC Merced, NWPU and AID data sets. Experimental results show that SSIC framework has obvious advantages over the latest technology, and it also performs well in low resolution image classification. © 2022 Editorial Board of Jilin University. All rights reserved.
引用
收藏
页码：1850 / 1856
页数：6
相关论文
共 17 条
  • [11] Wu Z, Xiong Y, Yu S, Et al., Unsupervised feature learning via non-parametric instance discrimination [C], Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3733-3742, (2018)
  • [12] Yang Y, Newsam S., Bag-of-visual-words and spatial extensions for land-use classification, Proceedings of the 18th SIGSPATIAL International Conference on Advances in Geographic Information Systems, pp. 270-279, (2010)
  • [13] Cheng G, Han J, Lu X., Remote sensing image scene classification: benchmark and state of the art, Proceedings of the IEEE, 105, 10, pp. 1865-1883, (2017)
  • [14] Xia G S., AID: a benchmark data set for performance evaluation of aerial scene classification, IEEE Transactions on Geoscience and Remote Sensing, 55, 7, pp. 3965-3981, (2017)
  • [15] Qi B, Kun Q, Zhang H, Et al., APDC-Net: attention pooling-based convolutional network for aerial scene classification, IEEE Geoscience and Remote Sensing Letters, 15, 10, pp. 1603-1607, (2020)
  • [16] Sun H, Li S, Zheng X, Et al., Remote sensing scene classification by gated bidirectional network, IEEE Transactions on Geoscience and Remote Sensing, 58, 1, pp. 82-96, (2020)
  • [17] Tan M, Le Q V., EfficientNet: rethinking model scaling for convolutional neural networks, International Conference on Machine Learning, pp. 10691-10700, (2019)
12