"Yes, My LoRD." Guiding Language Model Extraction with Locality Reinforced Distillation
Not on the map. Why?
Abstract (arXiv)
Model extraction attacks (MEAs) on large language models (LLMs) have received increasing attention in recent research. However, existing attack methods typically adapt the extraction strategies originally developed for deep neural networks (DNNs). They neglect the underlying inconsistency between the training tasks of MEA and LLM alignment, leading to suboptimal attack performance. To tackle this issue, we propose Locality Reinforced Distillation (LoRD), a novel model extraction algorithm specifically designed for LLMs. In particular, LoRD employs a newly defined policy-gradient-style training task that utilizes the responses of victim model as the signal to guide the crafting of preference for the local model. Theoretical analyses demonstrate that I) The convergence procedure of LoRD in model extraction is consistent with the alignment procedure of LLMs, and II) LoRD can reduce query complexity while mitigating watermark protection through our exploration-based stealing. Extensive experiments validate the superiority of our method in extracting various state-of-the-art commercial LLMs. Our code is available at: https://github.com/liangzid/LoRD-MEA .
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 41 of its 72 references that are in the map. First posted on arXiv 4 Sep 2024.
Builds on
41 of 72 references.
References · 41
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2022Constitutional AI: Harmlessness from AI Feedbackcited by 2,950
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,919
29 more
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2015Trust Region Policy Optimizationcited by 2,503
- 2018Spider: A Large-Scale Human-Labeled Dataset for Complex and Cross-Domain Semantic Parsing and Text-to-SQL Taskcited by 1,120
- 2015Teaching Machines to Read and Comprehendcited by 1,039
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
- 2016Practical Black-Box Attacks against Machine Learningcited by 893
- 2022Discovering Language Model Behaviors with Model-Written Evaluationscited by 805
- 2019Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learningcited by 750
- 2016Stealing Machine Learning Models via Prediction APIscited by 602
- 2019SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarizationcited by 530
- 2022Improving alignment of dialogue agents via targeted human judgementscited by 527
- 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 386
- 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 252
- 2019High Accuracy and High Fidelity Extraction of Neural Networkscited by 208
- 2020Entangled Watermarks as a Defense against Model Extractioncited by 144
- 2019Thieves on Sesame Street! Model Extraction of BERT-based APIscited by 136
- 2023Aligning Language Models with Preferences through f-divergence Minimizationcited by 118
- 2019Evaluating the State-of-the-Art of End-to-End Natural Language Generation: The E2E NLG Challengecited by 86
- 2022On Reinforcement Learning and Distribution Matching for Fine-Tuning Language Models with no Catastrophic Forgettingcited by 80
- 2020Imitation Attacks and Defenses for Black-box Machine Translation Systemscited by 78
- 2021Protecting Intellectual Property of Language Generation APIs with Lexical Watermarkcited by 78
- 2021On the Safety of Conversational Models: Taxonomy, Dataset, and Benchmarkcited by 73
- 2022CATER: Intellectual Property Protection on Text Generation APIs via Conditional Watermarkscited by 68
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Networkcited by 47
- 2022SSLGuard: A Watermarking Scheme for Self-supervised Learning Pre-trained Encoderscited by 43
- 2022Distillation-Resistant Watermarking for Model Protection in NLPcited by 29
- 2021Student Surpasses Teacher: Imitation Attack for Black-Box NLP APIscited by 16
- 2024MEA-Defender: A Robust Watermark against Model Extraction Attackcited by 16
- 2023On Extracting Specialized Code Abilities from Large Language Models: A Feasibility Studycited by 13
Built on by
Citing papers per year: 2025: 3; 2026: 2.
5 papers.
- 2025Safety at Scale: A Comprehensive Survey of Large Model and Agent Safetycited by 63
- 2025How Much Do Large Language Model Cheat on Evaluation? Benchmarking Overestimation under the One-Time-Pad-Based Frameworkcited by 6
- 2025SecureInfer: Heterogeneous TEE-GPU Architecture for Privacy-Critical Tensors for Large Language Model Deploymentcited by 2
- 2026JudgeStealer: Extracting LLM Judging Capabilities across Evaluation Protocolscited by 0
- 2026AgentLeak: Cloning Stronger LLM Agent Capabilities onto Weaker Agents Beyond Skill Stealingcited by 0
- 2026AgentLeak: Cloning Stronger LLM Agent Capabilities onto Weaker Agents Beyond Skill Stealing7 Sep 2026
- 2026JudgeStealer: Extracting LLM Judging Capabilities across Evaluation Protocols27 Aug 2026
- 2025SecureInfer: Heterogeneous TEE-GPU Architecture for Privacy-Critical Tensors for Large Language Model Deployment22 Oct 2025
- 2025How Much Do Large Language Model Cheat on Evaluation? Benchmarking Overestimation under the One-Time-Pad-Based Framework25 Jul 2025
- 2025Safety at Scale: A Comprehensive Survey of Large Model and Agent Safety2 Feb 2025
Similar, not cited
Close in wording, no citation either way
- 2025AlignTree: Efficient Defense Against LLM Jailbreak Attacks
- 2025DELMAN: Dynamic Defense Against Large Language Model Jailbreaking with Model Editing
- 2025Targeting Alignment: Extracting Safety Classifiers of Aligned LLMs
- 2023Defending Against Alignment-Breaking Attacks via Robustly Aligned LLM
- 2023Survey of Vulnerabilities in Large Language Models Revealed by Adversarial Attacks
- 2024Jailbreaking Large Language Models Through Alignment Vulnerabilities in Out-of-Distribution Settings
- 2023Large Language Model Alignment: A Survey
- 2026Alignment Defends LLMs from Property Inference Attacks