Stop Diverse OOD Attacks: Knowledge Ensemble for Reliable Defense
Enhancing defense through model ensemble is an emerging trend, where the challenge lies in how to use ensemble knowledge to counter Out-of-Distribution (OOD) attacks. In this paper, we propose the Reliable Defense Ensemble (REE) to address this issue. REE optimizes the ensemble knowledge of models through aggregation and enhances multidimensional robust performance through collaboration. It employs the Dynamic Synergy Amplification for weight allocation and strategy adjustment. Furthermore, we design a new Kernel Anomaly Smoothing Detection Module, which detects anomalous attacks using a smoothing feature function based on Gaussian kernel mean embedding and a multi-layer feedback structure. Particularly, we build a framework that uses reinforcement learning to iteratively fine-tune the parameters of inter-model communication and consensus. Extensive experimental results show that REE outperforms current state-of-the-art methods by a large margin in defending against OOD attacks.
Authors
- Liusheng Huang (ORCID: https://orcid.org/0000-0001-8417-3256)
- Xiaoman Liu
- Zhenbo Shi
- Shuchang Wang (ORCID: https://orcid.org/0000-0001-8191-1188)
- Yuxuan Zhang (ORCID: https://orcid.org/0000-0002-3760-1083)
- Wei Yang (ORCID: https://orcid.org/0000-0002-5338-7347)
- Rui Shu (ORCID: https://orcid.org/0000-0001-8937-7274)
- Zhidong Yu (ORCID: https://orcid.org/0000-0001-7168-6423)
Institutions
- University of Science and Technology of China (CN)
- Hefei University of Technology (CN)
- Suzhou University of Science and Technology (CN)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i19.34251
- Primary Topic
- Network Security and Intrusion Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- China Postdoctoral Science Foundation
- Government of Jiangsu Province
- Natural Science Foundation of Jiangsu Province