Sign in

Jörg Franke

@jfranke.bsky.social
2 followers 3 following 5 posts

PhD student in the Machine Learning Lab at the University of Freiburg - Core Deep Learning Research with some applications in bio.

PostsRepliesMedia
Jörg Franke @jfranke.bsky.social · 09/12/2024
🧵5/5 - Come discuss with us at NeurIPS: 📍 East Exhibit Hall A-C 🎯 Poster #2803 ⏰ Thu Dec 12, 11am-2pm PST Or check our paper: arxiv.org/abs/2311.09058
arxiv.org
Improving Deep Learning Optimization through Constrained Parameter Regularization
Regularization is a critical component in deep learning. The most commonly used approach, weight decay, applies a constant penalty coefficient uniformly across all parameters. This may be overly restr...
000
Jörg Franke @jfranke.bsky.social · 09/12/2024
🧵4/5 - For example, when pretrain GPT2s, AdamCPR outperforms AdamW with the same budget or only requires 2/3 of the budget to reach the same score.
100
Jörg Franke @jfranke.bsky.social · 09/12/2024
🧵3/5 - CPR can be used with any gradient-based optimization algorithm, e.g. Adam. You can find our AdamCPR implementation at github.com/automl/CPR or via pip install pytorch-cpr
100
Jörg Franke @jfranke.bsky.social · 09/12/2024
🧵2/5 - We reformulate regularization as an inequality-constrained optimization problem which leads to a couple of benefits: ✅ Individual and dynamic weight regularization ✅ Outperforms weight decay ✅ No additional or fewer hyperparameters ✅ Minor or no runtime overhead
100
Jörg Franke @jfranke.bsky.social · 09/12/2024
Thrilled to present our work on Constrained Parameter Regularization (CPR) at #NeurIPS2024! Our novel deep learning regularization outperforms weight decay across various tasks. neurips.cc/virtual/2024... This is joint work with Michael Hefenbrock, Gregor Köhler, and Frank Hutter 🧵👇
neurips.cc
NeurIPS Poster Improving Deep Learning Optimization through Constrained Parameter RegularizationNeurIPS 2024
121