让大模型推理与训练共用多卡,提升资源利用率。
LeMix: Unified Scheduling for LLM Training and Inference on Multi-GPU Systems
- 动态调度系统,根据负载特征分配资源。
- 吞吐量最高提升3.53倍,响应延迟达标率提高2.12倍。
- 适合需要持续更新的大模型生产部署场景。
大型语言模型的部署常需同时支持在线推理和持续重训练以保持与新数据和用户反馈对齐。现有做法将两者分置于独立服务器上分阶段执行,导致显著效率损失(如GPU空闲)和分布式环境中对新数据响应延迟。我们实证分析发现,效率损失源于服务期间动态请求到达及流水线并行训练中的负载异构性。为此,我们提出LeMix,一个统一调度系统,用于共置并管理并发的LLM推理与训练任务。LeMix结合离线性能分析、执行预测机制与运行时调度,动态调整资源分配,依据任务特性和系统状态应对共享节点上的任务干扰。通过理解任务行为与共执行冲突,LeMix在不牺牲推理响应速度的前提下,显著提升资源利用率与服务质量。评估显示,相比传统分离方案,LeMix可提升吞吐量达3.53倍,降低推理误差0.61倍,并使响应时间服务等级协议达成率提高2.12倍。据我们所知,这是首个揭示并利用推理与训练联合机会的工作,为生产环境中的高效大模型部署开辟新路径。
原文摘要 · Abstract (English)
Modern deployment of large language models (LLMs) frequently involves both inference serving and continuous retraining to stay aligned with evolving data and user feedback. Common practices separate these workloads onto distinct servers in isolated phases, causing substantial inefficiencies (e.g., GPU idleness) and delayed adaptation to new data in distributed settings. Our empirical analysis reveals that these inefficiencies stem from dynamic request arrivals during serving and workload heterogeneity in pipeline-parallel training. To address these challenges, we propose LeMix, a system for co-locating and managing concurrent LLM serving and training workloads. LeMix integrates offline profiling, execution prediction mechanisms, and runtime scheduling to dynamically adapt resource allocation based on workload characteristics and system conditions. By understanding task-specific behaviors and co-execution interference across shared nodes, LeMix improves utilization and serving quality without compromising serving responsiveness. Our evaluation shows that LeMix improves throughput by up to 3.53x, reduces inference loss by up to 0.61x, and delivers up to 2.12x higher response time SLO attainment over traditional separate setups. To our knowledge, this is the first work to uncover and exploit the opportunities of joint LLM inference and training, paving the way for more resource-efficient deployment of LLMs in production environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。