In the evolving landscape of foundation models, fine-tuning pre-trained models with first-order (FO) optimizers like SGD and Adam has become standard practice. However, as these models grow in size, the substantial memory overhead from back-propagation (BP) for FO gradient computation presents a significant challenge. Addressing this issue is crucial, particularly for applications like on-device training where memory efficiency is paramount. This chapter introduces a shift towards back-propagation-free, zeroth-order (ZO) optimization as a solution to reduce memory costs during model fine-tuning. We will explore both the algorithmic foundations of ZO optimization and its applications in memory-efficient fine-tuning.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Memory-Efficient Fine-Tuning for Foundation Models

  • Pin-Yu Chen,
  • Sijia Liu

摘要

In the evolving landscape of foundation models, fine-tuning pre-trained models with first-order (FO) optimizers like SGD and Adam has become standard practice. However, as these models grow in size, the substantial memory overhead from back-propagation (BP) for FO gradient computation presents a significant challenge. Addressing this issue is crucial, particularly for applications like on-device training where memory efficiency is paramount. This chapter introduces a shift towards back-propagation-free, zeroth-order (ZO) optimization as a solution to reduce memory costs during model fine-tuning. We will explore both the algorithmic foundations of ZO optimization and its applications in memory-efficient fine-tuning.