Caliber Defense Protects AI Models from Extraction Attacks.

Chi Wang, Hanwen Wang, Yu Xia, Zihan Wang, Guangdong Bai· August 4, 2026 View original

Key takeaways

  • Caliber defends AI models from extraction by adding noise to logits.
  • It quantifies the cost for attackers to recover original model information.
  • The defense balances signal degradation for attackers with model utility.
  • Caliber shows strong performance in making model extraction more difficult.

Who benefits

SoftwareCloud ServicesFinancial ServicesHealthcareDefense

Summary

Caliber is a new defense mechanism that protects AI models from extraction attacks by adding controlled Gaussian noise to internal logits, making it harder for attackers to replicate the model. It quantifies the trade-off between degrading the attacker's signal and the provable query cost needed to recover clean logits.

A new defense mechanism, named Caliber, has been introduced to protect AI models accessible via score-returning APIs from model extraction attacks. These attacks involve an adversary querying a target model to create a surrogate model that mimics its behavior. Caliber counters this by strategically adding independent and identically distributed Gaussian noise to the model's internal logits before returning scores. This perturbation aims to degrade the quality of the supervision signal an attacker can use for knowledge distillation. The core innovation of Caliber lies in its ability to formulate noise selection as a calibration problem, balancing the degradation of the surrogate's training signal with the provable per-input query cost required for an attacker to recover the original, clean logits. The research establishes two key properties: a monotone degradation of agreement with clean predictions as noise increases, and a closed-form minimax lower bound on the queries needed for recovery. By normalizing noise variance and fitting a logistic curve, Caliber effectively controls the noise-utility relationship, demonstrating strong performance across various models and datasets while making extraction significantly more expensive for attackers.

Why it matters

Protecting proprietary AI models from intellectual property theft and unauthorized replication is critical for businesses that invest heavily in AI development and deployment.

How to implement this in your domain

  1. 1Evaluate Caliber's applicability for existing AI models exposed via APIs.
  2. 2Integrate noise injection mechanisms into API endpoints to deter model extraction.
  3. 3Monitor the trade-off between defense strength and model utility for legitimate users.
  4. 4Develop internal guidelines for securing AI intellectual property.

Original post by Chi Wang, Hanwen Wang, Yu Xia, Zihan Wang, Guangdong Bai

"arXiv:2608.01023v1 Announce Type: new Abstract: We present Caliber, an output-perturbation defense against model extraction that formulates noise selection as a calibration problem: how much the defense degrades the supervision signal used to train a surrogate, and the provable p…"

View on X

Originally posted by Chi Wang, Hanwen Wang, Yu Xia, Zihan Wang, Guangdong Bai on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses