A Partially Labeled Anomaly Data Detection Approach Based on Prioritized Deep Reinforcement Learning for Consumer Electronics Security

被引:0
作者
Qin, Shuqi [1 ]
Liu, Shenghao [1 ]
Ye, Shengjie [1 ]
Fan, Xiaoxuan [1 ]
Cheng, Minmin [1 ]
He, Yuanyuan [1 ]
Deng, Xianjun [1 ]
Park, Jong Hyuk [2 ]
机构
[1] Huazhong Univ Sci & Technol, Hubei Engn Res Ctr Big Data Secur, Sch Cyber Sci & Engn, Hubei Key Lab Distributed Syst Secur, Wuhan 430074, Peoples R China
[2] Seoul Natl Univ Sci & Technol, Dept Comp Sci & Engn, Seoul 01811, South Korea
基金
中国国家自然科学基金;
关键词
Consumer electronics; Uncertainty; Training; Security; Aerospace electronics; Detectors; Deep reinforcement learning; Consumer electronics security; anomaly data detection; deep reinforcement learning;
D O I
10.1109/TCE.2024.3445629
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Anomalies within data flows in the Internet of Things environment can potentially result in security vulnerabilities in consumer electronics. Therefore, it is crucial to effectively detect anomaly data to safeguard the reliability and continuous functionality of consumer electronics. Existing related works either learn from unlabeled data using unsupervised methods or leverage the limited labeled data to improve detection performance by semi-supervised methods. However, these methods usually overfit specific types of known anomalies or ignore the uncertainty when model training. To this end, we design a novel approach to jointly optimize the end-to-end detection of labeled and unlabeled anomalies. Specifically, the anomaly data detection problem investigated is first reformulated as a Markov decision process. Then, a partially labeled anomaly data detection approach (PANDA) based on prioritized deep deterministic policy gradient is proposed, which considers uncertainty when the agent makes decisions and can learn from the labeled known anomalies while continuously exploring and detecting prospective anomalies in unlabeled data. Extensive experiments on 13 datasets show that PANDA improves the AUC-ROC and AUC-PR by 3.0%-10.3% and 10.0%-73.5% and its robustness under the impact of anomaly contamination rates compared with four state-of-the-art competing methods.
引用
收藏
页码:6452 / 6462
页数:11
相关论文
共 31 条
[1]   A Survey of Machine and Deep Learning Methods for Internet of Things (IoT) Security [J].
Al-Garadi, Mohammed Ali ;
Mohamed, Amr ;
Al-Ali, Abdulla Khalid ;
Du, Xiaojiang ;
Ali, Ihsan ;
Guizani, Mohsen .
IEEE COMMUNICATIONS SURVEYS AND TUTORIALS, 2020, 22 (03) :1646-1685
[2]  
Boyd Kendrick, 2013, Machine Learning and Knowledge Discovery in Databases. European Conference, ECML PKDD 2013. Proceedings: LNCS 8190, P451, DOI 10.1007/978-3-642-40994-3_29
[3]   Catching Both Gray and Black Swans: Open-set Supervised Anomaly Detection [J].
Ding, Choubo ;
Pang, Guansong ;
Shen, Chunhua .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :7378-7388
[4]   Network Abnormal Traffic Detection Model Based on Semi-Supervised Deep Reinforcement Learning [J].
Dong, Shi ;
Xia, Yuanjun ;
Peng, Tao .
IEEE TRANSACTIONS ON NETWORK AND SERVICE MANAGEMENT, 2021, 18 (04) :4197-4212
[5]   Cyber Attacks via Consumer Electronics: Studying the Threat of Covert Malware in Smart and Autonomous Vehicles [J].
Haghighi, Mohammad Sayad ;
Farivar, Faezeh ;
Jolfaei, Alireza ;
Asl, Azin Bayrami ;
Zhou, Wei .
IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2023, 69 (04) :825-832
[6]   Towards safe and robust weakly-supervised anomaly detection under subpopulation shift [J].
He, Rundong ;
Han, Zhongyi ;
Yin, Yilong .
KNOWLEDGE-BASED SYSTEMS, 2022, 250
[7]   An Intelligent Intrusion Detection System for Smart Consumer Electronics Network [J].
Javeed, Danish ;
Saeed, Muhammad Shahid ;
Ahmad, Ijaz ;
Kumar, Prabhat ;
Jolfaei, Alireza ;
Tahir, Muhammad .
IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2023, 69 (04) :906-913
[8]   Interpretable machine learning with an ensemble of gradient boosting machines [J].
Konstantinov, Andrei, V ;
Utkin, Lev, V .
KNOWLEDGE-BASED SYSTEMS, 2021, 222
[9]  
Lillicrap T., 2016, ARXIV
[10]  
Liu BY, 2021, PROCEEDINGS OF THE THIRTIETH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2021, P1505, DOI 10.24963/ijcai.2021/208