Unsupervised Contrastive Learning for Automatic Grouping of Aerial Swarms

Leyan Li, Rennong Yang, Huanyu Li, Maolong Lv*, Longfei Li Yue, Ao Wu, Li Mo*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

The self-organization capabilities of massive aerial swarms pose challenges to conventional methods of grouping aerial targets. These traditional approaches struggle with issues such as stability, reliability, and the recognition of deep intentions. To overcome these challenges, we propose an architecture called Unsupervised Contrastive Learning for Aerial Targets Grouping (UCL-ATG). Our approach utilizes random time periods within the scenario as training batches for contrastive learning. The UCL-ATG model consists of four key modules: feature extraction, time series generation, fine-tuning, and clustering. The feature extraction and time series generation modules form the network architecture of Contrastive Predictive Coding (CPC). Positive and negative samples are obtained using different sampling methods. To effectively collaborate with the clustering module, we design a contrastive predictive loss function specifically tailored for clustering. This enables the extraction of low-dimensional representations that capture the high-dimensional temporal characteristics of aircraft. Furthermore, we introduce a real-time fine-tuner to enhance the model's transferability to specific tasks. The inclusion of the real-time fine-tuner greatly alleviates problems associated with evolving confrontation styles and unevenly distributed datasets, as confirmed by the performance on the validation set. Extensive comparative experimental results demonstrate our model's superior training outcomes and its ability to extract improved clustering features. In wargame applications, our model no longer relies solely on static position information of the aircraft. It exhibits outstanding capabilities in historical information memory, information synthesis, and even demonstrates aptitude in identifying friend or foe and making tactical inferences.

Original languageEnglish
Pages (from-to)6249-6258
Number of pages10
JournalIEEE Transactions on Vehicular Technology
Volume73
Issue number5
DOIs
Publication statusPublished - 1 May 2024

Keywords

  • Unmanned aerial vehicle
  • aerial swarms grouping
  • situational awareness
  • unsupervised contrastive learning

Fingerprint

Dive into the research topics of 'Unsupervised Contrastive Learning for Automatic Grouping of Aerial Swarms'. Together they form a unique fingerprint.

Cite this