← Search

Ilia Shumailov

19 accepted papers

2026

Extracting alignment data in open models

ICML 2026poster

In this work, we show that it is possible to extract significant amounts of alignment training data from a post-trained model -- useful to steer the model to improve certain capabilities such as long-context reasoning, safety, instruction following, and maths. While the majority of related work on m…

Cited by 0SourceScholar
2025

Breach By A Thousand Leaks: Unsafe Information Leakage in 'Safe' AI Responses

ICLR 2025poster

Vulnerability of Frontier language models to misuse has prompted the development of safety measures like filters and alignment training seeking to ensure safety through robustness to adversarially crafted prompts. We assert that robustness is fundamentally insufficient for ensuring safety goals due…

Cited by 2SourcePDFScholar
2025

Exploring the limits of strong membership inference attacks on large language models

NeurIPS 2025poster

State-of-the-art membership inference attacks (MIAs) typically require training many reference models, making it difficult to scale these attacks to large pre-trained language models (LLMs). As a result, prior research has either relied on weaker attacks that avoid training references (e.g., fine-tu…

Cited by 0SourceScholar
2025

Hardware and Software Platform Inference

ICML 2025poster

It is now a common business practice to buy access to large language model (LLM) inference rather than self-host, because of significant upfront hardware infrastructure and energy costs. However, as a buyer, there is no mechanism to verify the authenticity of the advertised service including the ser…

Cited by 0SourcePDFScholar
2025

Interpreting the Repeated Token Phenomenon in Large Language Models

ICML 2025poster

Large Language Models (LLMs), despite their impressive capabilities, often fail to accurately repeat a single word when prompted to, and instead output unrelated text. This unexplained failure mode represents a *vulnerability*, allowing even end users to diverge models away from their intended behav…

2025

Machine Unlearning Doesn't Do What You Think: Lessons for Generative AI Policy and Research

NeurIPS 2025oral

"Machine unlearning" is a popular proposed solution for mitigating the existence of content in an AI model that is problematic for legal or moral reasons, including privacy, copyright, safety, and more. For example, unlearning is often invoked as a solution for removing the effects of specific infor…

Cited by 0SourceScholar
2025

Measuring memorization in language models via probabilistic extraction

NAACL 2025long

Large language models (LLMs) are susceptible to memorizing training data, raising concerns about the potential extraction of sensitive information at generation time. Discoverable extraction is the most common method for measuring this issue: split a training example into a prefix and suffix, then p…

Cited by 2SourcePDFScholar
2025

Position: Machine Learning Models Have a Supply Chain Problem

ICML 2025poster

Powerful machine learning (ML) models are now readily available online, which creates exciting possibilities for users who lack the deep technical expertise or substantial computing resources needed to develop them. On the other hand, this type of open ecosystem comes with many risks. In this paper,…

Cited by 0SourcePDFScholar
2024

Beyond Slow Signs in High-fidelity Model Extraction

NeurIPS 2024poster

Deep neural networks, costly to train and rich in intellectual property value, are increasingly threatened by model extraction attacks that compromise their confiden- tiality. Previous attacks have succeeded in reverse-engineering model parameters up to a precision of float64 for models trained on r…

2024

Position: Fundamental Limitations of LLM Censorship Necessitate New Approaches

ICML 2024poster

Large language models (LLMs) have exhibited impressive capabilities in comprehending complex instructions. However, their blind adherence to provided instructions has led to concerns regarding risks of malicious use. Existing defence mechanisms, such as model fine-tuning or output censorship methods…

Cited by 2SourcePDFScholar
2023

Architectural Backdoors in Neural Networks

CVPR 2023poster

Machine learning is vulnerable to adversarial manipulation. Previous literature has demonstrated that at the training stage attackers can manipulate data (Gu et al.) and data sampling procedures (Shumailov et al.) to control model behaviour. A common attack goal is to plant backdoors i.e. force the…

Cited by 23SourcePDFScholar
2023

Revisiting Automated Prompting: Are We Actually Doing Better?

ACL 2023short

Current literature demonstrates that Large Language Models (LLMs) are great few-shot learners, and prompting significantly increases their performance on a range of downstream tasks in a few-shot learning setting. An attempt to automate human-led prompting followed, with some progress achieved. In p…

2023

Revisiting Block-based Quantisation: What is Important for Sub-8-bit LLM Inference?

EMNLP 2023long main

The inference of Large language models (LLMs) requires immense computation and memory resources. To curtail these costs, quantisation has emerged as a promising solution, but existing LLM quantisation mainly focuses on 8-bit. In this work, we explore the statistical and learning properties of the LL…

Cited by 0SourcecodeScholar
2022

In Differential Privacy, There is Truth: on Vote-Histogram Leakage in Ensemble Private Learning

NeurIPS 2022accept

When learning from sensitive data, care must be taken to ensure that training algorithms address privacy concerns. The canonical Private Aggregation of Teacher Ensembles, or PATE, computes output labels by aggregating the predictions of a (possibly distributed) collection of teacher models via a vot…

Cited by 6SourcePDFScholar
2022

On the Limitations of Stochastic Pre-processing Defenses

NeurIPS 2022accept

Defending against adversarial examples remains an open problem. A common belief is that randomness at inference increases the cost of finding adversarial inputs. An example of such a defense is to apply a random transformation to inputs prior to feeding them to the model. In this paper, we empirical…

2022

Rapid Model Architecture Adaption for Meta-Learning

NeurIPS 2022accept

Network Architecture Search (NAS) methods have recently gathered much attention. They design networks with better performance and use a much shorter search time compared to traditional manual tuning. Despite their efficiency in model deployments, most NAS algorithms target a single task on a fixed h…

Cited by 6SourcePDFScholar
2022

Rethinking Image-Scaling Attacks: The Interplay Between Vulnerabilities in Machine Learning Systems

ICML 2022oral

As real-world images come in varying sizes, the machine learning model is part of a larger system that includes an upstream image scaling algorithm. In this paper, we investigate the interplay between vulnerabilities of the image scaling procedure and machine learning models in the decision-based bl…

2021

Markpainting: Adversarial Machine Learning meets Inpainting

ICML 2021spotlight

Inpainting is a learned interpolation technique that is based on generative modeling and used to populate masked or missing pieces in an image; it has wide applications in picture editing and retouching. Recently, inpainting started being used for watermark removal, raising concerns. In this paper w…