CoLA: A Choice Leakage Attack Framework to Expose Privacy Risks in Subset Training
Organizations: King Abdullah University of Science and Technology · Provable Responsible AI and Data Analytics Lab · National University of Singapore · Johns Hopkins University
Abstract
Training models on a carefully chosen portion of data rather than the full dataset is now a standard preprocess for modern ML. From vision coreset selection to large-scale filtering in language models, it enables scalability with minimal utility loss. A common intuition is that training on fewer samples should also reduce privacy risks. In this paper, we challenge this assumption. We show that subset training is not privacy free: the very choices of which data are included or excluded can introduce new privacy surface and leak more sensitive information. Such information can be captured by adversaries either through side-channel metadata from the subset selection process or via the outputs of the target model. To systematically study this phenomenon, we propose CoLA (Choice Leakage Attack), a unified framework for analyzing privacy leakage in subset selection. In CoLA, depending on the adversary's knowledge of the side-channel information, we define two practical attack scenarios: Subset-aware Side-channel Attacks and Black-box Attacks. Under both scenarios, we investigate two privacy surfaces unique to subset training: (1) Training-membership MIA (TM-MIA), which concerns only the privacy of training data membership, and (2) Selection-participation MIA (SP-MIA), which concerns the privacy of all samples that participated in the subset selection process. Notably, SP-MIA enlarges the notion of membership from model training to the entire data-model supply chain. Experiments on vision and language models show that existing threat models underestimate subset-training privacy risks: the expanded privacy surface leaks both training and selection membership, extending risks from individual models to the broader ML ecosystem.
Figures & tables
| Intensity | Setting | NN | NN_top3 | NN_cls | LiRA | CoLA | |||||
| AUC | TPR@5%FPR | AUC | TPR@5%FPR | AUC | TPR@5%FPR | AUC | TPR@5%FPR | AUC | TPR@5%FPR | ||
| Light | SP-MIA | 51.23 2.56 | 5.83 1.34 | 51.77 3.02 | 3.34 4.45 | 51.59 2.66 | 6.37 2.22 | 51.26 4.85 | 6.43 3.70 | 61.39 2.48 | 14.24 2.02 |
| TM-MIA | 64.00 12.15 | 12.13 5.96 | 61.57 15.05 | 8.93 13.52 | 67.24 16.18 | 19.30 17.14 | 69.86 22.08 | 15.74 18.19 | 83.77 2.44 | 42.19 4.51 | |
| Medium | SP-MIA | 52.33 3.56 | 6.31 1.48 | 53.59 4.30 | 3.56 2.28 | 54.51 4.51 | 6.64 1.58 | 54.99 4.69 | 5.31 0.42 | 81.93 3.50 | 42.66 5.81 |
| TM-MIA | 59.84 12.84 | 10.80 4.97 | 60.37 10.53 | 2.91 3.32 | 66.84 11.79 | 12.51 5.85 | 62.96 13.69 | 4.61 2.52 | 88.53 2.55 | 60.10 7.62 | |
| Heavy | SP-MIA | 52.21 3.83 | 12.20 15.48 | 53.20 4.85 | 2.80 2.43 | 53.53 5.94 | 12.37 15.44 | 53.69 5.68 | 4.26 1.77 | 96.86 2.60 | 88.60 5.51 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Privacy-Preserving ML
Privacy-preserving machine learning develops methods to learn, share, and run models while limiting exposure of sensitive data.
See all 433 papers