Skip to main navigation Skip to search Skip to main content

Deep Reinforcement Learning-Based Block Coordinate Descent for Downlink Weighted Sum-Rate Maximization on AI-Native Wireless Networks

  • Siya Chen*
  • , Chee Wei Tan
  • , H. Vincent Poor
  • *Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

Abstract

This paper introduces a deep reinforcement learning-based block coordinate descent (DRL-based BCD) algorithm to address the nonconvex weighted sum-rate maximization (WSRM) problem with a total power constraint. Firstly, we present an efficient block coordinate descent (BCD) method to solve the problem. While this method may not always achieve globally optimal solutions, it provides a pathway for integrating machine learning and domain-specific techniques with theoretical analysis of the underlying convexity of the subproblems. We then integrate deep reinforcement learning (DRL) techniques into the BCD method and propose the DRL-based BCD algorithm. This approach combines the data-driven learning capability of machine learning techniques with the navigational and decision-making characteristics of the optimization-theoretic-based BCD method. This combination significantly improves the algorithm’s performance by reducing its sensitivity to initial points and mitigating the risk of entrapment in local optima. The primary advantages of the proposed DRL-based BCD algorithm lie in its ability to adhere to the constraints of the WSRM problem and significantly enhance accuracy, potentially achieving the exact optimal solution. Moreover, unlike many pure machine-learning approaches, the DRL-based BCD algorithm capitalizes on the underlying theoretical analysis of the WSRM problem’s structure. This enables it to be easily trained and computationally efficient while maintaining a level of interpretability. Moreover, the DRL-based BCD framework demonstrates strong extensibility and can effectively be applied to other scenarios, such as joint beamforming for sum rate maximization, as demonstrated in this paper. Through numerical experiments, the DRL-based BCD algorithm demonstrates substantial advantages in effectiveness, efficiency, robustness, and interpretability for maximizing sum rates, which also provides valuable potential for designing resource-constrained AI-native wireless optimization strategies in next-generation wireless networks. © 2025 IEEE.
Original languageEnglish
Pages (from-to)3658-3674
Number of pages17
JournalIEEE Transactions on Wireless Communications
Volume25
Online published10 Sept 2025
DOIs
Publication statusPublished - 2025
Externally publishedYes

Funding

This work is supported in part by an Innovation Grant from Princeton NextG and in part by the Singapore Ministry of Education Academic Research Fund MOE-T2EP20224-0009.

Research Keywords

  • block coordinate descent
  • nonlinear Perron-Frobenius theory
  • reinforcement learning
  • Weighted sum-rate maximization

Fingerprint

Dive into the research topics of 'Deep Reinforcement Learning-Based Block Coordinate Descent for Downlink Weighted Sum-Rate Maximization on AI-Native Wireless Networks'. Together they form a unique fingerprint.

Cite this