# Making Models Shallow Again: Jointly Learning to Reduce Non-Linearity and Depth for Latency-Efficient Private Inference

Source: [Making Models Shallow Again: Jointly Learning to Reduce Non-Linearity and Depth for Latency-Efficient Private Inference](<https://www.sota2.com/research/paper/68fa6f241e209131cea43fd2>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2025-10-26T01:14:37.532Z

Authors: Souvik Kundu, Yuke Zhang, Dake Chen, Peter A. Beerel

Published: 2023-04-26T04:23:34.000Z

## Abstract

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.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Image Classification on ImageNet (test) (Accuracy and Communication Overhead)](<https://www.sota2.com/research/sota/image-classification-on-imagenet-test-accuracy-and-communication-overhead.md>): Image Classification; ImageNet (test); Top-1 Accuracy: 69.1.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2304.13274>)

[PDF](<https://arxiv.org/pdf/2304.13274.pdf>)
