DOI QR코드

DOI QR Code

Semi-supervised SAR Image Classification with Threshold Learning Module

임계값 학습 모듈을 적용한 준지도 SAR 이미지 분류

  • 도재준 (한국항공대학교 인공지능학과) ;
  • 김선옥 (한국항공대학교 소프트웨어학과)
  • Received : 2023.11.24
  • Accepted : 2023.12.11
  • Published : 2023.12.31

Abstract

Semi-supervised learning (SSL) is an effective approach to training models using a small amount of labeled data and a larger amount of unlabeled data. However, many papers in the field use a fixed threshold when applying pseudo-labels without considering the feature-wise differences among images of different classes. In this paper, we propose a SSL method for synthetic aperture radar (SAR) image classification that applies different thresholds for each class instead of using a single fixed threshold for all classes. We propose a threshold learning module into the model, considering the differences in feature distributions among classes, to dynamically learn thresholds for each class. We compare the application of a SSL SAR image classification method using different thresholds and examined the advantages of employing class-specific thresholds.

준지도 학습(Semi-supervised learning)은 소량의 라벨이 있는 데이터와 다량의 라벨이 없는 데이터를 사용하여 모델을 훈련하는 효과적인 방법이다. 그러나 많은 논문에서 준지도 학습시 하나의 고정된 임계값을 사용하여 각 클래스별 서로 다른 이미지들의 특징별 차이를 고려하지 않고 임의 라벨을 만든다. 본 논문에서는 합성개구 레이더(SAR) 이미지 분류 준지도 학습시 모든 클래스가 하나의 고정된 임계값을 사용하는 대신 각 클래스에 대해 서로 다른 임계값을 적용한다. 모델에 임계값 학습 모듈을 추가하여 임계값을 학습하여 클래스별로 학습되는 차이를 고려하여 클래스별로 서로 다른 임계값을 얻는다. 서로 다른 임계값을 사용한 준지도 학습기반의 SAR 이미지 분류 방법을 적용유무를 비교하여 클래스별 임계값을 사용하는 이점에 대해 고찰하였다.

Keywords

Acknowledgement

이 출판물은 2021년도 한국항공대학교 교비지원 연구비에 의하여 지원된 연구의 결과임.

References

  1. F, Pierre, et al. "Sharpness-aware minimization for efficiently improving generalization". arXiv preprint arXiv:2010.01412, 2020.
  2. M, Wortsman, et al. "Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time." International Conference on Machine Learning. PMLR, 2022.
  3. H. M. Kabir, "Reduction of Class Activation Uncertainty with Background Information." arXiv preprint arXiv:2305.03238, 2023.
  4. Z, Yang, et al. "SAR image classification method based on improved capsule network." Journal of Physics: Conference Series. Vol. 1693. No. 1. IOP Publishing, 2020.
  5. S, Chen, et al. "Target classification using the deep convolutional networks for SAR images." IEEE transactions on geoscience and remote sensing 54.8 4806-4817. 2016 https://doi.org/10.1109/TGRS.2016.2551720
  6. H, Ren, et al. "Extended convolutional capsule network with application on SAR automatic target recognition." Signal Processing 183 : 108021. 2021
  7. Q, Xie, et al. "Self-training with noisy student improves imagenet classification." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2020.
  8. E, Arazo, et al. "Pseudo-labeling and confirmation bias in deep semi-supervised learning." 2020 International Joint Conference on Neural Networks (IJCNN). IEEE, 2020.
  9. D, Berthelot, et al. "Mixmatch: A holistic approach to semi-supervised learning." Advances in neural information processing systems 32, 2019.
  10. D, Berthelot, et al. "Remixmatch: Semi-supervised learning with distribution alignment and augmentation anchoring." arXiv preprint arXiv:1911.09785, 2019.
  11. K, Sohn, Kihyuk, et al. "Fixmatch: Simplifying semi-supervised learning with consistency and confidence." Advances in neural information processing systems 33 : 596-608. 2020
  12. T, Chen, e t al. "A s imple framework for contrastive learning of visual representations." International conference on machine learning. PMLR, 2020.
  13. A, Anaby-Tavor, et al. "Do not have enough data? Deep learning to the rescue!." Proceedings of the AAAI Conference on Artificial Intelligence. Vol. 34. No. 05. 2020.
  14. B, Zhang, et al. "Censer: Curriculum semi-supervised learning for speech recognition based on self-supervised pre-training." arXiv preprint arXiv:2206.08189, 2022.
  15. V, Tsouvalas, et. al. "Federated self-training for semi-supervised audio recognition." ACM Transactions on Embedded Computing Systems 21.6 : 1-26. 2022 https://doi.org/10.1145/3520128
  16. M, Sajjadi, et. al. "Regularization with stochastic transformations and perturbations for deep semi-supervised learning." Advances in neural information processing systems 29, 2016.
  17. X, Wang, et, al. "Contrastive learning with stronger augmentations." IEEE Transactions on Pattern Analysis and Machine Intelligence. 2022.
  18. E, Cubuk et al. "Randaugment: Practical automated data augmentation with a reduced search space." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops. 2020.
  19. ED, Cubuk, et al. "Autoaugment: Learning augmentation strategies from data." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2019.
  20. Y, Lei et, al. "Synthetic Images Augmentation for Robust SAR Target Recognition." 2021 The 5th International Conference on Video and Image Processing. 2021.
  21. X, Zhang, et al. "A Novel Data Augmentation Method for SAR Image Target Detection and Recognition." 2021 IEEE International Geoscience and Remote Sensing Symposium IGARSS. IEEE, 2021.
  22. M, Zhang, et al. "Data augmentation method of SAR image dataset." IGARSS 2018-2018 IEEE International Geoscience and Remote Sensing Symposium. IEEE, 2018.
  23. Xu, Yi, et al. "Dash: Semi-supervised learning with dynamic thresholding." International Conference on Machine Learning. PMLR, 2021.
  24. Liang, Zechen, et al. "ADT-SSL: Adaptive Dual-Threshold for Semi-Supervised Learning." arXiv preprint arXiv:2205.10571 (2022).
  25. Zhang, Bowen, et al. "Flexmatch: Boosting semi-supervised learning with curriculum pseudo labeling." Advances in Neural Information Processing Systems 34 (2021): 18408-18419.
  26. ER, Keydel, et. al. "MSTAR extended operating conditions: A tutorial." Algorithms for Synthetic Aperture Radar Imagery III 2757, 228-242, 1996. https://doi.org/10.1117/12.242059
  27. C, Coman, "A deep learning SAR target classification experiment on MSTAR dataset." 2018 19th international radar symposium (IRS). IEEE, 2018.