Diffusion-Based Reinforcement Learning for Edge-Enabled AI-Generated Content Services

被引:16
|
作者
Du, Hongyang [1 ]
Li, Zonghang [2 ]
Niyato, Dusit [1 ]
Kang, Jiawen [3 ]
Xiong, Zehui [4 ]
Huang, Huawei [5 ]
Mao, Shiwen [6 ]
机构
[1] Nanyang Technol Univ, Energy Res Inst NTU, Sch Comp Sci & Engn, Interdisciplinary Grad Program, Singapore 639798, Singapore
[2] Univ Elect Sci & Technol China, Sch Informat & Commun Engn, Chengdu 610054, Peoples R China
[3] Guangdong Univ Technol, Sch Automat, Guangzhou 510006, Peoples R China
[4] Singapore Univ Technol & Design, Pillar Informat Syst Technol & Design, Singapore 487372, Singapore
[5] Sun Yat Sen Univ, Sch Software Engn, Zhuhai 519082, Peoples R China
[6] Auburn Univ, Dept Elect & Comp Engn, Auburn, AL 36849 USA
基金
新加坡国家研究基金会; 中国国家自然科学基金;
关键词
Computational modeling; Metaverse; Task analysis; Servers; Biological system modeling; Mathematical models; Performance evaluation; AI-generated content; and deep reinforcement learning; diffusion models; wireless networks; generative AI;
D O I
10.1109/TMC.2024.3356178
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
As Metaverse emerges as the next-generation Internet paradigm, the ability to efficiently generate content is paramount. AI-Generated Content (AIGC) emerges as a key solution, yet the resource-intensive nature of large Generative AI (GAI) models presents challenges. To address this issue, we introduce an AIGC-as-a-Service (AaaS) architecture, which deploys AIGC models in wireless edge networks to ensure broad AIGC services accessibility for Metaverse users. Nonetheless, an important aspect of providing personalized user experiences requires carefully selecting AIGC Service Providers (ASPs) capable of effectively executing user tasks, which is complicated by environmental uncertainty and variability. Addressing this gap in current research, we introduce the AI-Generated Optimal Decision (AGOD) algorithm, a diffusion model-based approach for generating the optimal ASP selection decisions. Integrating AGOD with Deep Reinforcement Learning (DRL), we develop the Deep Diffusion Soft Actor-Critic (D2SAC) algorithm, enhancing the efficiency and effectiveness of ASP selection. Our comprehensive experiments demonstrate that D2SAC outperforms seven leading DRL algorithms. Furthermore, the proposed AGOD algorithm has the potential for extension to various optimization problems in wireless networks, positioning it as a promising approach for future research on AIGC-driven services.
引用
收藏
页码:8902 / 8918
页数:17
相关论文
共 50 条
  • [1] Exploring Collaborative Distributed Diffusion-Based AI-Generated Content (AIGC) in Wireless Networks
    Du, Hongyang
    Zhang, Ruichen
    Niyato, Dusit
    Kang, Jiawen
    Xiong, Zehui
    Kim, Dong In
    Shen, Xuemin
    Poor, H. Vincent
    IEEE NETWORK, 2024, 38 (03): : 178 - 186
  • [2] Enabling AI-Generated Content Services in Wireless Edge Networks
    Du, Hongyang
    Li, Zonghang
    Niyato, Dusit
    Kang, Jiawen
    Xiong, Zehui
    Shen, Xuemin
    Kim, Dong In
    IEEE WIRELESS COMMUNICATIONS, 2024, 31 (03) : 226 - 234
  • [3] Online AI-Generated Content Request Scheduling with Deep Reinforcement Learning
    Feng, Chenglong
    Zheng, Ying
    Xu, Yuedong
    IEEE INFOCOM 2024-IEEE CONFERENCE ON COMPUTER COMMUNICATIONS WORKSHOPS, INFOCOM WKSHPS 2024, 2024,
  • [4] Dynamic Reinforcement Learning based Scheduling for Energy-Efficient Edge-Enabled LoRaWAN
    Mhatre, Jui
    Lee, Ahyoung
    2022 IEEE INTERNATIONAL PERFORMANCE, COMPUTING, AND COMMUNICATIONS CONFERENCE, IPCCC, 2022,
  • [5] QoE-Driven Content-Centric Caching With Deep Reinforcement Learning in Edge-Enabled IoT
    He, Xiaoming
    Wang, Kun
    Xu, Wenyao
    IEEE COMPUTATIONAL INTELLIGENCE MAGAZINE, 2019, 14 (04) : 12 - 20
  • [6] AI-Generated Network Design: A Diffusion Model-Based Learning Approach
    Huang, Yudong
    Xu, Minrui
    Zhang, Xinyuan
    Niyato, Dusit
    Xiong, Zehui
    Wang, Shuo
    Huang, Tao
    IEEE NETWORK, 2024, 38 (03): : 202 - 209
  • [7] Joint Caching and Computing Service Placement for Edge-Enabled IoT Based on Deep Reinforcement Learning
    Chen, Yan
    Sun, Yanjing
    Yang, Bin
    Taleb, Tarik
    IEEE INTERNET OF THINGS JOURNAL, 2022, 9 (19) : 19501 - 19514
  • [8] Edge-Enabled Two-Stage Scheduling Based on Deep Reinforcement Learning for Internet of Everything
    Zhou, Xiaokang
    Liang, Wei
    Yan, Ke
    Li, Weimin
    Wang, Kevin I-Kai
    Ma, Jianhua
    Jin, Qun
    IEEE INTERNET OF THINGS JOURNAL, 2023, 10 (04) : 3295 - 3304
  • [9] Evading Watermark based Detection of AI-Generated Content
    Jiang, Zhengyuan
    Zhang, Jinghuai
    Gong, Neil Zhenqiang
    PROCEEDINGS OF THE 2023 ACM SIGSAC CONFERENCE ON COMPUTER AND COMMUNICATIONS SECURITY, CCS 2023, 2023, : 1168 - 1181
  • [10] AI-Generated Content-Based Edge Learning for Fast and Efficient Few-Shot Defect Detection in IIoT
    Li, Siyuan
    Lin, Xi
    Xu, Wenchao
    Li, Jianhua
    IEEE TRANSACTIONS ON SERVICES COMPUTING, 2024, 17 (06) : 3140 - 3153