跳到正文
原文
arXiv· arXiv:2405.19544· Xinmeng Huang, Shuo Li, Edgar Dobriban, Osbert Bastani, Hamed Hassani, Dongsheng Ding·本站收录· 原文发表

通过最优对偶化实现大语言模型的一次性安全对齐

One-Shot Safety Alignment for Large Language Models via Optimal Dualization

AI 导读

研究者提出通过最优对偶化实现大语言模型安全对齐的方法,将带安全约束的 RLHF 转化为等价的无约束对齐问题。该方法预先优化一个具有闭式形式的光滑凸对偶函数,免去繁琐的原始对偶策略迭代,从而大幅降低计算负担并提升训练稳定性。论文在基于模型和基于偏好两种设定下分别给出实用算法 MoCAN 与 PeCAN,较广范围的实验验证了两种算法的有效性。

阅读原文arxiv.org