Cost-Aware GRPO: weighting RL samples by informativeness and cost to cut training compute

burkov · x · 2026-11-12

Andriy Burkov highlights an article on cutting RL training costs for language models. Since each training example costs very different compute — long generated responses cost far more than short ones — the work proposes Cost-Aware SGD, adapted for LM policy optimization as Cost-Aware GRPO. The method upweights samples that are informative for the model update while accounting for their processing cost: response length serves as the cost proxy, and the absolute value of each response's advantage (training signal strength) is used as a practical informativeness estimate. The goal is hitting a target performance level at lower total compute, not just fewer training steps.

Original post →

More from Infra

Infra channel →