Search papers, labs, and topics across Lattice.
This paper introduces Online Dynamic Batching (ODB), a novel system that enhances batch formation during LLM training by synchronizing it with observable preprocessing costs, thus addressing inefficiencies in traditional offline batch samplers. ODB achieves significant throughput improvements—1.58-2.51x on single-node and 1.71-3.78x on two-node setups—while maintaining comparable quality to fixed-batch methods. The approach is implemented without requiring changes to the model, optimizer, or attention kernel, making it a practical drop-in solution for high-heterogeneity fine-tuning scenarios.
ODB achieves up to 4.43x throughput gains in LLM training without sacrificing quality, revolutionizing how we approach dynamic batching in heterogeneous environments.
Modern LLM training breaks a core assumption behind offline batch samplers: the true training cost of a sample is only observable after preprocessing, augmentation, templating, tokenization, and multimodal visual-token expansion. Unless one pays for a preprocessing- and augmentation-dependent length cache, batch construction is therefore blind to the quantity that determines padding, memory use, and GPU saturation. We introduce Online Dynamic Batching (ODB), a DataLoader-side drop-in system that moves batch formation to this point of accurate observability while preserving DDP step alignment. We formalize this synchronization requirement as the Distributed Group Alignment Problem and prove deadlock-free bounded termination with default join-mode identity coverage and opt-in non-join sample-quota closure. ODB requires no model, optimizer, or attention-kernel changes and is released as online-dynamic-batching with lightweight trainer adapters. Across public 2B/8B Qwen3-VL runs on UltraChat/LLaVA/ShareGPT4o, ODB improves literal emitted-sample throughput vs. fixed-batch Standard by 1.58-2.51x on single-node Full FT/LoRA and 1.71-3.78x on two-node Full FT, with Standard-comparable quality; production MM-Mix reaches 4.43x. Against GMT/BMT offline token-budget oracles, ODB is within 15% on UltraChat/LLaVA and faster on high-CV ShareGPT4o: 2.24-2.39x single-node Full FT/LoRA and 3.06-3.69x two-node Full FT. Together, ODB occupies the online/drop-in regime for high-heterogeneity LLM fine-tuning: large throughput gains at Standard-comparable quality, formal DGAP guarantees, and no length-cache precompute or kernel rewrites.
