首页 > AI前沿 > Bayesian Low-Rank Posteriors for Scalable Membership Inference

Bayesian Low-Rank Posteriors for Scalable Membership Inference

arXiv机器学习 2026-11-07 12:08 6 阅读 查看原文

背景

Membership inference attacks (MIAs) aim to determine whether a sample was used during the training of a target model. Strong MIAs rely on reference (or shadow) models to estimate target-conditional score distributions, but training such ensembles is computationally expensive for modern large-scale Vision--Language Models (VLMs).

方法

方法原理

In this work, we propose a scalable alternative that replaces explicit reference-model training with a Bayesian approximation over low-rank adaptation parameters.

Specifically, we model uncertainty only within the LoRA subspace and construct a posterior from a single training run, from which we sample a diverse set of virtual reference models.

模型实例化

We instantiate this approach using Stochastic Weight Averaging Gaussian (SWAG), enabling efficient approximation of target-conditional score distributions at a fraction of the cost of conventional shadow-model ensembles.

评估与实验

实验设计与验证

We evaluate the resulting attack on three downstream-adapted VLMs across two privacy-sensitive visual question answering tasks, namely Document VQA and Medical VQA.

To isolate the effect of pretraining knowledge, we additionally introduce a controlled synthetic dataset.

Our approach outperforms standard reference-based attacks while requiring only a single reference-model fine-tuning run, making membership inference more practical for large VLMs.