Sysformer: Safeguarding Frozen Large Language Models with Adaptive System Prompts

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sharma, Kartik, Jin, Yiqiao, Rakesh, Vineeth, Dou, Yingtong, Pan, Menghai, Das, Mahashweta, Kumar, Srijan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911491101294592
author Sharma, Kartik
Jin, Yiqiao
Rakesh, Vineeth
Dou, Yingtong
Pan, Menghai
Das, Mahashweta
Kumar, Srijan
author_facet Sharma, Kartik
Jin, Yiqiao
Rakesh, Vineeth
Dou, Yingtong
Pan, Menghai
Das, Mahashweta
Kumar, Srijan
contents As large language models (LLMs) are deployed in safety-critical settings, it is essential to ensure that their responses comply with safety standards. Prior research has revealed that LLMs often fail to grasp the notion of safe behaviors, resulting in either unjustified refusals to harmless prompts or the generation of harmful content. While substantial efforts have been made to improve their robustness, existing defenses often rely on costly fine-tuning of model parameters or employ suboptimal heuristic techniques. In this work, we take a novel approach to safeguard LLMs by learning to adapt the system prompts in instruction-tuned LLMs. While LLMs are typically pre-trained to follow a fixed system prompt, we investigate the impact of tailoring the system prompt to each specific user input on the safety of the responses. To this end, we propose $\textbf{Sysformer}$, a trans$\textbf{former}$ model that updates an initial $\textbf{sys}$tem prompt to a more robust system prompt in the LLM input embedding space while attending to the user prompt. While keeping the LLM parameters frozen, the Sysformer is trained to refuse to respond to a set of harmful prompts while responding ideally to a set of safe ones. Through extensive experiments on $5$ LLMs from different families and $2$ recent benchmarks, we demonstrate that Sysformer can significantly enhance the robustness of LLMs, leading to upto $80\%$ gain in the refusal rate on harmful prompts while enhancing the compliance with the safe prompts by upto $90\%$. Results also generalize well to sophisticated jailbreaking attacks, making LLMs upto $100\%$ more robust against different attack strategies. We hope our findings lead to cheaper safeguarding of LLMs and motivate future investigations into designing variable system prompts.
format Preprint
id arxiv_https___arxiv_org_abs_2506_15751
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Sysformer: Safeguarding Frozen Large Language Models with Adaptive System Prompts
Sharma, Kartik
Jin, Yiqiao
Rakesh, Vineeth
Dou, Yingtong
Pan, Menghai
Das, Mahashweta
Kumar, Srijan
Artificial Intelligence
Computation and Language
Machine Learning
As large language models (LLMs) are deployed in safety-critical settings, it is essential to ensure that their responses comply with safety standards. Prior research has revealed that LLMs often fail to grasp the notion of safe behaviors, resulting in either unjustified refusals to harmless prompts or the generation of harmful content. While substantial efforts have been made to improve their robustness, existing defenses often rely on costly fine-tuning of model parameters or employ suboptimal heuristic techniques. In this work, we take a novel approach to safeguard LLMs by learning to adapt the system prompts in instruction-tuned LLMs. While LLMs are typically pre-trained to follow a fixed system prompt, we investigate the impact of tailoring the system prompt to each specific user input on the safety of the responses. To this end, we propose $\textbf{Sysformer}$, a trans$\textbf{former}$ model that updates an initial $\textbf{sys}$tem prompt to a more robust system prompt in the LLM input embedding space while attending to the user prompt. While keeping the LLM parameters frozen, the Sysformer is trained to refuse to respond to a set of harmful prompts while responding ideally to a set of safe ones. Through extensive experiments on $5$ LLMs from different families and $2$ recent benchmarks, we demonstrate that Sysformer can significantly enhance the robustness of LLMs, leading to upto $80\%$ gain in the refusal rate on harmful prompts while enhancing the compliance with the safe prompts by upto $90\%$. Results also generalize well to sophisticated jailbreaking attacks, making LLMs upto $100\%$ more robust against different attack strategies. We hope our findings lead to cheaper safeguarding of LLMs and motivate future investigations into designing variable system prompts.
title Sysformer: Safeguarding Frozen Large Language Models with Adaptive System Prompts
topic Artificial Intelligence
Computation and Language
Machine Learning
url https://arxiv.org/abs/2506.15751