atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

OSSCAR: One-Shot Structured Pruning in Vision and Language Models with Combinatorial Optimization

ICML 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Structured network pruning · 2 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 OSSCAR, a one-shot structured pruning framework that reformulates a layer-wise reconstruction objective as a combinatorial problem solved by a low-rank-update local search, scaling to OPT models up to 30B for pruning heads and FFN neurons and to CNNs.

Problem

One-shot (post-training, no retraining) structured pruning of large vision and language models, where existing methods are too slow or memory-hungry for billion-parameter models and lose accuracy at high sparsity.

Approach

Formulate structured pruning as a layer-wise quadratic problem with group-structured binary selection, reduce the quadratic matrix to a compact form by exploiting shared coefficients, and solve the combinatorial selection with local swap search that uses low-rank (Woodbury-style) updates to re-evaluate the objective cheaply.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use OSSCAR instead of the strongest listed one-shot baselineResNet20 · CIFAR10, 3.5x speedupaccuracy: 58.68 -> 61.82 (±0.98)Helped
Use OSSCAR instead of the strongest listed one-shot baselineMobileNetV1 · ImageNet, 2.2x speedupaccuracy: 45.77 -> 52.07 (±0.57)Helped
Use OSSCAR instead of the strongest listed one-shot baselineResNet50 · ImageNet, 1.9x speedupaccuracy: 29.23 -> 38.45 (±0.63)Helped
Use OSSCAR instead of ZipLMOPT-1.3B · raw-WikiText2 test, 2.0x speedupperplexity: 3529 ($\pm$346) -> 37.87 ($\pm$0.64)Helped
Use OSSCAR instead of ZipLMOPT-2.7B · raw-WikiText2 test, 2.0x speedupperplexity: 3611 ($\pm$840) -> 29.48 ($\pm$0.46)Helped
Use OSSCAR instead of ZipLMOPT-1.3B · pruningpruning time: 9161 ($\pm$670) -> 1455 ($\pm$4.00)Helped
Use OSSCAR instead of ZipLMOPT-2.7B · pruningpruning time: 22839 ($\pm$999) -> 2842 ($\pm$12.0)Helped
Increase calibration sample size from 128 to 2048 (ablation)OSSCAR · OPT models, 2x speedupperplexity: up to 3.5 pointsHelped
Vary the local-search parameters $\hat t$ and $\hat p$ (ablation)OSSCAR · structured pruning; benchmark-specific ablationperformanceNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: C4 CIFAR10 ImageNet

Benchmarks: CIFAR10 ImageNet WikiText2 PTB C4 validation

Metrics: accuracy perplexity inference speedup pruning time

Baselines: Magnitude pruning MP+ ZipLM CHIP FPGM Lasso ThiNet

Builds on: ZipLM Optimal Brain Surgeon ThiNet CHIP

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