Home

Awesome

ProxSGD

This is the TensorFlow implementation of the ProxSGD algorithm proposed in following paper:

"ProxSGD: Training Structured Neural Networks under Regularization and Constraints," International Conference on Learning Representation (ICLR), Apr. 2020. URL

by Yang Yang, Yaxiong Yuan, Avraan Chatzimichailidis, Ruud JG van Sloun, Lei Lei, and Symeon Chatzinotas.

This repository contains the simulations in Section 4.1 and 4.3. The simulations in Section 4.2 are implemented in PyTorch and available at here.