atlas

Research map / Large language models / Retrieval and knowledge

Seeking Neural Nuggets: Knowledge Transfer in Large Language Models from a Parametric Perspective

ICLR 2024 · Large language models › Retrieval and knowledge · approach: Knowledge editing in LLMs · 0 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 parametric knowledge transfer: sensitivity-based extraction of task-specific submatrices from a larger LLaMA teacher, injected into a smaller student's LoRA module, showing gains over plain LoRA fine-tuning on four benchmarks.

Problem

Whether task-specific implicit knowledge stored in the parameters of a larger LLM can be transferred to smaller LLMs of different scale, as an alternative to output-based distillation.

Approach

Extract task-relevant parameters from a teacher via sensitivity scores (first-order Taylor approximation on ~32 seed samples), select layers and submatrices matching the student's dimensions, factorize them with SVD, and inject them into the student's LoRA initialization (with the extracted part subtracted from the base weights so training starts at the pretrained point).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
transfer sensitivity-extracted parameters from LLaMA-1 30B instead of using 7B-LoRA aloneLLaMA-1 7B · four-benchmark averageaverage score: 27.00 -> 28.63Helped
transfer sensitivity-extracted parameters from LLaMA-2 13B instead of using 7B-LoRA aloneLLaMA-2 7B · four-benchmark averageaverage score: 32.28 -> 34.47Helped
increase teacher scale from 13B to 30BLLaMA-1 7B · GSMaccuracy: 18.73 -> 18.63Hurt
extract parameters using task-specific seed data rather than generic transfer data (ablation)LLaMA-2 7B · GSM 0-shotaccuracy: 23.38 -> 25.30Helped
initialize LoRA with the proposed weight-preserving strategy rather than the LoRA-paper initialization (ablation)LLaMA-1 7B · four-benchmark averageaverage scoreHelped
increase the number of sensitivity seed samples (ablation)student model · knowledge transferperformance varianceHelped
select teacher layers by sensitivity rather than random, top, or last-layer selection (ablation)LLaMA-1 teacher-to-student transfers · knowledge transferstudent performanceHelped
transfer all available modules rather than the embedding layer alone (ablation)student model · knowledge transferstudent performanceHelped
transfer coherent submatrices rather than isolated single weights (ablation)student model · knowledge transferstudent performanceHelped
use sensitivity-based rather than random submatrix extraction (ablation)LLaMA-1 13B to 7B · four-benchmark averageaverage score: 27.83 -> 28.22Helped
use sensitivity-based rather than random submatrix extraction (ablation)LLaMA-1 30B to 7B · four-benchmark averageaverage score: 28.11 -> 28.63Helped
increase LoRA rank beyond 16 when transferring all modules (ablation)LLaMA-1 7B · four-benchmark averageaverage score: 28.22 -> 28.27; 28.63 -> 28.54No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: GSM8K MMLU Super-NaturalInstructions LIMA AlpacaFarm

Benchmarks: GSM8K MMLU Super-NaturalInstructions AlpacaFarm

Metrics: accuracy exact match ROUGE-L win rate

Baselines: LoRA fine-tuning (no transfer) KD SeqKD Gaussian LoRA initialization random submatrix extraction

Builds on: LoRA knowledge distillation first-order Taylor parameter sensitivity pruning

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