Skip to content
/ ThinMQM Public

[NeurIPS 2025] Are Large Reasoning Models Good Translation Evaluators? Analysis and Performance Boost

License

Notifications You must be signed in to change notification settings

NLP2CT/ThinMQM

Repository files navigation

Are Large Reasoning Models Good Translation Evaluators? Analysis and Performance Boost

Diagnose the key issues in LRM-as-a-Judge for MT and calibrate the thoughts.

overview

Paper Github Hugging Face Collection Hugging Face Collection



πŸ“’News

  • [2025/10/24] 🌻 ThinMQM paper is available on arXiv.
  • [2025/09/19] πŸŽ‰ ThinMQM has been accepted to NeurIPS 2025!

πŸ“–Introduction

Evaluating machine translation (MT) quality is a complex task that extends beyond simple string matching. Large Reasoning Models (LRMs) are capable of modeling intricate reasoning processes, yet their role in MT evaluation remains insufficiently understood. In this work, we present a systematic investigation into the use of LRMs as evaluators for MT quality, specifically exploring their ability to replicate the Multidimensional Quality Metrics (MQM) assessment process. Our analysis across various LRMs reveals that evaluation materials must be carefully tailored, as these models tend to overanalyze simple cases and exhibit overestimation biases. To address these challenges, we introduce a simple yet effective method for calibrating LRM reasoning by training them on synthetic, human-like MQM evaluation trajectories. Our experiments show that this approach not only reduces the thinking budget required by LRMs but also enhances evaluation performance across different model scales. These findings underscore the potential of efficiently calibrated LRMs to advance fine-grained, automatic MT evaluation.


πŸš€Quick Start

Installation

# Clone the repository
git clone https://github.com/NLP2CT/ThinMQM.git
cd ThinMQM

# Install dependencies
pip install -r requirements.txt

# Install mt-metrics-eval evaluation package & Prepare benchmark data
git clone https://github.com/google-research/mt-metrics-eval.git
cd mt-metrics-eval
pip install .
mkdir $HOME/.mt-metrics-eval
cd $HOME/.mt-metrics-eval
wget https://storage.googleapis.com/mt-metrics-eval/mt-metrics-eval-v2.tgz
tar xfz mt-metrics-eval-v2.tgz

Basic Usage

1. Complete Workflow for running WMT24 experiments.

# Step 1: Generate responses (using existing scripts)
# For ThinMQM model
bash scripts/run_thinmqm.sh 
# For general-purpose LRMs using GEMBA prompt
bash scripts/run_gemba.sh

# Step 2: Extract answers and run meta-evaluation
bash scripts/run_metaeval.sh

Please refer to the comments in the scripts to adjust for your environment. For hyperparameter options, see πŸ”§Configuration.

2. Custom Input Files.

You can evaluate your own translation data with custom input files:

Example Data:

# Run the example script to see how it works
python example_custom_evaluation.py

Example CLI Usage:

MODEL_NAME_OR_PATH="/path/to/rzzhan/ThinMQM-32B"  # Replace with your actual model path

# Set your data paths
SOURCE_FILE="cli_example_data/source.txt"
REFERENCE_FILE="cli_example_data/reference.txt"
SYSTEM_OUTPUTS_DIR="cli_example_data/system_outputs"
OUTPUT_DIR="cli_example_data/results"

SOURCE_LANG="English"
TARGET_LANG="Chinese"
TEMPLATE="thinking"  # For ThinMQM: "thinking" (32B) or "thinking_ref" (7/8B)

# Run ThinMQM evaluation
python main.py custom_thinmqm \
    --model_name="$MODEL_NAME_OR_PATH" \
    --source_file="$SOURCE_FILE" \
    --reference_file="$REFERENCE_FILE" \
    --system_outputs="$SYSTEM_OUTPUTS_DIR" \
    --output_dir="$OUTPUT_DIR" \
    --source_lang="$SOURCE_LANG" \
    --target_lang="$TARGET_LANG" \
    --template="$TEMPLATE" \
    --max_new_tokens=4096 \
    --temperature=0.6

πŸ”§Configuration

πŸ“ Project Structure

β”œβ”€β”€ config/                 # Configuration management
β”‚   └── experiment_config.py
β”œβ”€β”€ evaluators/             # Specific evaluator implementations
β”‚   β”œβ”€β”€ base_evaluator.py   # Core base classes
β”‚   β”œβ”€β”€ thinmqm_evaluator.py
β”‚   β”œβ”€β”€ gemba_evaluator.py
β”‚   └── meta_evaluator.py
β”œβ”€β”€ utils/                  # Utility functions
β”‚   β”œβ”€β”€ answer_extractor.py
β”‚   β”œβ”€β”€ template_utils.py
β”‚   β”œβ”€β”€ mqm_parser.py
β”‚   └── process_results.py
β”œβ”€β”€ scripts/                # Shell scripts for easy execution
β”‚   β”œβ”€β”€ run_thinmqm.sh
β”‚   β”œβ”€β”€ run_gemba.sh
β”‚   └── run_pipeline.sh
β”œβ”€β”€ main.py                 # Main entry point
└── meta_eval_pipeline.md   # Meta-evaluation entry point

Model & Data Card

Released Models HF Model Template Trained Dataset
rzzhan/ThinMQM-32B https://huggingface.co/rzzhan/ThinMQM-32B thinking https://huggingface.co/datasets/rzzhan/ThinMQM-12k/ thinmqm12k_src
rzzhan/ThinMQM-8B https://huggingface.co/rzzhan/ThinMQM-8B thinking_ref https://huggingface.co/datasets/rzzhan/ThinMQM-12k/ thinmqm12k_ref
rzzhan/ThinMQM-7B https://huggingface.co/rzzhan/ThinMQM-7B thinking_ref https://huggingface.co/datasets/rzzhan/ThinMQM-12k/ thinmqm12k_ref

Recommended decoding with temperature=0.6, top_p=0.95.

ThinMQM Model Templates

  • thinking: Source + translation evaluation
  • thinking_ref: Source + reference + translation evaluation

GEMBA Templates

  • src: Source + translation evaluation
  • ref: Reference + translation evaluation
  • joint: Source + reference + translation evaluation

πŸ“Š Meta-Evaluation

ThinMQM reduces thinking budgets while improving the evaluation performance of LRMs at different model scales.

meta-eval

✨Acknowledgments

We thank the open-source community for the excellent tools and libraries that made this work possible, including:


πŸ“¬Contact

For questions, feedback, or collaboration opportunities, feel free to reach out:

πŸ“„License

This project is licensed under the Apache 2.0 License - see the LICENSE file for details.

πŸ“Citation

If you find our model, data, or evaluation code useful, please kindly cite our paper:

@article{zhan2025thinmqm,
      title={Are Large Reasoning Models Good Translation Evaluators? Analysis and Performance Boost}, 
      author={Zhan, Runzhe and Huang, Zhihong and Yang, Xinyi and Chao, Lidia S and Yang, Min and Wong, Derek F},
      year={2025},
      journal = {ArXiv preprint},
      volume = {2510.20780},
      url={https://arxiv.org/abs/2510.20780}, 
}

About

[NeurIPS 2025] Are Large Reasoning Models Good Translation Evaluators? Analysis and Performance Boost

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published