TY - JOUR
T1 - Unsupervised Contrastive Learning for Automatic Grouping of Aerial Swarms
AU - Li, Leyan
AU - Yang, Rennong
AU - Li, Huanyu
AU - Lv, Maolong
AU - Yue, Longfei Li
AU - Wu, Ao
AU - Mo, Li
N1 - Publisher Copyright:
© 1967-2012 IEEE.
PY - 2024/5/1
Y1 - 2024/5/1
N2 - 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.
AB - 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.
KW - Unmanned aerial vehicle
KW - aerial swarms grouping
KW - situational awareness
KW - unsupervised contrastive learning
UR - http://www.scopus.com/inward/record.url?scp=85182388118&partnerID=8YFLogxK
U2 - 10.1109/TVT.2023.3342186
DO - 10.1109/TVT.2023.3342186
M3 - Article
AN - SCOPUS:85182388118
SN - 0018-9545
VL - 73
SP - 6249
EP - 6258
JO - IEEE Transactions on Vehicular Technology
JF - IEEE Transactions on Vehicular Technology
IS - 5
ER -