Classifier-free guidance in LLMs Safety

Code

• github.com/rgsmirnov/cfg_safety_llm
  OfficialIn paperpytorch★ 1

Abstract

The paper describes LLM unlearning without a retaining dataset, using the ORPO reinforcement learning method with inference enhanced by modified classifier-free guidance. Significant improvement in unlearning, without degradation of the model, is achieved through direct training on synthetic replacement data in CFG-aware training regime, with classifier-free guidance applied during the inference. This article is an extended version of the NeurIPS 2024 LLM-PC submission, which was awarded second prize.

Tasks

Reproductions