Counter-Interference Adapter for Multilingual Machine Translation

Overview

This is the (early-access) implementation of EMNLP'21 Finding paper Counter-Interference Adapter for Multilingual Machine Translation

The code is based on NeurST Toolkits

Data prepare

Please follow data/DataProcess.md to download and preprocess data.

Training and Inference

The major class for the paper is TransformerCIAT in neurst/models/transformer_CIAT.py

The example config files are in configs/, please modify the file path in those yml files.

Pretaining:

python3 -m neurst.cli.run_exp.py --config configs/pretrain.yml

Tune on En-De with adapter:

python3 -m neurst.cli.run_exp.py --config configs/tune-on-en2de.yml

Inference:

python3 -m neurst.cli.run_exp.py --config configs/inference.yml