atlas

Research map / Large language models / Retrieval and knowledge

Knowledge Fusion of Large Language Models

ICLR 2024 · Large language models › Retrieval and knowledge · approach: Knowledge-augmented question answering · 8 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 FuseLLM, which fuses several heterogeneous source LLMs (Llama-2, OpenLLaMA, MPT) into one target LLM by continually training it to match the source models' fused next-token distributions, using MinED token alignment and MinCE fusion.

Problem

Combining the capabilities of multiple pre-trained LLMs with different architectures and tokenizers is impractical via weight merging or ensembling, which require identical architectures or costly multi-model inference.

Approach

Externalize each source LLM's next-token distributions on a raw corpus, align them across tokenizers with minimum edit distance, fuse them with a min-cross-entropy rule, and train the target LLM to match the fused distribution alongside the CLM loss.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
fuse distributions from Llama-2, OpenLLaMA, and MPT and continually train the targetLlama-2 7B · BBH, Avg. 27 Tasksexact match accuracy: 39.70 -> 41.75Helped
fuse distributions from Llama-2, OpenLLaMA, and MPT and continually train the targetLlama-2 7B · Common Sense, Avg. 5 Tasksaccuracy: 63.76 -> 64.56Helped
fuse distributions from Llama-2, OpenLLaMA, and MPT and continually train the targetLlama-2 7B · MultiPL-E, Avg. 10 Taskspass@1: 14.63 -> 15.56Helped
use three source models instead of two (ablation)FuseLLM · BBHexact match accuracy: 41.00 -> 41.75; 41.16 -> 41.75Helped
use three source models instead of two (ablation)FuseLLM · Common Senseaccuracy: 64.50 -> 64.56; 64.51 -> 64.56No clear effect
use three source models instead of two (ablation)FuseLLM · MultiPL-Epass@1: 15.51 -> 15.56; 15.47 -> 15.56No clear effect
replace exact-match token alignment with MinED (ablation)FuseLLM · BBHexact match accuracy: 41.57 -> 41.75Helped
replace exact-match token alignment with MinED (ablation)FuseLLM · MultiPL-Epass@1: 15.49 -> 15.56Helped
replace exact-match token alignment with MinED (ablation)FuseLLM · Common Senseaccuracy: 64.24 -> 64.56Helped
replace AvgCE weighted averaging with MinCE selection (ablation)FuseLLM · BBHexact match accuracy: 41.04 -> 41.75Helped
replace AvgCE weighted averaging with MinCE selection (ablation)FuseLLM · MultiPL-Epass@1: 15.39 -> 15.56Helped
replace AvgCE weighted averaging with MinCE selection (ablation)FuseLLM · Common Senseaccuracy: 63.98 -> 64.56Helped

Surprising findings

Limitations

Experimental setup

Datasets: MiniPile The Pile RedPajama Evol-Instruct ShareGPT Open-Platypus

Benchmarks: Big-Bench Hard CommonSense (ARC, BoolQ, HellaSwag, OpenBookQA) MultiPL-E TriviaQA DROP LAMBADA IWSLT2017 SciBench Vicuna Benchmark

Metrics: exact match accuracy accuracy pass@1 BLEU bits per UTF-8 byte perplexity

Baselines: Llama-2 CLM OpenLLaMA MPT Llama-2 KD LLM-Blender Ensemble Weight Merging

Builds on: knowledge distillation LLM-Blender exact-match token alignment teacher annealing

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