arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.14733cs.LGcs.AIcs.CL
收藏

硬提示变得可解释:用于强化学习提示调优的稀疏熵正则化

Hard Prompts Made Interpretable: Sparse Entropy Regularization for Prompt Tuning with RL

Microsoft Research Asia(微软亚洲研究院);KAIST(韩国科学技术院)

Yunseon Choi, Sangmin Bae, Seonghyun Ban, Minchan Jeong, Chuheng Zhang, Lei Song, Li Zhao, Jiang Bian, Kee-Eung Kim

AI总结:

针对RLPrompt产生不自然提示的问题,提出稀疏Tsallis熵正则化过滤低概率词元,在少样本分类、风格迁移和文本反演任务上显著提升提示的自然性和可解释性。

摘要:

随着基础模型的出现,提示调优已成为引导模型行为和引发期望响应的重要技术。提示调优涉及选择包含在输入中的适当关键词,从而在不调整或微调模型参数的情况下适应下游任务。提示调优领域有广泛的工作,从直接利用模型反向传播梯度信号的方法,到采用黑盒优化(如强化学习(RL)方法)的方法。我们的主要关注点是RLPrompt,它旨在利用软Q学习找到最优的提示词元。虽然结果显示出前景,但我们观察到提示经常显得不自然,这阻碍了其可解释性。我们通过使用稀疏Tsallis熵正则化来解决这一限制,这是一种从考虑中过滤掉不太可能的词元的原则性方法。我们在各种任务上广泛评估了我们的方法,包括少样本文本分类、无监督文本风格迁移和从图像进行文本反演。结果表明,与基线相比有显著改进,突出了我们的方法在解决提示调优挑战方面的有效性。此外,我们表明,使用我们的方法发现的提示比其他基线发现的提示更自然、更可解释。

英文摘要:

With the advent of foundation models, prompt tuning has positioned itself as an important technique for directing model behaviors and eliciting desired responses. Prompt tuning regards selecting appropriate keywords included into the input, thereby adapting to the downstream task without adjusting or fine-tuning the model parameters. There is a wide range of work in prompt tuning, from approaches that directly harness the backpropagated gradient signals from the model, to those employing black-box optimization such as reinforcement learning (RL) methods. Our primary focus is on RLPrompt, which aims to find optimal prompt tokens leveraging soft Q-learning. While the results show promise, we have observed that the prompts frequently appear unnatural, which impedes their interpretability. We address this limitation by using sparse Tsallis entropy regularization, a principled approach to filtering out unlikely tokens from consideration. We extensively evaluate our approach across various tasks, including few-shot text classification, unsupervised text style transfer, and textual inversion from images. The results indicate a notable improvement over baselines, highlighting the efficacy of our approach in addressing the challenges of prompt tuning. Moreover, we show that the prompts discovered using our method are more natural and interpretable compared to those from other baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑