liveBook · Manning

Mode covering

As explained in Reinforcement Learning from Human Feedback

Mode covering is a distribution-approximation tendency that spreads probability across the major modes of a target distribution.

Mode covering is a distribution-approximation tendency in which a model spreads probability across the major modes of a target distribution. Forward KL encourages this behavior because it evaluates errors under the target distribution and penalizes missing regions where the target has mass. section 15.2.2

This pressure contrasts with reverse KL, which evaluates regions under the model’s own distribution and therefore tends to concentrate on one high-probability mode. In multimodal policies, that difference produces an unexpected forgetting pattern: forward KL can stretch toward a new mode and pull mass from an old one, while reverse KL can shift the new mode without disturbing the old mode. section 15.2.2

Old and new policy modes

Imagine an old mode encoding prior knowledge and a new mode representing a target task. Forward-KL training tries to cover both target modes, whereas reverse-KL training can focus on the high-reward new mode while preserving the old mode. section 15.2.2

Equation 15.5
\[\label{eq:sft_forward_kl_1} \text{KL}(\pi_\star \| \pi_\theta) = \mathbb{E}_{(x,y) \sim \mathcal{D}} \left[ \log \pi_\star(y \mid x) - \log \pi_\theta(y \mid x) \right].\]

Use the forward-KL equation to inspect how expectation under the target distribution creates pressure to cover its modes. section 15.2.2

How Reinforcement Learning from Human Feedback covers it

How deep it goes
Defines the pressure, identifies its expectation-based cause, and contrasts its multimodal forgetting behavior with reverse-KL mode seeking.
Examples
Uses old and new policy modes plus a forward-KL-versus-reverse-KL forgetting diagram.
What you can do after
Evaluate/understand how KL direction changes probability allocation and forgetting across old and new modes.
How it connects
Connects forward KL to SFT, reverse KL to RL, and mode covering to multimodal policy forgetting.

Related concepts

This explainer was generated by Manning's knowledge graph from the text of Reinforcement Learning from Human Feedback. The section links point to the parts of the book it draws on.