Contrastive GNN-based traffic anomaly analysis against imbalanced dataset in IoT-based ITS
- Authors: Wang, Yang , Lin, Xi , Wu, Jun , Bashir, Ali , Yang, Wu , Li, Jianhua , Imran, Muhammad
- Date: 2022
- Type: Text , Conference paper
- Relation: 2022 IEEE Global Communications Conference, GLOBECOM 2022, Virtual, online, 4-8 December 2022, 2022 IEEE Global Communications Conference, GLOBECOM 2022 - Proceedings p. 3557-3562
- Full Text: false
- Reviewed:
- Description: The traffic anomaly analysis in IoT-based intelligent transportation system (ITS) is crucial to improving public transportation safety and efficiency. The issue is also challenging due to the unbalanced distribution of anomaly data in IoT-based ITS, which may cause overfitting or underfitting in the training phase. However, some research on traffic anomaly analysis injected limited data to address the shortage of anomaly samples or even neglects this issue, which overlooks the potential representation of nodes in graph neural networks. In this paper, we propose an improved contrastive GNN-based learning framework for traffic anomaly analysis that alleviates the problem of imbalanced datasets in the training phase. In this framework, we provide a graph augmentation approach with coupled features to learn different views of graph data. Besides, we design an effective training method based on the contrastive loss for our framework, which can learn the better performance of latent representations utilized in the downstream tasks. Finally, we conduct extensive experiments to evaluate the performance of our proposed frame-works based on real-world datasets. We demonstrate that our framework achieves as high as 6.45% precision improvement compared to the state-of-the-art. © 2022 IEEE.
Metric learning-based few-shot malicious node detection for IoT backhaul/fronthaul networks
- Authors: Zhou, Ke , Lin, Xi , Wu, Jun , Bashir, Ali , Li, Jianhua , Imran, Muhammad
- Date: 2022
- Type: Text , Conference paper
- Relation: 2022 IEEE Global Communications Conference, GLOBECOM 2022, Virtual, online, 4-8 December 2022, 2022 IEEE Global Communications Conference, GLOBECOM 2022 - Proceedings p. 5777-5782
- Full Text: false
- Reviewed:
- Description: The development of backhaul/fronthaul networks can enable low latency and high reliability, but nodes in future networks like Internet of Things (IoT) can conduct malicious activities like flooding attack and DDoS attack, which can decrease QoS of smart backhaul/fronthaul network. Timely detection of malicious nodes in future networks is significant for low-latency backhaul/fronthaul networks. However, conventional supervised learning-based detection models require abundant malicious training samples, while capturing adequate malicious samples can not meet the requirement of timely detection. In this paper, we propose a novel few-shot malicious node detection system for improving QoS of IoT backhaul/fronthaul network, which can detect malicious nodes with unknown malicious activities through a limited number of network traffic samples. In our proposed system, we first design a fresh IoT traffic sample processing approach, which integrates normal activity samples and known malicious activity samples to generate training pairs. Then, we design a metric learning-based malicious node detection model training method, which employs a contrastive loss over distance metric to distinguish between similar and dissimilar pairs of samples. Besides, the trained model can detect nodes with unknown malicious activities by comparing real-time samples with few-shot samples of malicious nodes. Finally, the proposed system is evaluated on a real-world IoT network dataset named N-BaIoT. The exhaustive experiment results show that our model can achieve an average accuracy around 97.67 % when detecting malicious nodes with unknown malicious activities, which is comparable to state-of-the-art supervised learning models while our model only needs 5-shot samples of malicious node. © 2022 IEEE.