Switch language한국어
Back to the list

Trust-Region Behavior Blending for On-Policy Distillation

TL;DR AI

Key summary

2 min read
  1. Trust-Region Behavior Blending is a new warmup method for on-policy distillation.

  2. It replaces weak early student rollouts with the closest teacher-like behavior within a student-centered KL trust region, while keeping the distillation loss unchanged.

  3. The KL constraint is gradually annealed to zero, so training returns to pure student rollouts later.

  4. In math-reasoning distillation tests, it achieved the best average performance among the compared methods.

Read the original