Instead of using a quadratic penalty, [1]}, [2]} replace the objective in (REF ) by the augmented Lagrangian function
\(\sum _{t = 1}^T f_t(x_t) + \beta ^\top z + \frac{\theta }{2} \left\Vert z \right\Vert ^2,\) 
