arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.00740cs.CLcs.LG
收藏

LaSEr-Edit:基于能量定位的局部跨度级错误编辑

LaSEr-Edit: Localized Span-level Error Editing with Energy-based Localization

Graduate School of Data Science(数据科学研究生院);Seoul National University(首尔国立大学);Georgia Institute of Technology(佐治亚理工学院)

Hye Ryung Son, Saehee Eom, Mooho Song, Jay-Yoon Lee

AI总结:

研究针对大语言模型满足约束问题,提出LaSEr-Edit方法。利用轻量级特定任务的基于能量的模型进行错误定位,提出LaSEr-LLM Edit和LaSEr-EBM Edit两种文本修订方法,实验表明该方法能有效控制文本,多约束下也表现良好。

摘要:

随着大语言模型(LLMs)在实际应用中广泛采用,确保其满足安全约束(如无毒和逻辑一致性)以及特定任务和情境约束变得至关重要。通过指令控制输出虽简单但不可靠。多数近期受控文本生成(CTG)方法需访问语言模型内部组件,与基于API的流行LLMs不兼容。本文提出LaSEr-Edit,一种可应用于任何LLMs的约束满足文本修订方法。首先发现轻量级、特定任务的基于能量的模型(EBMs)在错误定位性能上具有竞争力,基于此提出两种结合能量错误定位的文本修订方法变体。通过实验表明,LaSEr-LLM Edit在多数任务中比普通基于LLM的编辑更好地控制文本,LaSEr-EBM Edit进一步提升控制性能,在多约束控制时也表现良好。

英文摘要:

As large language models (LLMs) are widely adopted in real-world applications, it has become critical to ensure LLMs satisfy safety constraints, such as non-toxicity and logical consistency, as well as task- and situation-specific constraints. Controlling the output through instructions is a simple and tempting approach; however, it remains brittle, is opaque in how it influences model behavior, and thus cannot reliably ensure constraint satisfaction. Moreover, most recent controlled text generation (CTG) methods require access to the internal components of language models--such as weights or logits--making them incompatible with popular API-based LLMs. In this work, we propose LaSEr-Edit, a constraint-satisfying text revision method that can be applied to any LLMs, black- or white-box. We first find that lightweight, task-specific energy-based models (EBMs) achieve error-localization performance competitive with or even better than that of much larger LLMs, while operating substantially faster. Based on this finding, we propose two variants of text revision methods that incorporate energy-based error localization: LaSEr-LLM Edit, which instructs an LLM to edit text given EBM-predicted error spans, and LaSEr-EBM Edit, which uses the EBM not only for localization but also for editing by reranking edit candidates. Through experiments in diverse single-constraint control tasks, we show that LaSEr-LLM Edit controls text better than plain LLM-based editing in most of the tasks. We also find that LaSEr-EBM Edit further improves the control performance of LaSEr-LLM Edit and achieves among the strongest controllability across all tasks. Furthermore, we find that LaSEr-Edit, especially LaSEr-EBM Edit, performs well even when multiple constraints are controlled simultaneously.

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

↑