/proxsgd

ProxSGD algorithm in TensorFlow

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

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.