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. SAMPa: Sharpness-aware Minimization Parallelized
 
conference poster not in proceedings

SAMPa: Sharpness-aware Minimization Parallelized

Xie, Wanyun  
•
Pethick, Thomas Michaelsen  
•
Cevher, Volkan  orcid-logo
December 2024
38th Annual Conference on Neural Information Processing Systems

Sharpness-aware minimization (SAM) has been shown to improve the generalization of neural networks. However, each SAM update requires sequentially computing two gradients, effectively doubling the per-iteration cost compared to base optimizers like SGD. We propose a simple modification of SAM, termed SAMPa, which allows us to fully parallelize the two gradient computations. SAMPa achieves a twofold speedup of SAM under the assumption that communication costs between devices are negligible. Empirical results show that SAMPa ranks among the most efficient variants of SAM in terms of computational time. Additionally, our method consistently outperforms SAM across both vision and language tasks. Notably, SAMPa theoretically maintains convergence guarantees even for fixed perturbation sizes, which is established through a novel Lyapunov function. We in fact arrive at SAMPa by treating this convergence guarantee as a hard requirement—an approach we believe is promising for developing SAM-based methods in general. Our code is available at https://github.com/LIONS-EPFL/SAMPa.

  • Details
  • Metrics
Type
conference poster not in proceedings
Author(s)
Xie, Wanyun  

EPFL

Pethick, Thomas Michaelsen  

EPFL

Cevher, Volkan  orcid-logo

EPFL

Date Issued

2024-12

Subjects

ML-AI

URL

Link to the code

https://github.com/LIONS-EPFL/SAMPa
Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LIONS  
Event nameEvent acronymEvent placeEvent date
38th Annual Conference on Neural Information Processing Systems

NeurIPS 2024

Vancouver, BC, Canada

2024-12-10 - 2024-12-15

Available on Infoscience
January 10, 2025
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/242683
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