PTR: Prompt Tuning with Rules for Text Classification

• Published in: AI open 2022, Vol.3, p.182-192
• Main Authors: Han, Xu, Zhao, Weilin, Ding, Ning, Liu, Zhiyuan, Sun, Maosong
• Format: Article
• Language: English
• Subjects: Pre-trained language models; Prompt tuning
• ISSN: 2666-6510; 2666-6510
• DOI: 10.1016/j.aiopen.2022.11.003

Recently, prompt tuning has been widely applied to stimulate the rich knowledge in pre-trained language models (PLMs) to serve NLP tasks. Although prompt tuning has achieved promising results on some few-class classification tasks, such as sentiment classification and natural language inference, manually designing prompts is cumbersome. Meanwhile, generating prompts automatically is also difficult and time-consuming. Therefore, obtaining effective prompts for complex many-class classification tasks still remains a challenge. In this paper, we propose to encode the prior knowledge of a classification task into rules, then design sub-prompts according to the rules, and finally combine the sub-prompts to handle the task. We name this Prompt Tuning method with Rules “PTR”. Compared with existing prompt-based methods, PTR achieves a good trade-off between effectiveness and efficiency in building prompts. We conduct experiments on three many-class classification tasks, including relation classification, entity typing, and intent classification. The results show that PTR outperforms both vanilla and prompt tuning baselines, indicating the effectiveness of utilizing rules for prompt tuning. The source code of PTR is available at https://github.com/thunlp/PTR.