Decrease the Prompt Uncertainty: Adversarial Prompt Learning for Few-shot Text Classification
With few-shot learning abilities, pre-trained language models (PLMs) have achieved remarkable success in classification tasks. However, recent studies have shown that the performance of PLM is vulnerable due to different prompts and the instability of the prompt-based learning process. To address this challenge, we explore appropriate perturbation addition of adversarial training and integrate the global knowledge of the full-parameter fine-tuned pre-trained language model(PLM). Specifically, we propose an adversarial prompt learning model (ATPET) and ATPET with fine-tuning(ATPET-FT), incorporating ATPET with fine-tuning knowledge into the prompt learning process. Through extensive experiments on several few-shot classification tasks and challenging data settings, we demonstrate that our methods consistently improve the robustness while maintaining the effectiveness of PLMs.
Authors
- Zhaoguang Zhang
- Chenxu Niu (ORCID: https://orcid.org/0009-0004-8213-1182)
- Jing Ya-Qi
- Heyan Huang (ORCID: https://orcid.org/0000-0002-3427-868X)
- Jinta Weng (ORCID: https://orcid.org/0000-0002-6235-6454)
- Yue Hu (ORCID: https://orcid.org/0000-0002-4416-3974)
Institutions
- Beijing Institute of Technology (CN)
- Guangzhou University (CN)
- National Computer Network Emergency Response Technical Team/Coordination Center of Chinar (CN)
- University of Chinese Academy of Sciences (CN)
Publication Details
- Published
- 2024-10-06
- DOI
- https://doi.org/10.1109/smc54092.2024.10831613
- Primary Topic
- Adversarial Robustness in Machine Learning
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China