atlas

Research map / Large language models / Language, safety and interpretability

MindMerger: Efficiently Boosting LLM Reasoning in non-English Languages

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Multilingual LLM data and culture · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes MindMerger, which merges a frozen multilingual encoder (mT5-xl) with a frozen LLM via a trained mapping layer in a two-stage scheme, improving multilingual reasoning especially for low-resource languages.

Problem

LLM reasoning in non-English languages, especially low-resource ones, lags behind English; existing relearning and translate-then-replace methods underuse the LLM's built-in reasoning and language understanding.

Approach

Feed the LLM the undecoded hidden states of a multilingual encoder through a trainable mapping layer, concatenated with the LLM's own query embeddings (augmentation rather than replacement), trained first on general bilingual translation pairs (mapping stage) and then on query-translation task data (augmentation stage), with both base models frozen.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use MindMerger-Soft instead of the strongest reported baselineLlama 2-7B · MGSMaverage accuracy: 50.6 -> 57.3Helped
Use MindMerger-Soft instead of the strongest reported baselineLlama 2-7B · MSVAMPaverage accuracy: 57.2 -> 60.4Helped
Use MindMerger-Soft instead of the strongest reported baselineLlama 2-7B · X-CSQAaverage accuracy: 56.9 -> 61.0Helped
Use MindMerger-Soft instead of the strongest reported baselineLlama 2-7B · XNLIaverage accuracy: 76.5 -> 78.4Helped
Add the mapping stage (ablation)MindMerger-Soft · MGSMlow-resource average accuracy: 31.5 -> 51.7; 39.1 -> 53.5Helped
Add the mapping stage (ablation)MindMerger-Soft · MSVAMPaverage accuracy: 50.8 -> 59.1; 53.1 -> 60.4Helped
Add the augmentation stage (ablation)MindMerger-Soft · MGSMaverage accuracy: 42.5 -> 56.3; 42.6 -> 57.3Helped
Add the augmentation stage (ablation)MindMerger-Soft · MSVAMPaverage accuracy: 49.1 -> 59.1; 52.4 -> 60.4Helped
Add the original LLM query representation alongside multilingual features (ablation)MindMerger-Soft · MGSMaverage accuracy: 54.6 -> 56.3; 55.9 -> 57.3Helped
Add the original LLM query representation alongside multilingual features (ablation)MindMerger-Soft · MSVAMPaverage accuracy: 55.6 -> 59.1; 57.3 -> 60.4Helped
Replace 2-layer MLP mapping with linear mapping (ablation)MindMerger-Soft · MGSMaverage accuracy: 57.3 -> 55.3Hurt
Replace 2-layer MLP mapping with QFormer (ablation)MindMerger-Soft · MGSMaverage accuracy: 57.3 -> 22.8Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Lego-MT MetaMathQA MultiNLI X-CSQA OpenBookQA ARC QASC XNLI dev set MGSM translated queries M2M100-1.2B translations

Benchmarks: MGSM MSVAMP X-CSQA XNLI Flores-101

Metrics: accuracy average accuracy across low-resource and high-resource langu spBLEU cosine similarity Recall@1

Baselines: MonoReason MultiReason-SFT MultiReason-Lora QAlign LangBridge Translate-En

Builds on: LangBridge QAlign MetaMath

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy