Making Models Shallow Again: Jointly Learning to Reduce Non-Linearity and Depth for Latency-Efficient Private Inference
Not on the map. Why?
Abstract (arXiv)
Large number of ReLU and MAC operations of Deep neural networks make them ill-suited for latency and compute-efficient private inference. In this paper, we present a model optimization method that allows a model to learn to be shallow. In particular, we leverage the ReLU sensitivity of a convolutional block to remove a ReLU layer and merge its succeeding and preceding convolution layers to a shallow block. Unlike existing ReLU reduction methods, our joint reduction method can yield models with improved reduction of both ReLUs and linear operations by up to 1.73x and 1.47x, respectively, evaluated with ResNet18 on CIFAR-100 without any significant accuracy-drop.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 14 of its 26 references that are in the map. First posted on arXiv 26 Apr 2023.
Builds on
14 of 26 references.
References · 14
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Wide Residual Networkscited by 3,057
- 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 491
- 2018Gazelle: A Low Latency Framework for Secure Neural Network Inferencecited by 193
- 2021DeepReDuce: ReLU Reduction for Fast Private Inferencecited by 50
- 2021CryptGPU: Fast Privacy-Preserving Machine Learning on the GPUcited by 46
- 2020CryptoNAS: Private Inference on a ReLU Budgetcited by 40
- 2020A Tunable Robust Pruning Framework Through Dynamic Network Rewiring of DNNscited by 37
- 2022Selective Network Linearization for Efficient Private Inferencecited by 31
- 2023Learning to Linearize Deep Neural Networks for Secure and Efficient Private Inferencecited by 20
- 2022DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networkscited by 19
- 2021Sphynx: ReLU-Efficient Network Design for Private Inferencecited by 15
Built on by
Citing papers per year: 2023: 2; 2024: 2.
4 papers.
- 2023Mitigate Replication and Copying in Diffusion Models with Generalized Caption and Dual Fusion Enhancementcited by 5
- 2023FakeSwarm: Improving Fake News Detection with Swarming Characteristicscited by 1
- 2024Linearizing Models for Efficient yet Robust Private Inferencecited by 0
- 2024CiMNet: Towards Joint Optimization for DNN Architecture and Configuration for Compute-In-Memory Hardwarecited by 0
- 2024CiMNet: Towards Joint Optimization for DNN Architecture and Configuration for Compute-In-Memory Hardware19 Feb 2024
- 2024Linearizing Models for Efficient yet Robust Private Inference8 Feb 2024
- 2023Mitigate Replication and Copying in Diffusion Models with Generalized Caption and Dual Fusion Enhancement13 Sep 2023
- 2023FakeSwarm: Improving Fake News Detection with Swarming Characteristics30 May 2023