← Search

Reza Shokri

19 accepted papers

2026

(Token-Level) \textbf{InfoRMIA}: Stronger Membership Inference and Privacy Assessment for LLMs

ICLR 2026poster

Machine learning models are known to leak sensitive information, as they inevitably memorize (parts of) their training data. More alarmingly, large language models (LLMs) are now trained on nearly all available data, which amplifies the magnitude of information leakage and raises serious privacy ris…

Cited by 0SourcecodeScholar
2026

Decomposing Extrapolative Problem Solving: Spatial Transfer and Length Scaling with Map Worlds

ICLR 2026poster

Someone who learns to walk shortest paths in New York can, upon receiving a map of Paris, immediately apply the same rule to navigate, despite never practicing there. This ability to recombine known rules to solve novel problems exemplifies compositional generalization (CG), a hallmark of human cogn…

Cited by 0SourcecodeScholar
2025

Context-Aware Membership Inference Attacks against Pre-trained Large Language Models

EMNLP 2025

Membership Inference Attacks (MIAs) on pre-trained Large Language Models (LLMs) aim at determining if a data point was part of the model’s training set. Prior MIAs that are built for classification models fail at LLMs, due to ignoring the generative nature of LLMs across token sequences. In this pap

Cited by 0SourcePDFScholar
2025

Contextual Integrity in LLMs via Reasoning and Reinforcement Learning

NeurIPS 2025poster

As the era of autonomous agents making decisions on behalf of users unfolds, ensuring contextual integrity (CI) -- what is the appropriate information to share while carrying out a certain task -- becomes a central question to the field. We posit that CI demands a form of reasoning where the agent…

Cited by 0SourceScholar
2025

How much of my dataset did you use? Quantitative Data Usage Inference in Machine Learning

ICLR 2025oral

How much of my data was used to train a machine learning model? This is a critical question for data owners assessing the risk of unauthorized usage of their data to train models. However, previous work mistakenly treats this as a binary problem—inferring whether all-or-none or any-or-none of the da…

Cited by 0SourcePDFScholar
2025

Minerva: A Programmable Memory Test Benchmark for Language Models

ICML 2025poster

How effectively can LLM-based AI assistants utilize their memory (context) to perform various tasks? Traditional data benchmarks, which are often manually crafted, suffer from several limitations: they are static, susceptible to overfitting, difficult to interpret, and lack actionable insights--fail…

Cited by 0SourcePDFScholar
2025

The Canary’s Echo: Auditing Privacy Risks of LLM-Generated Synthetic Text

ICML 2025poster

How much information about training samples can be leaked through synthetic data generated by Large Language Models (LLMs)? Overlooking the subtleties of information flow in synthetic data generation pipelines can lead to a false sense of privacy. In this paper, we assume an adversary has access to…

Cited by 0SourcePDFScholar
2024

Can LLMs Keep a Secret? Testing Privacy Implications of Language Models via Contextual Integrity Theory

ICLR 2024spotlight

Existing efforts on quantifying privacy implications for large language models (LLMs) solely focus on measuring leakage of training data. In this work, we shed light on the often-overlooked interactive settings where an LLM receives information from multiple sources and generates an output to be sha…

Cited by 86SourcePDFScholar
2024

Leave-one-out Distinguishability in Machine Learning

ICLR 2024poster

We introduce an analytical framework to quantify the changes in a machine learning algorithm's output distribution following the inclusion of a few data points in its training set, a notion we define as leave-one-out distinguishability (LOOD). This is key to measuring data **memorization** and info…

2023

Initialization Matters: Privacy-Utility Analysis of Overparameterized Neural Networks

NeurIPS 2023poster

We analytically investigate how over-parameterization of models in randomized machine learning algorithms impacts the information leakage about their training data. Specifically, we prove a privacy bound for the KL divergence between model distributions on worst-case neighboring datasets, and explor…

Cited by 11SourcePDFScholar
2023

Share Your Representation Only: Guaranteed Improvement of the Privacy-Utility Tradeoff in Federated Learning

ICLR 2023poster

Repeated parameter sharing in federated learning causes significant information leakage about private data, thus defeating its main purpose: data privacy. Mitigating the risk of this information leakage, using state of the art differentially private algorithms, also does not come for free. Randomi…

2023

Unified Enhancement of Privacy Bounds for Mixture Mechanisms via $f$-Differential Privacy

NeurIPS 2023poster

Differentially private (DP) machine learning algorithms incur many sources of randomness, such as random initialization, random batch subsampling, and shuffling. However, such randomness is difficult to take into account when proving differential privacy bounds because it induces mixture distributio…

Cited by 7SourcePDFScholar
2022

Quantifying Privacy Risks of Masked Language Models Using Membership Inference Attacks

EMNLP 2022main

The wide adoption and application of Masked language models (MLMs) on sensitive data (from legal to medical) necessitates a thorough quantitative investigation into their privacy vulnerabilities. Prior attempts at measuring leakage of MLMs via membership inference attacks have been inconclusive, imp…

Cited by 177SourcePDFScholar
2021

Differential Privacy Dynamics of Langevin Diffusion and Noisy Gradient Descent

NeurIPS 2021spotlight

What is the information leakage of an iterative randomized learning algorithm about its training data, when the internal state of the algorithm is \emph{private}? How much is the contribution of each specific training epoch to the information leakage through the released model? We study this problem…

Cited by 81SourcePDFScholar
2021

Quantifying the Privacy Risks of Learning High-Dimensional Graphical Models

AISTATS 2021poster

Models leak information about their training data. This enables attackers to infer sensitive information about their training sets, notably determine if a data sample was part of the model’s training set. The existing works empirically show the possibility of these membership inference (tracing) att…

Cited by 30SourcePDFScholar