Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

YouTube

Accelerating the Training of Large Language Models - Efficient Activation Rematerialization and Optimal Hybrid Parallelism

USENIX via YouTube

Overview

Explore cutting-edge techniques for accelerating the training of large language models in this 19-minute conference talk from USENIX ATC '24. Dive into novel strategies for efficient activation rematerialization and optimal hybrid parallelism presented by researchers from Kuaishou Technology. Learn about Pipeline-Parallel-Aware Offloading, which maximizes host memory utilization for storing activations, and Compute-Memory Balanced Checkpointing, which balances activation memory and computational efficiency. Discover an efficient searching method for optimizing hybrid parallelism parameters, considering both offloading and checkpointing for optimal performance. Examine the results of extensive experiments on public benchmarks, showcasing significant improvements in Model FLOPs Utilization (MFU) for large-scale models with varying context window sizes. Gain insights into the latest advancements in training large-scale models, focusing on optimizing activation strategies and exploring various parallel training options to enhance the balance between computation and memory utilization.

Syllabus

USENIX ATC '24 - Accelerating the Training of Large Language Models using Efficient Activation...

Taught by

USENIX

Reviews

Start your review of Accelerating the Training of Large Language Models - Efficient Activation Rematerialization and Optimal Hybrid Parallelism

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Someone learning on their laptop while sitting on the floor.