A Deep Reinforcement Learning With Transformer Integration for Directed Acyclic Graph Scheduling in Edge Networks
The rapid adoption of 5G technology and Internet of things (IoT) devices has fueled significant growth in intelligent applications, increasing their complexity beyond simple task definitions. Scheduling intelligent applications modeled as directed acyclic graphs (DAGs) has thus emerged as a crucial challenge. Our proposed solution is a deep reinforcement learning (DRL) framework that uniquely integrates proximal policy optimization (PPO) with a transformer-based module for scheduling DAG applications. Unlike other approaches that rely on predefined priorities or static optimization algorithms, our approach enables agents to autonomously explore task execution orders and dynamically adapt to changing network resource conditions, learning optimal scheduling strategies. The algorithm leverages transformers to handle complex task dependencies, minimizing application duration and user energy consumption by jointly optimizing application processing order, task priorities, transmit power, offloading decisions, and computational frequency. Through a series of simulations, we prove the effectiveness of the proposed algorithm and demonstrate the performance comparison under different settings, providing a more flexible and robust solution for DAG scheduling in edge networks.
Authors
- Xifei Song (ORCID: https://orcid.org/0000-0002-8086-0013)
- Qingqi Pei (ORCID: https://orcid.org/0000-0001-7601-5434)
- Lei Liu (ORCID: https://orcid.org/0000-0001-8173-0408)
- Ning Zhang (ORCID: https://orcid.org/0000-0002-8781-4925)
- F. Richard Yu (ORCID: https://orcid.org/0000-0003-1006-7594)
- Jie Feng (ORCID: https://orcid.org/0000-0002-5474-3286)
Institutions
- Xidian University (CN)
- Xi’an University of Posts and Telecommunications (CN)
- Carleton University (CA)
- University of Windsor (CA)
Publication Details
- Journal
- IEEE Transactions on Wireless Communications
- Published
- 2025-10-15
- DOI
- https://doi.org/10.1109/twc.2025.3619100
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.00