Provable Defense Framework for LLM Jailbreaks via Noise-Augumented Alignment

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cheng, Zehua, Yang, Jianwei, Dai, Wei, Sun, Jiahao
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918318617657344
author Cheng, Zehua
Yang, Jianwei
Dai, Wei
Sun, Jiahao
author_facet Cheng, Zehua
Yang, Jianwei
Dai, Wei
Sun, Jiahao
contents Large Language Models (LLMs) remain vulnerable to adaptive jailbreaks that easily bypass empirical defenses like GCG. We propose a framework for certifiable robustness that shifts safety guarantees from single-pass inference to the statistical stability of an ensemble. We introduce Certified Semantic Smoothing (CSS) via Stratified Randomized Ablation, a technique that partitions inputs into immutable structural prompts and mutable payloads to derive rigorous lo norm guarantees using the Hypergeometric distribution. To resolve performance degradation on sparse contexts, we employ Noise-Augmented Alignment Tuning (NAAT), which transforms the base model into a semantic denoiser. Extensive experiments on Llama-3 show that our method reduces the Attack Success Rate of gradient-based attacks from 84.2% to 1.2% while maintaining 94.1% benign utility, significantly outperforming character-level baselines which degrade utility to 74.3%. This framework provides a deterministic certificate of safety, ensuring that a model remains robust against all adversarial variants within a provable radius.
format Preprint
id arxiv_https___arxiv_org_abs_2602_01587
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Provable Defense Framework for LLM Jailbreaks via Noise-Augumented Alignment
Cheng, Zehua
Yang, Jianwei
Dai, Wei
Sun, Jiahao
Computation and Language
Artificial Intelligence
Large Language Models (LLMs) remain vulnerable to adaptive jailbreaks that easily bypass empirical defenses like GCG. We propose a framework for certifiable robustness that shifts safety guarantees from single-pass inference to the statistical stability of an ensemble. We introduce Certified Semantic Smoothing (CSS) via Stratified Randomized Ablation, a technique that partitions inputs into immutable structural prompts and mutable payloads to derive rigorous lo norm guarantees using the Hypergeometric distribution. To resolve performance degradation on sparse contexts, we employ Noise-Augmented Alignment Tuning (NAAT), which transforms the base model into a semantic denoiser. Extensive experiments on Llama-3 show that our method reduces the Attack Success Rate of gradient-based attacks from 84.2% to 1.2% while maintaining 94.1% benign utility, significantly outperforming character-level baselines which degrade utility to 74.3%. This framework provides a deterministic certificate of safety, ensuring that a model remains robust against all adversarial variants within a provable radius.
title Provable Defense Framework for LLM Jailbreaks via Noise-Augumented Alignment
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2602.01587