Md. Asif Uddin
    I.8.X04

    An adaptive denominator changes a penalty

    counterexample▲▲△

    Consider the fixed diagonal preconditioner P=diag⁡(1,0.01)P=\operatorname{diag}(1,0.01), weights w=(1,1)w=(1,1), data gradient g=(1,100)g=(1,100), step size η=0.01\eta=0.01, and λ=0.1\lambda=0.1. Compare w−ηP(g+λw)w-\eta P(g+\lambda w) with (1−ηλ)w−ηPg(1-\eta\lambda)w-\eta Pg. State what this example proves, and what it leaves out of Adam.

    Hint

    In the first expression the penalty is multiplied by PP. In the second it is not.

    Solution

    The coupled gradient is (1.1,100.1)(1.1,100.1). Preconditioning gives (1.1,1.001)(1.1,1.001), so the first update is

    wcoupled=(0.989, 0.98999).w_{\mathrm{coupled}}=(0.989,\ 0.98999).

    For the decoupled expression, Pg=(1,1)Pg=(1,1) and (1−ηλ)w=(0.999,0.999)(1-\eta\lambda)w=(0.999,0.999), giving

    wdecoupled=(0.989, 0.989).w_{\mathrm{decoupled}}=(0.989,\ 0.989).

    The first coordinate agrees because its preconditioner equals one. The second does not. The coupled penalty displacement there is 0.000010.00001, one hundredth of the decoupled displacement 0.0010.001.

    This is an exact counterexample for a fixed preconditioned update. It isolates one reason the plain-gradient identity (I.8.6) fails. Actual Adam does more: a penalty included in the gradient also changes the history-dependent first and second moments, so its denominator is not a fixed PP shared by both algorithms. Do not present this frozen-preconditioner calculation as a complete Adam trajectory.

    Draws on