Adversarial Training: Enhancing Out-of-Distribution Generalization for Learning Wireless Resource Allocation
Deep neural networks (DNNs) have widespread applications for optimizing resource allocation. Yet, their performance is vulnerable to distribution shifts between training and test data, say channels. In this letter, we resort to adversarial training (AT) for enhancing out-of-distribution (OOD) generalizability of DNNs trained in unsupervised manner. We reformulate AT to capture the OOD degradation, and propose a one-step gradient ascent method for AT. The proposed method is validated by optimizing hybrid precoding. Simulation results showcase the enhanced OOD performance of multiple kinds of DNNs across various channel distributions, when only Rayleigh fading channels are used for training.
Code (0)
등록된 구현이 없습니다.
Tasks
Out-of-Distribution GeneralizationSimilar Papers 제목 키워드 기반
RegMix: Adversarial Mutual and Generalization Regularization for Enhancing DNN Robustness
Adversarial training is the most effective defense against adversarial attacks. The effectiveness of the adversarial attacks has been on the design of its loss function and regularization term. The most widely used loss …
Knowledge DistillationAdversarial RobustnessAdversarial Training Improves Generalization Under Distribution Shifts in Bioacoustics
Adversarial training is a promising strategy for enhancing model robustness against adversarial attacks. However, its impact on generalization under substantial data distribution shifts in audio classification remains la…
Adversarial RobustnessAudio ClassificationRevisiting the Robust Generalization of Adversarial Prompt Tuning
Understanding the vulnerability of large-scale pre-trained vision-language models like CLIP against adversarial attacks is key to ensuring zero-shot generalization capacity on various downstream tasks. State-of-the-art d…
Adversarial RobustnessPrompt LearningZero-shot GeneralizationJoint Interference Detection and Identification via Adversarial Multi-task Learning
Precise interference detection and identification are crucial for enhancing the survivability of communication systems in non-cooperative wireless environments. While deep learning (DL) has advanced this field, existing …
Multi-Task LearningImproving Generalization of Adversarial Training via Robust Critical Fine-Tuning
Deep neural networks are susceptible to adversarial examples, posing a significant security risk in critical applications. Adversarial Training (AT) is a well-established technique to enhance adversarial robustness, but …
Adversarial Robustness