高级检索

    面向多Chiplet加速器的多智能体强化学习多DNN负载在线调度算法

    Multi-Agent Reinforcement Learning based Online Scheduling Algorithm for Multi-DNN Workloads on Multi-Chiplet Accelerators

    • 摘要: 随着深度神经网络(DNN)应用日益广泛,基于芯粒(Chiplet)架构的加速器成为了服务器平台上支撑大规模DNN推理服务的关键设施。然而,并行多DNN负载的实时性与动态性特征给调度算法带来了严峻挑战,现有算法在决策开销、服务水平协议(SLA)保障及动态适应性方面存在局限。为此,提出了一种面向多Chiplet加速器的多智能体强化学习多DNN负载在线调度算法(MAOSA),以最大化整体SLA满足率为目标。该算法将DNN任务划分为DNN块并建模为智能体,利用注意力机制并行生成包含执行顺序与PE级资源分配的调度策略,显著降低了决策开销;设计了考虑时间裕量的协作奖励函数以优化SLA,并引入周期性统一抢占机制来适应负载动态变化。实验结果表明,与当前最先进算法相比,所提出算法的SLA满足率最高可提升18%,STP提升31.9%,尾延迟降低28.7%,调度决策时间降低约11倍,有效提升了系统性能与服务质量。

       

      Abstract: With the increasing popularity of Deep Neural Network (DNN) applications, chiplet-based accelerators have become critical infrastructure for supporting large-scale DNN inference services on server platforms. However, the real-time and dynamic characteristics of parallel multi-DNN workloads pose severe challenges to scheduling algorithms, while existing algorithms suffer from limitations in decision overhead, Service Level Agreement (SLA) assurance, and dynamic adaptability. To address these issues, this paper proposes MAOSA, a multi-agent reinforcement learning based online scheduling algorithm for multi-DNN workloads on multi-chiplet accelerators, aiming to maximize the overall SLA satisfaction rate. The proposed algorithm partitions DNN tasks into DNN blocks and models them as agents. It utilizes the attention mechanism to generate scheduling strategies in parallel, which include execution order and PE-level resource allocation, thereby significantly reducing decision overhead. Furthermore, a cooperative reward function considering time slack is designed to optimize SLA, and a periodic unified preemption mechanism is introduced to adapt to dynamic load changes. Experimental results show that, compared with state-of-the-art algorithms, the proposed algorithm improves the SLA satisfaction rate by up to 18% and system throughput (STP) by 31.9%, reduces tail latency by 28.7%, and decreases scheduling execution time by approximately 11x, effectively enhancing system performance and service quality.

       

    /

    返回文章
    返回