# Securing Neural Networks with Knapsack Optimization

Source: [Securing Neural Networks with Knapsack Optimization](<https://www.sota2.com/research/paper/68fa71a41e209131cea523ec>)

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

Data updated: 2025-10-26T02:09:33.631Z

Authors: Yakir Gorski, Amir Jevnisek, Shai Avidan

Published: 2023-04-20T16:40:10.000Z

## Abstract

MLaaS Service Providers (SPs) holding a Neural Network would like to keep the Neural Network weights secret. On the other hand, users wish to utilize the SPs' Neural Network for inference without revealing their data. Multi-Party Computation (MPC) offers a solution to achieve this. Computations in MPC involve communication, as the parties send data back and forth. Non-linear operations are usually the main bottleneck requiring the bulk of communication bandwidth. In this paper, we focus on ResNets, which serve as the backbone for many Computer Vision tasks, and we aim to reduce their non-linear components, specifically, the number of ReLUs. Our key insight is that spatially close pixels exhibit correlated ReLU responses. Building on this insight, we replace the per-pixel ReLU operation with a ReLU operation per patch. We term this approach 'Block-ReLU'. Since different layers in a Neural Network correspond to different feature hierarchies, it makes sense to allow patch-size flexibility for the various layers of the Neural Network. We devise an algorithm to choose the optimal set of patch sizes through a novel reduction of the problem to the Knapsack Problem. We demonstrate our approach in the semi-honest secure 3-party setting for four problems: Classifying ImageNet using ResNet50 backbone, classifying CIFAR100 using ResNet18 backbone, Semantic Segmentation of ADE20K using MobileNetV2 backbone, and Semantic Segmentation of Pascal VOC 2012 using ResNet50 backbone. Our approach achieves competitive performance compared to a handful of competitors. Our source code is publicly available: https://github.com/yg320/secure\_inference.

## 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 CIFAR-100 (Acc. (%) and Acc./#ReLU)](<https://www.sota2.com/research/sota/image-classification-on-cifar-100-acc-percent-and-acc-sharp-relu.md>): Image Classification; CIFAR-100; Accuracy (%): 75.51.

- [Segmentation on ADE20K](<https://www.sota2.com/research/sota/segmentation-on-ade20k.md>): Segmentation; ADE20K; mIoU: 33.23.

## Original sources and implementation

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

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