MortezaMardani / Neural-PGD

This code implements the neural proximal gradient descent (PGD) algorithm proposed in https://arxiv.org/abs/1806.03963. The idea is to unroll the proximal gradient descent algorithm and model the proximal using a neural network. Adopting residual network (ResNet) as the proximal, a recurrent neural net (RNN) is implemented to learn the proximal.…
24Updated 6 years ago

Related projects

Alternatives and complementary repositories for Neural-PGD