Optimization of Multi-Resource Scheduling Strategies in Distributed Training Environments
In large-scale model training, distributed training frameworks offer an effective solution to the limitations of a single GPU or node in handling massive model parameters and datasets. By distributing training tasks across multiple nodes, distributed model training enables parallel utilization of computational resources, significantly improving training efficiency. However, efficient scheduling of multi-resources (such as computation, storage, and network) in distributed environments remains a critical challenge. This paper proposes an optimized multi-resource scheduling strategy that integrates Kubernetes’ scheduling principles. By improving the scheduling algorithm, the strategy effectively enhances resource utilization and task completion efficiency. Experimental results demonstrate that the proposed method significantly reduces task waiting time and improves overall system throughput in distributed environments.
Authors
- Yan Jiang (ORCID: https://orcid.org/0000-0002-9433-4161)
- Yong Cui (ORCID: https://orcid.org/0000-0002-5171-739X)
- Yu Wang (ORCID: https://orcid.org/0000-0002-9807-2293)
- Dan Zhang (ORCID: https://orcid.org/0000-0002-5871-7990)
Institutions
- Shenyang Institute of Automation (CN)
- Shenyang University of Technology (CN)
- Chinese Academy of Sciences (CN)
- Wuhu Hit Robot Technology Research Institute (CN)
Publication Details
- Published
- 2024-12-20
- DOI
- https://doi.org/10.1109/aiim64537.2024.10934522
- Primary Topic
- Scheduling and Timetabling Solutions
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China