atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

In defense of parameter sharing for model-compression

ICLR 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Joint pruning and quantization · 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.

Shows randomized parameter sharing (RPS) beats pruning and smaller models at high compression, and fixes ROAST's stability and Pareto-continuity issues via STABLE-RPS (gradient scaling plus RANDOMFOLD mapping).

Problem

Reducing a model's parameter memory footprint at the start of training: whether pruning or randomized parameter sharing (RPS) gives a better memory-accuracy tradeoff, and ROAST's instability and lack of Pareto-continuity.

Approach

Proposes STABLE-RPS, which adds a gradient scaler (theory-driven or effective-update) to remove ROAST's sensitivity to init-stdev, and a RANDOMFOLD mapping with optimal load factor so that compression 1x recovers the full model; it argues parameter sharing beats pruning in capacity.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use STABLE-RPS gradient scaling instead of ROAST's unscaled updates (ablation)ROAST · CIFAR-10 (ResNet-20), across compression levelsaccuracy stability across init-stdev: ROAST: 10.0 to 91.5; STABLE-RPS: 91.4 to 91.2 (1.33×); ROAST: 10.0 to 87.2; STABLE-RPS: 87.3 to 87.4 (10×); ROAST: 10.0 Helped
Use STABLE-RPS gradient scaling instead of ROAST's unscaled updates (ablation)ROAST · CIFAR-100 (VGG-11), across compression levelsaccuracy stability across init-stdev: ROAST: 1.0 to 67.5; STABLE-RPS: 68.1 to 68.9 (1.33×); ROAST: 38.7 to 66.0; STABLE-RPS: 65.8 to 66.5 (10×); ROAST: 1.0 toHelped
Replace ROAST mapping with RANDOMFOLD mapping (ablation)ROAST · CIFAR-10 and CIFAR-100 vision tasks, across compression levelsaccuracyHelped
Use RANDOMFOLD mapping rather than element-wise or ROAST mapping at full model memory (ablation)RPS · Vision tasks at full model memoryaccuracyHelped
Use STABLE-RPS parameter sharing rather than RAND pruningModel compression · Vision tasks, all compression levelsaccuracyHelped
Use STABLE-RPS parameter sharing rather than moderately informed pruningModel compression · Vision tasks, especially higher compressionaccuracyHelped
Use STABLE-RPS rather than Lottery Ticket RewindingModel compression · Vision tasks, high compressionaccuracyHelped
Use STABLE-RPS rather than a smaller model with reduced hidden dimensionsModel compression · Vision tasksaccuracyHelped
Increase ROAST init-stdev from 0.1 to 10 (ablation)ROAST · CIFAR-10 (ResNet-20) and CIFAR-100 (VGG-11), across compression levelsaccuracy: CIFAR-10: 91.5 to 79.4 (1.33×), 87.2 to 77.3 (10×), 75.5 to 66.8 (100×), 53.1 to 52.1 (1000×); CIFAR-100: 67.5 to 19.8 (Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 CIFAR-100

Benchmarks: CIFAR-10 CIFAR-100

Metrics: accuracy compression factor norm of solution

Baselines: RAND MAG SNIP SYNFLOW GRASP LTR smaller model ROAST

Builds on: ROAST HashedNet ROBE Lottery Ticket Hypothesis

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