Self-CTRL: Self-Consistency Training with Reinforcement Learning

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

Self-Consistency Training with Reinforcement Learning (Self-CTRL) is a new method designed to enhance the alignment between a language model's self-explanations and its actual behavior. This approach optimizes for consistency by either refining explanations to better predict behavior or adjusting behavior to more closely match explanations. Applied to a probabilistic reasoning task, Self-CTRL improved the correlation between self-reported and behaviorally-measured latent biases from R^2=0.24 to R^2=0.64 on held-out distributions, achieving generalization comparable to direct ground-truth supervision. In a constitutional AI setting, the method generated rules that accurately described model behavior on unseen requests, boosting third-party auditor refusal predictions from 36% to 92%. Furthermore, behavior updates reduced the HarmBench failure rate from 15.0% to 0.5% without increasing refusals on harmless prompts. This work offers a general recipe for training more transparent, controllable, and safer AI models.

Key takeaway

For AI Ethicists or Machine Learning Engineers developing safer AI, Self-CTRL offers a robust approach to improve model transparency and control. You should consider integrating self-consistency training to align your language models' stated rules with their actual responses. This method can significantly reduce harmful outputs and enhance auditability, ensuring your models are more trustworthy and predictable in critical applications.

Key insights

Self-CTRL aligns LM explanations and behavior, improving transparency, safety, and control.

Principles

Method

Self-CTRL uses reinforcement learning to optimize for consistency. It updates either LM explanations to predict behavior or LM behavior to match explanations, iteratively aligning both.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Ethicist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.