ArXiv TLDR

Crafting Reversible SFT Behaviors in Large Language Models

🐦 Tweet
2605.06632

Yuping Lin, Pengfei He, Yue Xing, Yingqian Cui, Jiayuan Ding + 3 more

cs.LG

TLDR

This paper introduces LCDD to create sparse, controllable "carriers" for SFT behaviors in LLMs, enabling their selective reversal with SFT-Eraser.

Key contributions

  • Proposes Loss-Constrained Dual Descent (LCDD) to create sparse, causally necessary "carriers" for SFT behaviors.
  • Introduces SFT-Eraser, a soft prompt, to selectively reverse SFT behaviors by matching carrier activations.
  • Shows LCDD-generated carriers preserve behaviors and enable strong, selective reversal across LLM families.
  • Ablations confirm sparse carrier structure is key for reversal, not just trigger optimization.

Why it matters

This paper offers a novel method to mechanistically localize and control SFT-induced behaviors in LLMs. By compressing behaviors into sparse "carriers," it enables selective suppression of unwanted traits without weight modification. This significantly advances LLM safety and controllability for deployment.

Original Abstract

Supervised fine-tuning (SFT) induces new behaviors in large language models, yet imposes no structural constraint on how these behaviors are distributed within the model. Existing behavior interpretation methods, such as circuit attribution approaches, identify sparse subnetworks correlated with SFT-induced behaviors post-hoc. However, such correlations do not imply *causal necessity*, limiting the ability to selectively control SFT-induced behaviors at inference time. We pursue an alternative by asking: can an SFT-induced behavior be deliberately compressed into a sparse, mechanistically necessary subnetwork, termed a *carrier*, while remaining controllable at inference time without weight modification? We propose (a) **Loss-Constrained Dual Descent (LCDD)**, which constructs such carriers by jointly optimizing routing masks and model weights under an explicit utility budget, and (b) **SFT-Eraser**, a soft prompt optimized via activation matching on extracted carrier channels, to reverse the SFT-induced behavior. Across safety, fixed-response, and style behaviors on multiple model families, LCDD yields sparse carriers that preserve target behaviors while enabling strong reversion when triggered by SFT-Eraser. Ablations further establish that the sparse structure is the key precondition for reversal: the same trigger optimization fails on standard SFT models, confirming that structure rather than trigger design is the operative factor. These results provide direct evidence that the learned carriers are causally necessary for the behaviors, pointing to a new direction for systematically localizing and selectively suppressing SFT-induced behaviors in deployed models.

📬 Weekly AI Paper Digest

Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.