PSGD's QEQ update rule yields same rule as KL+AIRM

YouJiacheng · x · 2026-08-31

YouJiacheng notes that it is surprising PSGD's original objective with QEQ update gives exactly the same update rule as KL+AIRM. This finding adds to the interesting mathematical connections discussed in the previous post regarding the optimizer's cost function.

Related event: PSGD Objective Found Mathematically Equivalent to KL-Shampoo(6 posts)→

Original post →

More from Research

Research channel →