Stealing Part of a Production Language Model
Authors: name: Nicholas Carlini; name: Daniel Paleka; name: Krishnamurthy Dj Dvijotham; name: Thomas Steinke; name: Jonathan Hayase; name: A. Feder Cooper; name: Katherine Lee; name: Matthew Jagielski; name: Milad Nasr; name: Arthur Conmy; name: Eric Wallace; name: David Rolnick; name: Florian Tramèr
Published: 2024-03-11 11:46:12

Abstract: We introduce the first model-stealing attack that extracts precise,
nontrivial information from black-box production language models like OpenAI's
ChatGPT or Google's PaLM-2. Specifically, our attack recovers the embedding
projection layer (up to symmetries) of a transformer model, given typical API
access. For under \20 USD, our attack extracts the entire projection matrix of
OpenAI's Ada and Babbage language models. We thereby confirm, for the first
time, that these black-box models have a hidden dimension of 1024 and 2048,
respectively. We also recover the exact hidden dimension size of the
gpt-3.5-turbo model, and estimate it would cost under 2,000 in queries to
recover the entire projection matrix. We conclude with potential defenses and
mitigations, and discuss the implications of possible future work that could
extend our attack.