Models & training Paper
LoGRA cuts LLM RL training memory by up to 45.7%
LoGRA stores learning signals as low-rank gradient sketches and uses predicted-KL step control; it reports 45.7% lower average memory and stable 27B-model training for over 1,100 steps on one eight-GPU node. This could make RL post-training feasible where dense Adam runs out of memory.
- WhatLoGRA reduces RL post-training memory by retaining useful learning signals in low-rank gradient sketches.
- MethodThe sketches support model updates and policy synchronization; predicted-KL step control estimates policy changes and adjusts update magnitude.
- Why it mattersLoGRA cut average training memory by up to 45.7% and enabled stable training of a 27B-parameter model for over 1,100 steps on one eight-GPU node.