Prompt4BLI

This repository is the official PyTorch implementation of the following paper:

Yaoyiran Li, Anna Korhonen, and Ivan Vulić. 2023. On Bilingual Lexicon Induction with Large Language Models. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing (EMNLP 2023). [Paper] [OpenReview]

Prompt4BLI aims to address the Bilingual Lexicon Induction (BLI) / Word Translation tasks with autoregressive Large Language Models (LLMs). We for the first time demonstrate that prompting multilingual LLMs for BLI outperforms traditional BLI approaches which rely on calculating cross-lingual word embeddings (CLWEs). While we show that prompting off-the-shelf LLMs can already establish new state-of-the-art BLI performance on many BLI language pairs (our main experimental setup), the Prompt4BLI repo also provides code for BLI-oriented fine-tuning which can further improve the results (as a side experiment, demonstrated on smaller-scale LLMs).

Traditional methods rely on learning parameterized CLWE mappings or cross-lingual word pair scoring functions and usually tackle BLI in three setups: (1) Supervised, 5K seed translation pairs; (2) Semi-Supervised, 1K seed translation pairs; (3) Unsupervised, 0 seed translation pairs. (cf. our previous work ContrastiveBLI and BLICEr). Different from traditional methods, Prompt4BLI only makes use of off-the-shelf LLMs, not requiring LLM fine-tuning nor updating any learnable parameters. Our work considers the following prompting setups:

• Few-Shot Prompting: We propose to retrieve a subset of the seed translation pairs (nearest neighbour retrieval) as in-context examples for prompting. Corresponds to the traditional Supervised and Semi-Supervised BLI setups where the seed bilingual dictionary size is 5K and 1K respectively.
• Zero-Shot Prompting: No in-context examples are used. Corresponds to the traditional Unsupervised BLI setup.

Follow-up Work:

Update: please see our follow-up work SAIL (ACL 2024) where we further improve unsupervised BLI by (1) inferring a high-confidence word translation dictionary with zero-shot prompting, (2) then optionally refining the high-confidence dictionary iteratively with few-shot prompting where the in-context examples are from the high-confidence dictionary in the previous iteration, and (3) finally conducting evaluation on the BLI test set with few-shot prompting also deriving in-context samples from the latest high-confidence dictionary. The whole process does not leverage any ground-truth word translation pairs for training/few-shot learning and improves the BLI scores by typically 10 ~ 15 P@1 points comparing to zero-shot prompting.

Dependencies

• PyTorch>=1.10.1
• Transformers>=4.28.1

LLMs Used in Our Work

LLM	(Hugging Face) Model ID
mT5-small	"google/mt5-small"
mT5-base	"google/mt5-base"
mT5-large	"google/mt5-large"
mT5-xl	"google/mt5-xl"
mT5-xxl	"google/mt5-xxl"
mT0-small	"bigscience/mt0-small"
mT0-base	"bigscience/mt0-base"
mT0-large	"bigscience/mt0-large"
mT0-xl	"bigscience/mt0-xl"
mT0-xxl	"bigscience/mt0-xxl"
XGLM-564M	"facebook/xglm-564M"
XGLM-1.7B	"facebook/xglm-1.7B"
XGLM-2.9B	"facebook/xglm-2.9B"
XGLM-4.5B	"facebook/xglm-4.5B"
XGLM-7.5B	"facebook/xglm-7.5B"
mGPT	"sberbank-ai/mGPT"
LLaMA-7B	"huggyllama/llama-7b"
LLaMA-13B	"huggyllama/llama-13b"
LLaMA2-7B	"meta-llama/Llama-2-7b-hf"
LLaMA2-13B	"meta-llama/Llama-2-13b-hf"

LLaMA2-7B and LLaMA2-13B models are investigated in our follow-up work SAIL (ACL 2024), and we also integrate the LLaMA2 models into the current code repo. SAIL also conducts zero-shot prompting with GPT-3.5, and GPT-4. Please refer to SAIL for the details.

Data

Following ContrastiveBLI and BLICEr, our data is obtained from the XLING (8 languages, 56 BLI directions in total) and PanLex-BLI (15 lower-resource languages, 210 BLI directions in total).

Get XLING data:

sh get_xling_data.sh

For PanLex-BLI, please see ./get_panlex_data, where we provide the code for deriving the monolingual word embeddings.

Run the Code

Prepare BLI Data and Extract In-Context Examples for Few-Shot Prompting (XLING):

python run_extract_vocabularies.py
python run_extract_bli_data.py

Prepare BLI Data and Extract In-Context Examples for Few-Shot Prompting (PanLex-BLI):

python run_extract_vocabularies_panlex.py
python run_extract_bli_data_panlex.py

(Optional) Run BLI-Oriented LLM Fine-Tuning (define LLM dirs, learning rate, batch size, and random seed in run_training.py):

python run_prepare_training_data.py
python run_training.py

Run BLI Evaluation (define seed dictionary size, n_shot, LLM dir, and language pairs to evaluate manually in run_bli.py):

python run_bli.py

Citation

Please cite our paper if you find Prompt4BLI useful.

@inproceedings{li-etal-2023-bilingual,
    title     = {On Bilingual Lexicon Induction with Large Language Models},
    author    = {Li, Yaoyiran and Korhonen, Anna and Vuli{\'c}, Ivan},
    booktitle = {Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing},    
    year      = {2023}
}

Follow-up Work (Code available at SAIL):

@inproceedings{li-etal-2024-self,
    title     = {Self-Augmented In-Context Learning for Unsupervised Word Translation},
    author    = {Li, Yaoyiran and Korhonen, Anna and Vuli{\'c}, Ivan},
    booktitle = {Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics},    
    year      = {2024}
}