Sorry, you need to enable JavaScript to visit this website.

Learning Gradients of Convex Functions with Monotone Gradient Networks

DOI:
10.60864/nrrn-zb12
Citation Author(s):
Shreyas Chaudhari*, Srinivasa Pranav*, José M.F. Moura
Submitted by:
SRINIVASA PRANAV
Last updated:
17 November 2023 - 12:07pm
Document Type:
Poster
Document Year:
2023
Event:
Presenters:
Srinivasa Pranav and Shreyas Chaudhari
Paper Code:
6453
 

While much effort has been devoted to deriving and analyzing effective convex formulations of signal processing problems, the gradients of convex functions also have critical applications ranging from gradient-based optimization to optimal transport. Recent works have explored data-driven methods for learning convex objective functions, but learning their monotone gradients is seldom studied. In this work, we propose C-MGN and M-MGN, two monotone gradient neural network architectures for directly learning the gradients of convex functions. We show that, compared to state of the art methods, our networks are easier to train, learn monotone gradient fields more accurately, and use significantly fewer parameters. We further demonstrate their ability to learn optimal transport mappings to augment driving image data.

up
1 user has voted: SRINIVASA PRANAV