SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit models - Inria - Institut national de recherche en sciences et technologies du numérique
Pré-Publication, Document De Travail Année : 2021

SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit models

Résumé

In recent years, implicit deep learning has emerged as a method to increase the depth of deep neural networks. While their training is memory-efficient, they are still significantly slower to train than their explicit counterparts. In Deep Equilibrium Models (DEQs), the training is performed as a bi-level problem, and its computational complexity is partially driven by the iterative inversion of a huge Jacobian matrix. In this paper, we propose a novel strategy to tackle this computational bottleneck from which many bi-level problems suffer. The main idea is to use the quasi-Newton matrices from the forward pass to efficiently approximate the inverse Jacobian matrix in the direction needed for the gradient computation. We provide a theorem that motivates using our method with the original forward algorithms. In addition, by modifying these forward algorithms, we further provide theoretical guarantees that our method asymptotically estimates the true implicit gradient. We empirically study this approach in many settings, ranging from hyperparameter optimization to large Multiscale DEQs applied to CIFAR and ImageNet. We show that it reduces the computational cost of the backward pass by up to two orders of magnitude. All this is achieved while retaining the excellent performance of the original models in hyperparameter optimization and on CIFAR, and giving encouraging and competitive results on ImageNet.
Fichier principal
Vignette du fichier
2106.00553.pdf (846.18 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03246167 , version 1 (27-07-2021)
hal-03246167 , version 2 (31-01-2022)
hal-03246167 , version 3 (10-03-2023)

Identifiants

Citer

Zaccharie Ramzi, Florian Mannel, Shaojie Bai, Jean-Luc Starck, Philippe Ciuciu, et al.. SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit models. 2021. ⟨hal-03246167v1⟩
146 Consultations
168 Téléchargements

Altmetric

Partager

More