AMC: AutoML for Model Compression and Acceleration on Mobile Devices
Not on the map. Why?
Abstract (arXiv)
Model compression is a critical technique to efficiently deploy neural network models on mobile devices which have limited computation resources and tight power budgets. Conventional model compression techniques rely on hand-crafted heuristics and rule-based policies that require domain experts to explore the large design space trading off among model size, speed, and accuracy, which is usually sub-optimal and time-consuming. In this paper, we propose AutoML for Model Compression (AMC) which leverage reinforcement learning to provide the model compression policy. This learning-based compression policy outperforms conventional rule-based compression policy by having higher compression ratio, better preserving the accuracy and freeing human labor. Under 4x FLOPs reduction, we achieved 2.7% better accuracy than the handcrafted model compression policy for VGG-16 on ImageNet. We applied this automated, push-the-button compression pipeline to MobileNet and achieved 1.81x speedup of measured inference latency on an Android phone and 1.43x speedup on the Titan XP GPU, with only 0.1% loss of ImageNet Top-1 accuracy.
Key facts. Cited by 377 papers in the map (data through 13 Oct 2026). First posted on arXiv 10 Feb 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 131; 2022: 77; 2023: 66; 2024: 42; 2025: 42; 2026: 19.
377 papers.
Sort:
By area:
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,103
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 814
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 318
- 2023A Survey on Model Compression for Large Language Modelscited by 268
- 2023DepGraph: Towards Any Structural Pruningcited by 203
- 2023A Survey on Deep Neural Network Pruning-Taxonomy, Comparison, Analysis, and Recommendationscited by 201
- 2023Structural Pruning for Diffusion Modelscited by 195
- 2022On-Device Training Under 256KB Memorycited by 160
- 2021Efficient Deep Learning: A Survey on Making Deep Learning Models Smaller, Faster, and Bettercited by 154
- 2021Learned Token Pruning for Transformerscited by 139
- 2021FedScale: Benchmarking Model and System Performance of Federated Learning at Scalecited by 136
- 2021Group Fisher Pruning for Practical Network Compressioncited by 131
38 more
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
- 2021QuantumNAS: Noise-Adaptive Search for Robust Quantum Circuitscited by 97
- 2021MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learningcited by 89
- 2022Unified Visual Transformer Compressioncited by 79
- 2021Only Train Once: A One-Shot Neural Network Training And Pruning Frameworkcited by 70
- 2022Vision Transformer Slimming: Multi-Dimension Searching in Continuous Optimization Spacecited by 66
- 2021A Winning Hand: Compressing Deep Networks Can Improve Out-Of-Distribution Robustnesscited by 62
- 2023SparseViT: Revisiting Activation Sparsity for Efficient High-Resolution Vision Transformercited by 60
- 2023TorchRL: A data-driven decision-making library for PyTorchcited by 59
- 2022Advancing Model Pruning via Bi-level Optimizationcited by 56
- 2021CDFI: Compression-Driven Network Design for Frame Interpolationcited by 55
- 2024Survey on Knowledge Distillation for Large Language Models: Methods, Evaluation, and Applicationcited by 55
- 2021Anycost GANs for Interactive Image Synthesis and Editingcited by 54
- 2021A Survey on Green Deep Learningcited by 52
- 2024Efficient Expert Pruning for Sparse Mixture-of-Experts Language Models: Enhancing Performance and Reducing Inference Costscited by 52
- 2021How Well Do Sparse Imagenet Models Transfer?cited by 50
- 2022Revisiting Random Channel Pruning for Neural Network Compressioncited by 50
- 2021FedGEMS: Federated Learning of Larger Server Models via Selective Knowledge Fusioncited by 49
- 2023ZipLM: Inference-Aware Structured Pruning of Language Modelscited by 46
- 2021Hessian-Aware Pruning and Optimal Neural Implantcited by 45
- 2022Efficient Spatially Sparse Inference for Conditional GANs and Diffusion Modelscited by 45
- 2021DeepPayload: Black-box Backdoor Attack on Deep Learning Models through Neural Payload Injectioncited by 44
- 2022Low-rank lottery tickets: finding efficient low-rank neural networks via matrix differential equationscited by 42
- 2022EfficientViT: Multi-Scale Linear Attention for High-Resolution Dense Predictioncited by 42
- 2021Recent Advances on Neural Network Pruning at Initializationcited by 41
- 2022NTIRE 2022 Challenge on Efficient Super-Resolution: Methods and Resultscited by 40
- 2022Structural Pruning via Latency-Saliency Knapsackcited by 40
- 2024Tiny Machine Learning: Progress and Futurescited by 40
- 2021Towards Compact CNNs via Collaborative Compressioncited by 36
- 2022CHEX: CHannel EXploration for CNN Model Compressioncited by 36
- 2022SPDY: Accurate Pruning with Speedup Guaranteescited by 34
- 2021AutoDS: Towards Human-Centered Automation of Data Sciencecited by 32
- 2021Applications and Techniques for Fast Machine Learning in Sciencecited by 32
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2022Pruning-as-Search: Efficient Neural Architecture Search via Channel Pruning and Structural Reparameterizationcited by 32
- 2023OTOV2: Automatic, Generic, User-Friendlycited by 32
- 2021Topology-Aware Network Pruning using Multi-stage Graph Embedding and Reinforcement Learningcited by 30
- 2021SOSP: Efficiently Capturing Global Correlations by Second-Order Structured Pruningcited by 30
50 most cited of 377 All 377 (JSON)
- 2026Unified Response Geometry for Structured Pruning16 Sep 2026
- 2026H3DNAS: Hardware-Aware ONNX-Native 3D Point Cloud Model Compression2 Sep 2026
- 2026MAPLE: MoE Adaptive Plug-and-play Layer-wise Expert allocation15 Aug 2026
- 2026Mitigating Compiler Fusion-Induced Power Bursts in Mobile NPU Inference as the Battery Depletes17 Jul 2026
- 2026Understanding Layer Patching in Model Size Interpolation9 Jul 2026
- 2026Scaling Closed-Loop Feature Channel Configuration with LLMs7 Jul 2026
- 2026HASTE: A Framework for Training-Free, Dynamic, and Steerable Compression of Pre-Trained Convolutional Neural Networks29 Jun 2026
- 2026Hierarchical Reinforcement Learning for Neural Network Compression (HiReLC): Pruning and Quantization24 Jun 2026
- 2026Edge-Constrained UAV Small-Object Detection with P2 Enhancement and Quantum-Inspired Lightweight Structure Search8 Jun 2026
- 2026KLAS: Using Similarity to Stitch Neural Networks for Improved Accuracy-Efficiency Tradeoffs28 May 2026
- 2026WINDQuant: Weight-Informed Neural Decision-Making for Global Mixed-Precision LLM Quantization26 May 2026
- 2026Neural Network Pruning via QUBO Optimization7 Apr 2026
38 more
- 2026Does it matter which Gaussians you pick in 4D Gaussian streaming?18 Mar 2026
- 2026ALADIN: Accuracy-Latency-Aware Design-space Inference Analysis for Embedded AI Accelerators12 Feb 2026
- 2026MatGPTQ: Efficient and Accurate Inference over Nested Quantized Models3 Feb 2026
- 2026AgenticPruner: MAC-Constrained Neural Network Compression via LLM-Driven Strategy Search18 Jan 2026
- 2026Spectral Complex Autoencoder Pruning: A Fidelity-Guided Criterion for Extreme Structured Channel Compression14 Jan 2026
- 2026Closed-Loop LLM Discovery of Non-Standard Channel Priors in Vision Models13 Jan 2026
- 2026GRASP LoRA: GRPO Guided Adapter Sparsity Policy for Cross Lingual Transfer10 Jan 2026
- 2025Tiny, On-Device Decision Makers with the MiniConv Library17 Dec 2025
- 2025SHARe-KAN: Post-Training Vector Quantization for Cache-Resident KAN Inference10 Dec 2025
- 2025HPM-KD: Hierarchical Progressive Multi-Teacher Framework for Knowledge Distillation and Efficient Model Compression10 Dec 2025
- 2025Accelerated Execution of Bayesian Neural Networks using a Single Probabilistic Forward Pass and Code Generation28 Nov 2025
- 2025Decomposed Trust: Privacy, Adversarial Robustness, Ethics, and Fairness in Low-Rank LLMs27 Nov 2025
- 2025FastForward Pruning: Efficient LLM Pruning via Single-Step Reinforcement Learning24 Nov 2025
- 2025Automatic Pruning Discovery for Large Language Models19 Nov 2025
- 2025CAMP-HiVe: Cyclic Pair Merging based Efficient DNN Pruning with Hessian-Vector Approximation for Resource-Constrained Systems9 Nov 2025
- 2025Which Heads Matter for Reasoning? RL-Guided KV Cache Compression9 Oct 2025
- 2025Resource-Aware Neural Network Pruning Using Graph-based Reinforcement Learning4 Sep 2025
- 2025One Shot vs. Iterative: Rethinking Pruning Strategies for Model Compression19 Aug 2025
- 2025FAIR-Pruner: A Flexible Framework for Automatic Layer-Wise Pruning via Tolerance of Difference4 Aug 2025
- 2025Tricks and Plug-ins for Gradient Boosting in Image Classification30 Jul 2025
- 2025Model Compression using Progressive Channel Pruning7 Jul 2025
- 2025SPEAR: Structured Pruning for Spiking Neural Networks via Synaptic Operation Estimation and Reinforcement Learning28 Jun 2025
- 2025Towards Universal & Efficient Model Compression via Exponential Torque Pruning27 Jun 2025
- 2025Structure-Aware Automatic Channel Pruning by Searching with Graph Embedding13 Jun 2025
- 2025Structured Pruning and Quantization for Learned Image Compression2 Jun 2025
- 2025Meta Pruning via Graph Metanetworks : A Universal Meta Learning Framework for Network Pruning24 May 2025
- 2025CoSeP: Complementary Separability Pruning via Class-Separability Clustering19 May 2025
- 2025One-for-All Pruning: A Universal Model for Customized Compression of Large Language Models18 May 2025
- 2025Efficient Unstructured Pruning of Mamba State-Space Models for Resource-Constrained Environments13 May 2025
- 2025A Wireless Collaborated Inference Acceleration Framework for Plant Disease Recognition5 May 2025
- 2025Optimizing LLMs for Resource-Constrained Environments: A Survey of Model Compression Techniques5 May 2025
- 2025BackSlash: Rate Constrained Optimized Training of Large Language Models23 Apr 2025
- 2025CUT: Pruning Pre-Trained Multi-Task Models into Compact Models for Edge Devices14 Apr 2025
- 2025Tin-Tin: Towards Tiny Learning on Tiny Devices with Integer-based Neural Network Training13 Apr 2025
- 2025Kernel-Level Energy-Efficient Neural Architecture Search for Tabular Dataset11 Apr 2025
- 2025A Multicore and Edge TPU-Accelerated Multimodal TinyML System for Livestock Behavior Recognition10 Apr 2025
- 2025Generative Artificial Intelligence for Internet of Things Computing: A Systematic Survey10 Apr 2025
- 2025Thanos: A Block-wise Pruning Algorithm for Efficient Large Language Model Compression6 Apr 2025
50 newest of 377 All 377 (JSON)
- 2024Survey on Knowledge Distillation for Large Language Models: Methods, Evaluation, and Applicationcited by 55
- 2021Multi-Scale Aligned Distillation for Low-Resolution Detectioncited by 24
- 2021Semantic Relation Preserving Knowledge Distillation for Image-to-Image Translationcited by 13
- 2021LANA: Latency Aware Network Accelerationcited by 11
- 2021Follow Your Path: a Progressive Method for Knowledge Distillationcited by 9
- 2021Scene-adaptive Knowledge Distillation for Sequential Recommendation via Differentiable Architecture Searchcited by 3
- 2024Robust Knowledge Distillation Based on Feature Variance Against Backdoored Teacher Modelcited by 2
- 2022A Novel Architecture Slimming Method for Network Pruning and Knowledge Distillationcited by 1
- 2022ORC: Network Group-based Knowledge Distillation using Online Role Changecited by 1
- 2025HPM-KD: Hierarchical Progressive Multi-Teacher Framework for Knowledge Distillation and Efficient Model Compressioncited by 1
- 2025HPM-KD: Hierarchical Progressive Multi-Teacher Framework for Knowledge Distillation and Efficient Model Compression10 Dec 2025
- 2024Survey on Knowledge Distillation for Large Language Models: Methods, Evaluation, and Application2 Jul 2024
- 2024Robust Knowledge Distillation Based on Feature Variance Against Backdoored Teacher Model1 Jun 2024
- 2022ORC: Network Group-based Knowledge Distillation using Online Role Change1 Jun 2022
- 2022A Novel Architecture Slimming Method for Network Pruning and Knowledge Distillation21 Feb 2022
- 2021Multi-Scale Aligned Distillation for Low-Resolution Detection14 Sep 2021
- 2021Follow Your Path: a Progressive Method for Knowledge Distillation20 Jul 2021
- 2021Scene-adaptive Knowledge Distillation for Sequential Recommendation via Differentiable Architecture Search15 Jul 2021
- 2021LANA: Latency Aware Network Acceleration12 Jul 2021
- 2021Semantic Relation Preserving Knowledge Distillation for Image-to-Image Translation30 Apr 2021
- 2024AutoDFP: Automatic Data-Free Pruning via Channel Similarity Reconstructioncited by 1
- 2024AutoSculpt: A Pattern-based Model Auto-pruning Framework Using Reinforcement Learning and Graph Learningcited by 1
- 2025Tiny, On-Device Decision Makers with the MiniConv Librarycited by 1
- 2024Emergency Computing: An Adaptive Collaborative Inference Method Based on Hierarchical Reinforcement Learningcited by 0
- 2025SPEAR: Structured Pruning for Spiking Neural Networks via Synaptic Operation Estimation and Reinforcement Learningcited by 0
- 2025Resource-Aware Neural Network Pruning Using Graph-based Reinforcement Learningcited by 0
- 2026Hierarchical Reinforcement Learning for Neural Network Compression (HiReLC): Pruning and Quantizationcited by 0
- 2026Hierarchical Reinforcement Learning for Neural Network Compression (HiReLC): Pruning and Quantization24 Jun 2026
- 2025Tiny, On-Device Decision Makers with the MiniConv Library17 Dec 2025
- 2025Resource-Aware Neural Network Pruning Using Graph-based Reinforcement Learning4 Sep 2025
- 2025SPEAR: Structured Pruning for Spiking Neural Networks via Synaptic Operation Estimation and Reinforcement Learning28 Jun 2025
- 2024AutoSculpt: A Pattern-based Model Auto-pruning Framework Using Reinforcement Learning and Graph Learning24 Dec 2024
- 2024AutoDFP: Automatic Data-Free Pruning via Channel Similarity Reconstruction13 Mar 2024
- 2024Emergency Computing: An Adaptive Collaborative Inference Method Based on Hierarchical Reinforcement Learning3 Feb 2024
- 2023Structural Pruning for Diffusion Modelscited by 195
- 2022Efficient Spatially Sparse Inference for Conditional GANs and Diffusion Modelscited by 45
- 2024Not All Prompts Are Made Equal: Prompt-based Pruning of Text-to-Image Diffusion Modelscited by 13
- 2024Mixture of Efficient Diffusion Experts Through Automatic Interval and Sub-Network Selectioncited by 11
- 2024Mixture of Efficient Diffusion Experts Through Automatic Interval and Sub-Network Selection23 Sep 2024
- 2024Not All Prompts Are Made Equal: Prompt-based Pruning of Text-to-Image Diffusion Models17 Jun 2024
- 2023Structural Pruning for Diffusion Models18 May 2023
- 2022Efficient Spatially Sparse Inference for Conditional GANs and Diffusion Models3 Nov 2022
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,103
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 814
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 318
- 2023A Survey on Model Compression for Large Language Modelscited by 268
- 2023DepGraph: Towards Any Structural Pruningcited by 203
- 2023A Survey on Deep Neural Network Pruning-Taxonomy, Comparison, Analysis, and Recommendationscited by 201
- 2022On-Device Training Under 256KB Memorycited by 160
- 2021Efficient Deep Learning: A Survey on Making Deep Learning Models Smaller, Faster, and Bettercited by 154
- 2021FedScale: Benchmarking Model and System Performance of Federated Learning at Scalecited by 136
- 2021Group Fisher Pruning for Practical Network Compressioncited by 131
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
- 2021QuantumNAS: Noise-Adaptive Search for Robust Quantum Circuitscited by 97
12 most cited of 348 All 348 (JSON)
- 2026Unified Response Geometry for Structured Pruning16 Sep 2026
- 2026H3DNAS: Hardware-Aware ONNX-Native 3D Point Cloud Model Compression2 Sep 2026
- 2026Mitigating Compiler Fusion-Induced Power Bursts in Mobile NPU Inference as the Battery Depletes17 Jul 2026
- 2026Understanding Layer Patching in Model Size Interpolation9 Jul 2026
- 2026Scaling Closed-Loop Feature Channel Configuration with LLMs7 Jul 2026
- 2026HASTE: A Framework for Training-Free, Dynamic, and Steerable Compression of Pre-Trained Convolutional Neural Networks29 Jun 2026
- 2026Edge-Constrained UAV Small-Object Detection with P2 Enhancement and Quantum-Inspired Lightweight Structure Search8 Jun 2026
- 2026KLAS: Using Similarity to Stitch Neural Networks for Improved Accuracy-Efficiency Tradeoffs28 May 2026
- 2026WINDQuant: Weight-Informed Neural Decision-Making for Global Mixed-Precision LLM Quantization26 May 2026
- 2026Neural Network Pruning via QUBO Optimization7 Apr 2026
- 2026Does it matter which Gaussians you pick in 4D Gaussian streaming?18 Mar 2026
- 2026ALADIN: Accuracy-Latency-Aware Design-space Inference Analysis for Embedded AI Accelerators12 Feb 2026
12 newest of 348 All 348 (JSON)