LGCRAug 5, 2024

Pre-trained Encoder Inference: Revealing Upstream Encoders In Downstream Machine Learning Services

arXiv:2408.02814v21 citationsh-index: 5Has Code
Originality Highly original
AI Analysis

This addresses a security vulnerability for ML service providers using pre-trained encoders, revealing a new threat that is more realistic than existing attacks.

The paper introduces the Pre-trained Encoder Inference (PEI) attack, which extracts sensitive encoder information from downstream ML services to facilitate other attacks, and empirically verifies its effectiveness on vision encoders for tasks like image classification and multimodal generation.

Pre-trained encoders available online have been widely adopted to build downstream machine learning (ML) services, but various attacks against these encoders also post security and privacy threats toward such a downstream ML service paradigm. We unveil a new vulnerability: the Pre-trained Encoder Inference (PEI) attack, which can extract sensitive encoder information from a targeted downstream ML service that can then be used to promote other ML attacks against the targeted service. By only providing API accesses to a targeted downstream service and a set of candidate encoders, the PEI attack can successfully infer which encoder is secretly used by the targeted service based on candidate ones. Compared with existing encoder attacks, which mainly target encoders on the upstream side, the PEI attack can compromise encoders even after they have been deployed and hidden in downstream ML services, which makes it a more realistic threat. We empirically verify the effectiveness of the PEI attack on vision encoders. we first conduct PEI attacks against two downstream services (i.e., image classification and multimodal generation), and then show how PEI attacks can facilitate other ML attacks (i.e., model stealing attacks vs. image classification models and adversarial attacks vs. multimodal generative models). Our results call for new security and privacy considerations when deploying encoders in downstream services. The code is available at https://github.com/fshp971/encoder-inference.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes