atlas

Research map / Architectures and efficiency / Neural architecture search

Search for Efficient Large Language Models

NeurIPS 2024 · Architectures and efficiency › Neural architecture search · approach: AutoML pipeline search · 5 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 a training-free evolutionary architecture search that selects structural subnets (rows/columns) of pretrained LLMs and reforms their weights with an ADMM-based reconstruction using 128 calibration samples.

Problem

Compressing LLMs by discovering efficient architectures (structural subnets) without retraining, where traditional NAS is too costly for large models and most compression methods focus on weight optimization rather than architecture.

Approach

Initialize subnets by importance scores (WoodFisher/SparseGPT-style saliency), run evolutionary search with a mask mutation operator over inheriting ratios, depth, and channel indices using perplexity on a few samples, then reform inherited weights via ADMM to compensate for omitted weights.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use the searched subnet instead of FLAP at an 80% inheriting ratioLLaMA-7B · WikiText2perplexity: 7.40 -> 6.89Helped
Use the searched subnet instead of FLAP at an 80% inheriting ratioLLaMA-7B · Common-sense reasoning averageaccuracy: 58.58 -> 59.71Helped
Use the searched subnet instead of FLAP at an 80% inheriting ratioLLaMA-13B · PTBperplexity: 23.33 -> 22.66Helped
Use the searched subnet instead of FLAP at an 80% inheriting ratioVicuna-7B · WikiText2perplexity: 8.90 -> 8.20Helped
Use the searched subnet instead of SliceGPT at an 80% inheriting ratioOPT-1.3B · WikiText2perplexity: 20.17 -> 19.23Helped
Use the searched subnet instead of FLAP at a 60% inheriting ratioLLaMA-7B · WikiText2perplexity: 12.34 -> 10.21Helped
Use the searched subnet instead of FLAP at a 50% inheriting ratioLLaMA-30B · WikiText2perplexity: 13.41 -> 8.01Helped
Search on PTB training data rather than WikiText2 (ablation)LLaMA-7B · WikiText2, 80% inheriting ratioperplexity: 6.89 -> 7.08Hurt
Search on PTB training data rather than WikiText2 (ablation)LLaMA-7B · PTB, 80% inheriting ratioperplexity: 36.06 -> 34.87Helped
Increase reformation calibration examples from 128 to 512 or 1024 (ablation)LLaMA-7B subnet · WikiText2perplexityNo clear effect
Increase ADMM reformation iterations toward 30 (ablation)LLaMA-7B subnet, 80% inheriting ratio · WikiText2perplexityHelped
Increase ADMM reformation iterations beyond 30 to 50 (ablation)LLaMA-7B subnet, 80% inheriting ratio · WikiText2perplexityNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: WikiText2 PTB

Benchmarks: WikiText2 PTB BoolQ PIQA HellaSwag WinoGrande ARC-easy ARC-challenge OpenBookQA

Metrics: perplexity zero-shot accuracy average accuracy GPU memory generation speed

Baselines: LLM-Pruner SliceGPT FLAP

Builds on: SparseGPT WoodFisher LLM-Pruner SliceGPT FLAP

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