Learning Compact Metrics for MT

TL;DR

Proposes RemBERT, a distilled multilingual evaluation metric reaching 92.6% of the large model's performance with only one-third parameters.

cs.CL 🔴 Advanced 2021-10-13 44 views
Amy Pu Hyung Won Chung Ankur P. Parikh Sebastian Gehrmann Thibault Sellam
multilingual learning knowledge distillation MT evaluation model compression cross-lingual transfer

Key Findings

Methodology

This study employs RemBERT as the backbone multilingual pre-trained model, integrating synthetic data generation and knowledge distillation to address capacity bottlenecks. The process involves: (1) generating 80 million sentences across 13 languages via Wikipedia perturbations; (2) training a teacher model (RemBERT-32) on this synthetic corpus; (3) applying pre-trained distillation to transfer knowledge to smaller student models (RemBERT-3/6/12); (4) fine-tuning on WMT metrics data for cross-lingual transfer evaluation. Experiments validate that larger models perform better, but distillation enables smaller models to approach similar performance.

Key Results

  • The distilled models achieve up to 92.6% of RemBERT's performance with only a third of its parameters, outperforming vanilla fine-tuning by 10.5%. On the WMT shared task, they surpass SOTA metrics like COMET and BLEURT-EXTENDED across multiple languages, including zero-shot settings, with significant improvements in correlation scores.
  • Synthetic data augmentation combined with multi-model distillation boosts average DaRR correlation by approximately 4%, demonstrating effective mitigation of the multilingual capacity curse. Larger models show better transfer, but the proposed methods enable efficient deployment of compact models with competitive accuracy.
  • Multi-language fine-tuning and specialized models for language clusters further improve performance, especially in low-resource languages, validating the approach's generalization and adaptability.

Significance

This work addresses the critical challenge of balancing model size and multilingual evaluation performance. By innovatively combining synthetic data and distillation, it enables deployment of high-performing, lightweight evaluation metrics suitable for real-world applications. The approach reduces computational costs, facilitates scalability to many languages, and enhances low-resource language support, thus advancing the state-of-the-art in multilingual NLP evaluation. It opens new avenues for model compression, transfer learning, and cross-lingual NLP tasks, with broad industry and research implications.

Technical Contribution

The paper introduces a novel framework that integrates synthetic multilingual data generation with multi-model knowledge distillation, achieving significant parameter reduction while maintaining high evaluation accuracy. Key innovations include: (1) synthetic data creation via Wikipedia perturbations to simulate translation errors; (2) large teacher model training on synthetic corpus; (3) multi-model distillation to produce compact models; (4) cross-lingual transfer enhancement through multi-language fine-tuning. These contributions collectively push the boundaries of model efficiency and scalability in multilingual evaluation.

Novelty

This is the first systematic combination of synthetic data generation and multi-model distillation tailored for multilingual evaluation metrics. Unlike prior work focusing solely on large models, this approach leverages synthetic perturbations to generate abundant training data and employs multiple smaller models trained on language-specific clusters. This dual strategy effectively alleviates the capacity curse, enabling high performance with significantly fewer parameters, representing a new paradigm in multilingual NLP evaluation.

Limitations

  • The synthetic data, while diverse, may not fully capture the complexity of real translation errors, potentially limiting generalization to real-world MT outputs.
  • Training multiple models for different language clusters increases overall complexity and resource requirements during development.
  • Despite significant compression, the approach still relies on large teacher models, which are costly to pre-train and fine-tune. Further research is needed to optimize training efficiency and extend coverage to more low-resource languages.

Future Work

Future directions include refining synthetic data generation to better mimic real translation errors, exploring additional compression techniques like pruning and quantization, and extending the framework to other NLP tasks such as summarization and question answering. Additionally, integrating unsupervised or semi-supervised learning could further reduce reliance on large annotated corpora, broadening the applicability of lightweight multilingual evaluation models.

AI Executive Summary

In recent years, the rapid growth of multilingual natural language processing (NLP) has driven the development of evaluation metrics that can accurately assess machine translation quality across many languages. Traditional metrics like BLEU are limited by their surface-level focus, prompting researchers to develop learned metrics such as COMET and BLEURT, which leverage large multilingual pre-trained models like XLM-RoBERTa. However, these models are computationally expensive and difficult to deploy at scale, especially for low-resource languages.

This paper introduces RemBERT, a compact yet high-performing multilingual evaluation metric achieved through a combination of synthetic data generation and knowledge distillation. The authors generate an extensive synthetic corpus of 80 million sentences across 13 languages by perturbing Wikipedia data with techniques like back-translation, word substitution, and random deletion. This synthetic data serves as training material for a large teacher model (RemBERT-32), which is then used to supervise smaller student models (RemBERT-3/6/12) via pre-trained distillation. The process effectively transfers knowledge, enabling the smaller models to approximate the performance of the larger one.

Experimental results on the WMT Metrics Shared Task demonstrate that the distilled models reach 92.6% of the teacher model's performance while using only a third of its parameters. They outperform baseline fine-tuning methods by up to 10.5%, showing significant improvements in correlation with human judgments across multiple languages, including zero-shot settings. The approach also incorporates multi-language fine-tuning, which further enhances transferability to unseen languages.

Beyond performance gains, the proposed framework offers practical benefits such as faster inference speeds—achieving 4.8 tuples per second without GPU—and reduced computational costs, making it suitable for real-world deployment. The authors highlight that their method addresses the capacity bottleneck in multilingual evaluation models, paving the way for scalable, efficient, and accurate multilingual NLP evaluation tools. Future work will focus on refining synthetic data techniques, extending language coverage, and integrating additional compression methods to further improve efficiency and applicability.

Deep Analysis

Background

Multilingual NLP has seen rapid advances with models like BERT, XLM-R, and mT5, enabling cross-lingual transfer and multi-task learning. Evaluation metrics evolved from lexical overlap measures such as BLEU to learned metrics like COMET and BLEURT, which utilize deep pre-trained models to better reflect human judgment. However, as models grow larger to accommodate more languages, they face scalability issues, especially in deployment scenarios with limited hardware resources. The 'curse of multilinguality' describes the phenomenon where increasing languages in a fixed-capacity model leads to performance degradation, limiting the effectiveness for low-resource languages. Researchers have sought methods to compress models without sacrificing accuracy, but balancing size and performance remains challenging.

Core Problem

Current multilingual evaluation models are hampered by their large size, which hampers deployment efficiency and scalability. The capacity bottleneck restricts the number of languages supported and diminishes performance in low-resource scenarios. Moreover, training large models is computationally expensive, limiting accessibility for many researchers and practitioners. Addressing these issues requires developing methods to retain high evaluation accuracy while significantly reducing model parameters, enabling practical deployment in diverse settings. The core challenge lies in effectively transferring knowledge from large models to smaller ones without losing critical semantic understanding, especially across many languages.

Innovation

This work introduces a hybrid approach combining synthetic data generation and multi-model knowledge distillation to create a lightweight yet high-performing multilingual evaluation metric. Key innovations include:

1) Synthetic data creation using Wikipedia perturbations, simulating translation errors across 13 languages, thus enriching training data without requiring additional labeled datasets.

2) Training a large teacher model (RemBERT-32) on this synthetic corpus to serve as a high-quality reference.

3) Applying pre-trained distillation to transfer knowledge from the teacher to smaller student models (RemBERT-3/6/12), drastically reducing parameters.

4) Employing multi-language fine-tuning and clustering strategies to improve transferability and performance in low-resource languages.

These innovations collectively address the capacity bottleneck, enabling efficient, scalable evaluation models.

Methodology

  • �� Synthetic data generation: Use Wikipedia perturbations—back-translation, word substitution with mBERT, and random deletions—to produce 80 million sentence pairs in 13 languages.
  • �� Teacher training: Fine-tune RemBERT-32 on synthetic corpus, optimizing for correlation with human ratings.
  • �� Knowledge distillation: Run teacher model on unlabeled data, generate pseudo-labels, then train smaller models (RemBERT-3/6/12) to mimic teacher outputs.
  • �� Multi-model training: Cluster languages based on linguistic similarity, train dedicated models for each cluster, and transfer knowledge via distillation.
  • �� Fine-tuning: Use WMT metrics data to adapt models for evaluation tasks, validating cross-lingual transfer.
  • �� Speed optimization: Implement length-based batching and model distillation to improve inference speed, enabling deployment in real-world scenarios.

Experiments

The experiments utilize WMT 2020 metrics shared task data, comprising 479k triplets across 12 languages. Models are trained with Adam optimizer, learning rate schedules, and evaluated using DaRR correlation scores. Baselines include COMET and BLEURT-EXTENDED. Synthetic data augmentation and multi-model distillation are systematically tested, with hyperparameters tuned for optimal performance. Additional ablation studies compare different synthetic perturbation strategies, model sizes, and fine-tuning configurations. Cross-lingual transfer is assessed through zero-shot evaluations on unseen languages, demonstrating robustness and scalability of the proposed approach.

Results

The distilled models achieve 92.6% of the teacher model’s performance, surpassing baseline fine-tuning by 10.5%. Across languages, they outperform SOTA metrics like COMET and BLEURT-EXTENDED, with improvements of up to 4% in DaRR scores. Synthetic data and multi-model distillation significantly boost low-resource language performance, especially in zero-shot settings. Speed tests show that the compressed models process 4.8 tuples/sec without GPU, representing a 2.5-3x speedup over the original large model. These results confirm that the proposed framework effectively balances model size, transferability, and accuracy.

Applications

The lightweight models can be deployed for real-time machine translation evaluation, quality estimation in multilingual systems, and low-resource language assessment. They are suitable for integration into translation pipelines, enabling faster and more cost-effective quality control. The approach also benefits research in low-resource NLP, facilitating the development of evaluation metrics for underrepresented languages. Future applications include expanding to other NLP tasks such as summarization and question answering, where model efficiency is critical for large-scale deployment.

Limitations & Outlook

While synthetic data improves robustness, it may not fully replicate real translation errors, limiting generalization. The multi-model training process increases development complexity and resource consumption. The reliance on large teacher models for initial training remains costly, and extending coverage to extremely low-resource languages requires additional data collection and adaptation. Future work should focus on further reducing training costs, improving synthetic data realism, and exploring unsupervised approaches to enhance scalability.

Plain Language Accessible to non-experts

想象你在一个大厨房里准备多种菜肴。厨师(老师模型)非常厉害,能做出各种美味,但太大太慢,不能同时给所有人用。于是,你让厨师用一种特别的方法,把菜谱变得简单一些(模型蒸馏),让厨房里的其他厨师(学生模型)也能学会做菜,但只用很少的材料和时间。为了让他们学得更快更好,厨师还会用一些模拟的练习,比如模仿不同的调料和做法(合成数据),让他们在不同情况下都能做出好菜。这样,厨房里就能用更少的厨师,做出和大厨一样好吃的菜。这就像论文用合成数据和蒸馏技术,把复杂的评估模型变得更小更快,但效果依然很棒。

ELI14 Explained like you're 14

想象你在学校里学做菜,老师用一道超级复杂的菜谱教你怎么做(老师模型)。你学会后,不一定能记住所有细节,但老师会给你一些提示,让你用更简单的方法做出差不多的味道(学生模型)。为了让你学得更快、更好,老师还会用一些模拟的练习题(合成数据),让你练习不同的变化。这样,你就不用每天跟老师学一模一样的菜谱,也能做出很好吃的菜。这个方法让你变得更聪明、更快,也节省了很多时间和精力。论文里的技术就像这样,用“老师”帮“学生”变得更厉害,模型变得更小更快,但效果依然很棒!

Abstract

Recent developments in machine translation and multilingual text generation have led researchers to adopt trained metrics such as COMET or BLEURT, which treat evaluation as a regression problem and use representations from multilingual pre-trained models such as XLM-RoBERTa or mBERT. Yet studies on related tasks suggest that these models are most efficient when they are large, which is costly and impractical for evaluation. We investigate the trade-off between multilinguality and model capacity with RemBERT, a state-of-the-art multilingual language model, using data from the WMT Metrics Shared Task. We present a series of experiments which show that model size is indeed a bottleneck for cross-lingual transfer, then demonstrate how distillation can help addressing this bottleneck, by leveraging synthetic data generation and transferring knowledge from one teacher to multiple students trained on related languages. Our method yields up to 10.5% improvement over vanilla fine-tuning and reaches 92.6% of RemBERT's performance using only a third of its parameters.

cs.CL