Chinmay Pateria
Author directory2026
Predict and Fix: A Unified Model for Translation Quality Estimation and Post-Editing
Maciej Modrzejewski | Yash Bhaskar | Chinmay Pateria
Proceedings of the 17th Conference of the Association for Machine Translation in the Americas (Volume 1: Research Track)
Maciej Modrzejewski | Yash Bhaskar | Chinmay Pateria
Proceedings of the 17th Conference of the Association for Machine Translation in the Americas (Volume 1: Research Track)
We propose a unified architecture for jointly modeling Translation Quality Estimation (QE) and Automatic Post-Editing (APE) within a single lightweight language model. Our approach integrates quality prediction and correction generation in a single decoding process using a decoder-only Qwen2.5 model (0.5B parameters), augmented with a dedicated QE regression head operating on hidden states at a special token position. The model produces structured outputs that include a continuous quality score, an edit decision, and a corrected translation when necessary. We train on datasets of 100K, 1M, and 1.84M manually annotated samples across eight language pairs, enabling analysis of both data scale and distribution. Experimental results show that the proposed model achieves strong QE performance (r=0.907) and high post-editing decision accuracy (88.4%), while reducing over-editing compared to both autoregressive baselines and large commercial LLMs.