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. Conferences, Workshops, Symposiums, and Seminars
  4. Improving SAM Requires Rethinking its Optimization Formulation
 
conference paper not in proceedings

Improving SAM Requires Rethinking its Optimization Formulation

Xie, Wanyun  
•
Latorre, Fabian  
•
Antonakopoulos, Kimon  
Show more
2024
41st International Conference on Machine Learning (ICML 2024)

This paper rethinks Sharpness-Aware Minimization (SAM), which is originally formulated as a zero-sum game where the weights of a network and a bounded perturbation try to minimize/maximize, respectively, the same differentiable loss. We argue that SAM should instead be reformulated using the 0-1 loss, as this provides a tighter bound on its generalization gap. As a continuous relaxation, we follow the simple conventional approach where the minimizing (maximizing) player uses an upper bound (lower bound) surrogate to the 0-1 loss. This leads to a novel formulation of SAM as a bilevel optimization problem, dubbed as BiSAM. Through numerical evidence, we show that BiSAM consistently results in improved performance when compared to the original SAM and variants, while enjoying similar computational complexity.

  • Files
  • Details
  • Metrics
Type
conference paper not in proceedings
Author(s)
Xie, Wanyun  
Latorre, Fabian  
Antonakopoulos, Kimon  
Pethick, Thomas Michaelsen  
Cevher, Volkan  orcid-logo
Date Issued

2024

Subjects

ML-AI

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LIONS  
Event nameEvent placeEvent date
41st International Conference on Machine Learning (ICML 2024)

Vienna, Austria

July 21-27, 2024

Available on Infoscience
July 1, 2024
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/208922
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