A Deep Reinforcement Learning Approach With Attention Mechanism for DAG Task Scheduling in Data Centers

ABSTRACT Task scheduling algorithms for data centers must be capable of making instantaneous decisions based on the current state of the system. However, due to information limitations, these scheduling algorithms often fail to achieve optimal scheduling plans. To address the information bottleneck faced in DAG (Directed Acyclic Graph) task scheduling within data centers, this paper proposes a deep reinforcement learning scheduling model based on a DAG attention mechanism. This model utilizes the attention mechanism to capture the potential relationships between dependent tasks, thereby improving scheduling efficiency and system performance under limited information conditions. The experimental results indicate that our DAG attention mechanism can significantly reduce makespan.