Constitutional AI: Harmlessness from AI Feedback

Yuntao Bai, Saurav Kadavath, Sandipan Kundu, Amanda Askell, Jackson Kernion, Andy Jones, Anna Chen, Anna Goldie, Azalia Mirhoseini, Cameron McKinnon, et al.

2022 · arXiv

Constitutional AI: Harmlessness from AI Feedback

Problem

Framing

RLHF needs large harmfulness-label datasets and tends to buy harmlessness with evasive refusals. Constitutional AI replaces those labels with a short natural-language constitution plus AI self-critique, revision, and preference judgments, yielding a more harmless, less evasive assistant.

Currently Used Methods

Foundational

Proposed Method

Architecture

The pipeline has two stages. Supervised CAI samples harmful responses from a helpful RLHF model, then applies critique \rightarrow revision under one sampled constitutional principle; RL-CAI then uses AI preference judgments to train a preference model and optimize the policy.

Verified pipeline diagram: a two-stage Constitutional AI process with supervised critique/revision finetuning followed by RLAIF preference-model training and RL.

Loss / Objective

The RL stage trains on AI preference targets for two candidate responses under a constitutional principle.

pθ(ABx)=exprθ(x,yA)exprθ(x,yA)+exprθ(x,yB)p_\theta(A \succ B \mid x) = \frac{\exp r_\theta(x, y_A)}{\exp r_\theta(x, y_A) + \exp r_\theta(x, y_B)}

Sampling Rule / Algorithm

The supervised stage repeatedly rewrites a sampled response with critique-conditioned revision.

y(t+1)pθ ⁣(yx,y(t),ct,critique(x,y(t),ct)),ctUnif(C)y^{(t+1)} \sim p_\theta\!\left(y \mid x, y^{(t)}, c_t, \text{critique}(x, y^{(t)}, c_t)\right), \quad c_t \sim \mathrm{Unif}(\mathcal{C})

Training Procedure

Evaluation

Datasets

Metrics

Headline results

Verified results plot: helpfulness-vs-harmlessness frontier where Constitutional RL traces a better Pareto curve than standard RLHF, with a CoT variant highest on harmlessness.

Ablations

Method Strengths and Weaknesses

Strengths

Weaknesses

Suggestions from the authors

Links

Prior Papers

Further Papers