mlsanityData to 14 Oct 2026

A Scalable Pipeline for LLM-Teacher Distillation Labeling: Work-Stealing Job Scheduling and Memory-Aware GPU Concurrency

arXiv 2608.15975 ↗First posted 17 Aug 2026cs.DC cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Labeling large text corpora with LLM teachers has become a practical route to training data at scale. At millions of items, hand-labeling every batch is not feasible, and two questions dominate: what label quality a teacher buys per dollar, and how to keep a fleet of GPU workers busy under skewed, failure-prone workloads. We present a simple, reproducible pipeline that addresses both. First, a work-stealing ring pool: each worker owns a queue, drains it first, and then steals from ring successors, with exactly-once task claims via atomic conditional writes and crash tolerance via stale-claim sweeping. The claim protocol requires only a compare-and-set primitive from its storage layer; we implement it on a single SQLite file, which makes the reference implementation dependency-free and the experiments reproducible on one machine. Second, a memory-aware concurrency rule that sizes per-node parallelism by how many model copies fit on the GPU, so the same code runs safely across device sizes. Third, a relabeling benchmark methodology in which the teacher relabels a public dataset that already has gold labels, so quality reduces to an agreement measurement and cost follows from measured throughput. Under skewed load the pool sustains up to 3.4 times the throughput of static sharding while matching it at zero skew, loses 0 of 2,000 tasks when half the workers are killed mid-run (static sharding loses 953), and yields measured quality and cost points for an instruction-tuned teacher on irony and sentiment tasks. All experiments run on public data and commodity hardware; code, tests, and run logs are released.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 7 of its 10 references that are in the map. First posted on arXiv 17 Aug 2026.

Builds on

7 of 10 references.

References · 7

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
  3. 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
  4. 2023ChatGPT Outperforms Crowd-Workers for Text-Annotation Taskscited by 870
  5. 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 758
  6. 2020TweetEval: Unified Benchmark and Comparative Evaluation for Tweet Classificationcited by 417
  7. 2017Snorkel: Rapid Training Data Creation with Weak Supervisioncited by 372

Built on by

Not cited yet.