目录
以下内置转自: Google Brain新提出的优化器“Lion”,效果要比Adam(W)更好 - 知乎
1 简单、内存高效、运行速度更快
与 AdamW 和各种自适应优化器需要同时保存一阶和二阶矩相比,Lion 只需要动量,将额外的内存占用减半。 这在训练大型模型和大Batch size时很有用。 例如,AdamW 需要至少 16 个 TPU V4 芯片来训练图像大小为 224、批量大小为 4,096 的 ViT-B/16,而 Lion 只需要8个。
另一个显而易见的好处是,由于 Lion 的简单性,Lion 在我们的实验中具有更快的运行时间(step/s),通常比 AdamW 和 Adafactor 提速 2-15%,具体取决于任务、代码库和硬件。
超参设置
- Lion 很简单,与 AdamW 和 Adafactor 相比,超参
Lion优化器比AdamW和Adafactor更简单、内存高效,运行速度更快。它只需保存动量,降低内存占用,尤其适合大规模模型训练。在实验中,Lion的运行速度通常提高2-15%,并简化超参数设置。在人脸关键点和YOLOv8训练中,Lion展现出优于AdamW的性能,但数据量增大时需要注意学习率和衰减策略。
订阅专栏 解锁全文
1427

被折叠的 条评论
为什么被折叠?



