arXiv AI

Beyond Average Safety: Chance-Constrained LLM Fine-tuning

The paper introduces a chance-constrained approach to fine‑tune large language models (LLMs) that limits the proportion of safety examples whose performance degrades beyond a set threshold relative to a reference model. By replacing the discontinuous violation indicator with a differentiable majorization, the authors derive a tractable, conservative constraint and a closed‑form, constraint‑aware gradient update that focuses on examples near or above the degradation threshold. Experiments on harmful fine‑tuning across three tasks and models show that this tail‑aware method consistently outperforms existing safety‑preserving baselines, suggesting that safety preservation should be treated as a reliability‑constrained optimization problem rather than average‑risk regularization.

arXiv AI
Nov 5

UpSafe$^\circ$C: Upcycling for Controllable Safety in Large Language Models

UpSafe°C is a unified framework that enhances large language model safety by upcycling safety‑critical layers into a sparse Mixture‑of‑Experts structure, with a router acting as a soft guardrail. It employs a two‑stage supervised fine‑tuning strategy to improve safety discrimination while preserving general capabilities, and introduces a safety temperature mechanism for dynamic inference‑time control of the safety‑utility trade‑off. Experiments across multiple benchmarks and model scales show robust safety improvements against harmful and jailbreak inputs while maintaining competitive general performance.

By Yuhao Sun, Zhuoer Xu, Shiwen Cui, Kun Yang, Lingyun Yu, Yongdong Zhang, Hongtao Xie
arXiv AI
Sep 10

Suan: Rectifying Direct Preference Safety Alignment in Large Language Models

The paper introduces Suan, a new preference optimization algorithm designed to improve safety alignment in large language models. Suan operates directly at the gradient level, avoiding traditional variational derivations, which yields more interpretable and robust training dynamics. Experiments show that Suan outperforms existing methods, achieving superior safety alignment while maintaining response utility.

By Oleksandr Cherednichenko, Roman Klypa