通过自纠正对抗训练,让模型从自身错误中学习,提升中文文本纠错泛化能力。
Learning from Mistakes: Self-correct Adversarial Training for Chinese Unnatural Text Correction
- 利用推理时模型生成的不一致预测作为错误样本进行训练
- 在多个中文数据集上超越当前最优方法,有效纠正拼写与对抗扰动错误
- 无需重新训练即可适配新模型和数据集,可即插即用
自然语言纠错旨在自动检测并修正句子中的拼写错误或对抗扰动错误。现有方法多依赖微调或对抗训练,虽取得显著成效,但因训练数据分布与真实场景差异导致泛化性能差,存在曝光偏差问题。本文提出一种任务与模型无关的自纠正对抗训练框架——LIMIT(Learning from Mistakes),充分利用推理阶段模型主动暴露的错误预测(即与目标不一致的输出),模拟真实应用中的潜在错误,缓解传统训练过程的曝光偏差。同时设计新型解码干预策略,保障语义一致性。大量实验表明,该方法在中文非自然文本纠错数据集上能有效纠正多种错误形式,优于当前最优方法。此外,在中英文数据集上的广泛验证证明,LIMIT可作为即插即用的防御模块,无需额外训练即可扩展至新模型与新数据集。
原文摘要 · Abstract (English)
Unnatural text correction aims to automatically detect and correct spelling errors or adversarial perturbation errors in sentences. Existing methods typically rely on fine-tuning or adversarial training to correct errors, which have achieved significant success. However, these methods exhibit poor generalization performance due to the difference in data distribution between training data and real-world scenarios, known as the exposure bias problem. In this paper, we propose a self-correct adversarial training framework for \textbf{L}earn\textbf{I}ng from \textbf{MI}s\textbf{T}akes (\textbf{LIMIT}), which is a task- and model-independent framework to correct unnatural errors or mistakes. Specifically, we fully utilize errors generated by the model that are actively exposed during the inference phase, i.e., predictions that are inconsistent with the target. This training method not only simulates potential errors in real application scenarios, but also mitigates the exposure bias of the traditional training process. Meanwhile, we design a novel decoding intervention strategy to maintain semantic consistency. Extensive experimental results on Chinese unnatural text error correction datasets show that our proposed method can correct multiple forms of errors and outperforms the state-of-the-art text correction methods. In addition, extensive results on Chinese and English datasets validate that LIMIT can serve as a plug-and-play defense module and can extend to new models and datasets without further training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。