Grammatical Error Correction Evaluation by Optimally Transporting Edit Representation
Takumi Goto, Yusuke Sakai, Taro Watanabe
TL;DR
This paper tackles the core challenge of evaluating grammatical error correction (GEC) systems by moving beyond token-level embedding similarity to edit-level similarity. It introduces edit vectors that encode the semantic impact of individual edits and uses unbalanced optimal transport (UOT) to transport these vectors from a hypothesis to a reference, producing an interpretable soft edit alignment. The proposed UOT-ERRANT metric derives precision, recall, and $F_{\beta}$ from the transport plan, allowing robust system ranking across diverse reference sets and improving correlations with human judgments, particularly in fluency-heavy domains. Although computationally heavier, the method offers valuable analytical capabilities for debugging and understanding GEC systems and is validated on SEEDA and GMEG-Data meta-evaluation tasks with strong performance in +Fluency scenarios. The approach also invites broader applications of edit vectors to other editing tasks and evaluation contexts.
Abstract
Automatic evaluation in grammatical error correction (GEC) is crucial for selecting the best-performing systems. Currently, reference-based metrics are a popular choice, which basically measure the similarity between hypothesis and reference sentences. However, similarity measures based on embeddings, such as BERTScore, are often ineffective, since many words in the source sentences remain unchanged in both the hypothesis and the reference. This study focuses on edits specifically designed for GEC, i.e., ERRANT, and computes similarity measured over the edits from the source sentence. To this end, we propose edit vector, a representation for an edit, and introduce a new metric, UOT-ERRANT, which transports these edit vectors from hypothesis to reference using unbalanced optimal transport. Experiments with SEEDA meta-evaluation show that UOT-ERRANT improves evaluation performance, particularly in the +Fluency domain where many edits occur. Moreover, our method is highly interpretable because the transport plan can be interpreted as a soft edit alignment, making UOT-ERRANT a useful metric for both system ranking and analyzing GEC systems. Our code is available from https://github.com/gotutiyan/uot-errant.
