Input Selection for Bandwidth-Limited Neural Network Inference

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review


  • Fulltext

    Accepted author manuscript, 2.74 MB, PDF document

Data are often accommodated on centralized storage servers. This is the case, for instance, in remote sensing and astronomy, where projects produce several petabytes of data every year. While machine learning models are often trained on relatively small subsets of the data, the inference phase typically requires transferring significant amounts of data between the servers and the clients. In many cases, the bandwidth available per user is limited, which then renders the data transfer to be one of the major bottlenecks. In this work, we propose a framework that automatically selects the relevant parts of the input data for a given neural network. The model as well as the associated selection masks are trained simultaneously such that a good model performance is achieved while only a minimal amount of data is selected. During the inference phase, only those parts of the data have to be transferred between the server and the client. We propose both instance-independent and instance-dependent selection masks. The former ones are the same for all instances to be transferred, whereas the latter ones allow for variable transfer sizes per instance. Our experiments show that it is often possible to significantly reduce the amount of data needed to be transferred without affecting the model quality much.

Original languageEnglish
Title of host publicationProceedings of the 2022 SIAM International Conference on Data Mining, SDM 2022
Number of pages9
Publication date2022
ISBN (Electronic)9781611977172
Publication statusPublished - 2022
Event2022 SIAM International Conference on Data Mining, SDM 2022 - Virtual, Online
Duration: 28 Apr 202230 Apr 2022


Conference2022 SIAM International Conference on Data Mining, SDM 2022
ByVirtual, Online

Bibliographical note

Publisher Copyright:
Copyright © 2022 by SIAM.

ID: 314303760