Convexity in ReLU Neural Networks: beyond ICNNs?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Gagneux, Anne, Massias, Mathurin, Soubies, Emmanuel, Gribonval, Rémi
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915232989839360
author Gagneux, Anne
Massias, Mathurin
Soubies, Emmanuel
Gribonval, Rémi
author_facet Gagneux, Anne
Massias, Mathurin
Soubies, Emmanuel
Gribonval, Rémi
contents Convex functions and their gradients play a critical role in mathematical imaging, from proximal optimization to Optimal Transport. The successes of deep learning has led many to use learning-based methods, where fixed functions or operators are replaced by learned neural networks. Regardless of their empirical superiority, establishing rigorous guarantees for these methods often requires to impose structural constraints on neural architectures, in particular convexity. The most popular way to do so is to use so-called Input Convex Neural Networks (ICNNs). In order to explore the expressivity of ICNNs, we provide necessary and sufficient conditions for a ReLU neural network to be convex. Such characterizations are based on product of weights and activations, and write nicely for any architecture in the path-lifting framework. As particular applications, we study our characterizations in depth for 1 and 2-hidden-layer neural networks: we show that every convex function implemented by a 1-hidden-layer ReLU network can be also expressed by an ICNN with the same architecture; however this property no longer holds with more layers. Finally, we provide a numerical procedure that allows an exact check of convexity for ReLU neural networks with a large number of affine regions.
format Preprint
id arxiv_https___arxiv_org_abs_2501_03017
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Convexity in ReLU Neural Networks: beyond ICNNs?
Gagneux, Anne
Massias, Mathurin
Soubies, Emmanuel
Gribonval, Rémi
Machine Learning
Convex functions and their gradients play a critical role in mathematical imaging, from proximal optimization to Optimal Transport. The successes of deep learning has led many to use learning-based methods, where fixed functions or operators are replaced by learned neural networks. Regardless of their empirical superiority, establishing rigorous guarantees for these methods often requires to impose structural constraints on neural architectures, in particular convexity. The most popular way to do so is to use so-called Input Convex Neural Networks (ICNNs). In order to explore the expressivity of ICNNs, we provide necessary and sufficient conditions for a ReLU neural network to be convex. Such characterizations are based on product of weights and activations, and write nicely for any architecture in the path-lifting framework. As particular applications, we study our characterizations in depth for 1 and 2-hidden-layer neural networks: we show that every convex function implemented by a 1-hidden-layer ReLU network can be also expressed by an ICNN with the same architecture; however this property no longer holds with more layers. Finally, we provide a numerical procedure that allows an exact check of convexity for ReLU neural networks with a large number of affine regions.
title Convexity in ReLU Neural Networks: beyond ICNNs?
topic Machine Learning
url https://arxiv.org/abs/2501.03017