English

Merging Improves Self-Critique Against Jailbreak Attacks

Computation and Language 2024-07-16 v2 Artificial Intelligence

Abstract

The robustness of large language models (LLMs) against adversarial manipulations, such as jailbreak attacks, remains a significant challenge. In this work, we propose an approach that enhances the self-critique capability of the LLM and further fine-tunes it over sanitized synthetic data. This is done with the addition of an external critic model that can be merged with the original, thus bolstering self-critique capabilities and improving the robustness of the LLMs response to adversarial prompts. Our results demonstrate that the combination of merging and self-critique can reduce the attack success rate of adversaries significantly, thus offering a promising defense mechanism against jailbreak attacks. Code, data and models released at https://github.com/vicgalle/merging-self-critique-jailbreaks .

Keywords

Cite

@article{arxiv.2406.07188,
  title  = {Merging Improves Self-Critique Against Jailbreak Attacks},
  author = {Victor Gallego},
  journal= {arXiv preprint arXiv:2406.07188},
  year   = {2024}
}

Comments

Published at ICML 2024 Workshop on Foundation Models in the Wild

R2 v1 2026-06-28T17:01:20.568Z