Mechanism

Replaces PPO's multiplicative ratio clipping with a squared-error loss between the log-policy ratio log⁡πθ(yt∣st)πref(yt∣st)\log \frac{\pi_\theta(y_t\mid s_t)}{\pi_{\text{ref}}(y_t\mid s_t)} and the scaled token advantage A^tβ\frac{\hat{A}_t}{\beta}, enforcing explicit trust-region control around the reference policy.

Practical Notes

Avoids flat zero-gradient regions outside the [1−ϵ,1+ϵ][1-\epsilon, 1+\epsilon] PPO clip interval.

References

  1. Fine-Tuning Language Models with Advantage-Induced Policy Alignment ↗

    Banghua Zhu, Hiteshi Sharma, Felipe Vieira Frujeri, Shi Dong, Chenguang Zhu, Michael I. Jordan et al. · 2023

    All 7 authors

    Banghua Zhu, Hiteshi Sharma, Felipe Vieira Frujeri, Shi Dong, Chenguang Zhu, Michael I. Jordan, Jiantao Jiao