Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. Improving Lipschitz-Constrained Neural Networks by Learning Activation Functions
 
research article

Improving Lipschitz-Constrained Neural Networks by Learning Activation Functions

Ducotterd, Stanislas  
•
Goujon, Alexis  
•
Bohra, Pakshal Narendra  
Show more
January 1, 2024
Journal of Machine Learning Research

Lipschitz-constrained neural networks have several advantages over unconstrained ones and can be applied to a variety of problems, making them a topic of attention in the deep learning community. Unfortunately, it has been shown both theoretically and empirically that they perform poorly when equipped with ReLU activation functions. By contrast, neural networks with learnable 1-Lipschitz linear splines are known to be more expressive. In this paper, we show that such networks correspond to global optima of a constrained functional optimization problem that consists of the training of a neural network composed of 1-Lipschitz linear layers and 1-Lipschitz freeform activation functions with second-order total-variation regularization. Further, we propose an efficient method to train these neural networks. Our numerical experiments show that our trained networks compare favorably with existing 1-Lipschitz neural architectures.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

ducotterd2401.pdf

Type

Publisher

Version

Published version

Access type

openaccess

License Condition

CC BY-NC-ND

Size

1.43 MB

Format

Adobe PDF

Checksum (MD5)

03810c645d454ec18d0bbee00015e062

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés