Score: 1

Bayesian Inference of Training Dataset Membership

Published: May 31, 2025 | arXiv ID: 2506.00701v1

By: Yongchao Huang

Potential Business Impact:

Finds if your private data was used in AI.

Business Areas:
Predictive Analytics Artificial Intelligence, Data and Analytics, Software

Determining whether a dataset was part of a machine learning model's training data pool can reveal privacy vulnerabilities, a challenge often addressed through membership inference attacks (MIAs). Traditional MIAs typically require access to model internals or rely on computationally intensive shadow models. This paper proposes an efficient, interpretable and principled Bayesian inference method for membership inference. By analyzing post-hoc metrics such as prediction error, confidence (entropy), perturbation magnitude, and dataset statistics from a trained ML model, our approach computes posterior probabilities of membership without requiring extensive model training. Experimental results on synthetic datasets demonstrate the method's effectiveness in distinguishing member from non-member datasets. Beyond membership inference, this method can also detect distribution shifts, offering a practical and interpretable alternative to existing approaches.

Country of Origin
🇬🇧 United Kingdom

Page Count
10 pages

Category
Computer Science:
Machine Learning (CS)