Just Enough Shifts: Mitigating Over-Refusal in Aligned Language Models with Targeted Representation Fine-Tuning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dabas, Mahavir, Chen, Si, Fleming, Charles, Jin, Ming, Jia, Ruoxi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912468528267264
author Dabas, Mahavir
Chen, Si
Fleming, Charles
Jin, Ming
Jia, Ruoxi
author_facet Dabas, Mahavir
Chen, Si
Fleming, Charles
Jin, Ming
Jia, Ruoxi
contents Safety alignment is crucial for large language models (LLMs) to resist malicious instructions but often results in over-refusals, where benign prompts are unnecessarily rejected, impairing user experience and model utility. We introduce ACTOR (Activation-Based Training for Over-Refusal Reduction), a robust and compute- and data-efficient training framework that minimizes over-refusals by leveraging internal activation patterns from diverse queries. ACTOR precisely identifies and adjusts the activation components that trigger refusals, providing stronger control over the refusal mechanism. By fine-tuning only a single model layer, ACTOR effectively reduces over-refusals across multiple benchmarks while maintaining the model's ability to handle harmful queries and preserve overall utility.
format Preprint
id arxiv_https___arxiv_org_abs_2507_04250
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Just Enough Shifts: Mitigating Over-Refusal in Aligned Language Models with Targeted Representation Fine-Tuning
Dabas, Mahavir
Chen, Si
Fleming, Charles
Jin, Ming
Jia, Ruoxi
Machine Learning
Artificial Intelligence
Safety alignment is crucial for large language models (LLMs) to resist malicious instructions but often results in over-refusals, where benign prompts are unnecessarily rejected, impairing user experience and model utility. We introduce ACTOR (Activation-Based Training for Over-Refusal Reduction), a robust and compute- and data-efficient training framework that minimizes over-refusals by leveraging internal activation patterns from diverse queries. ACTOR precisely identifies and adjusts the activation components that trigger refusals, providing stronger control over the refusal mechanism. By fine-tuning only a single model layer, ACTOR effectively reduces over-refusals across multiple benchmarks while maintaining the model's ability to handle harmful queries and preserve overall utility.
title Just Enough Shifts: Mitigating Over-Refusal in Aligned Language Models with Targeted Representation Fine-Tuning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.04250