atlas

Research map / Large language models / Instruction and fine-tuning

On Giant's Shoulders: Effortless Weak to Strong by Dynamic Logits Fusion

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Efficient LLM fine-tuning systems · 1 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 dynamic logit fusion that transfers knowledge from a series of task-specific small models (1.1B/7B LLaMA2 experts) to a 13B LLaMA2 at each decoding step, with per-step weights learned via a KL-constrained objective, requiring no gradient updates to the large model.

Problem

Logit arithmetic for weak-to-strong transfer uses a static, pre-set weight on the small model's fine-tuning delta and a single expert, which limits performance and generalization to unseen tasks; fine-tuning large models is memory-expensive.

Approach

At each decoding step, per-expert weights alpha_t are chosen by minimizing a symmetric KL-distance matching objective between the fused large-model distribution and each small expert's shift from its base, so knowledge from multiple task experts is adaptively fused into a large model without training it.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use dynamic logit fusion instead of Proxy TuningLLaMA2-13B with 7B experts · GSM8KEM: 37.68 -> 39.34Helped
Use dynamic logit fusion instead of Proxy TuningLLaMA2-13B with 7B experts · TruthfulQAaccuracy: 61.02 -> 61.56Helped
Use dynamic logit fusion instead of Proxy TuningLLaMA2-13B with 7B experts · TriviaQAEM: 52.81 -> 57.11Helped
Use dynamic logit fusion instead of Proxy TuningLLaMA2-13B with 7B experts · CNN/DMROUGE-2: 14.37 -> 15.31Helped
Use dynamic logit fusion instead of Proxy TuningLLaMA2-13B with 7B experts · MMLUaccuracy: 56.24 -> 57.15Helped
Transfer knowledge from 7B experts with dynamic fusion rather than full fine-tuning the 13B modelLLaMA2-13B · TriviaQAEM: 56.36 -> 57.11Helped
Transfer knowledge from 7B experts rather than full fine-tuning the 13B modelLLaMA2-13B · CNN/DMROUGE-2: 15.50 -> 15.31Hurt
Transfer knowledge from 7B experts with dynamic fusion rather than 7B multi-task tuningLLaMA2-13B · Multi-task averageaverage score: 34.89 -> 39.42Helped
Transfer knowledge from 7B experts with dynamic fusion rather than 7B multi-task tuningLLaMA2-13B · MMLU (unseen task)accuracy: 35.82 -> 51.31Helped
Transfer knowledge from 7B experts with dynamic fusion rather than 7B multi-task tuningLLaMA2-13B · Seen tasks averageaverage score: 33.96 -> 27.53Hurt
Replace predefined alpha values with dynamically adjusted alpha (ablation)Logit transfer method · Single-task and multi-task benchmarksperformanceHelped
Add in-context learning to dynamic logit fusionLLaMA2-13B with 7B experts · Multi-domain tasksperformance: 18.3%Helped

Surprising findings

Limitations

Experimental setup

Datasets: GSM8K TruthfulQA TriviaQA CNN/DailyMail MMLU

Benchmarks: GSM8K TruthfulQA TriviaQA CNN/DM MMLU

Metrics: exact match accuracy Rouge-2 average performance performance gap closed

Baselines: Proxy Tuning Full Fine-tuning (13B) LoRA Tuning (13B) Multi-Task Tuning In-Context Learning (5-shot) Task Arithmetic

Builds on: Proxy Tuning DPO Contrastive Decoding Task Arithmetic

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