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 Optimizer Objective Found Identical to KL-Shampoo(3 posts)→

Original post →

More from Research

Research channel →