Search papers, labs, and topics across Lattice.
This paper introduces Auto-PGD, an AutoML approach that optimizes wireless beamforming and waveforms by converting the proximal gradient descent (PGD) algorithm into a deep neural network with learnable parameters and a hybrid layer for gradient transformation. AutoGluon with a tree-structured parzen estimator (TPE) is used for hyperparameter optimization across a broad search space, including network depth and optimizer type. The resulting Auto-PGD achieves 98.8% of the spectral efficiency of a 200-iteration PGD solver with only five unrolled layers and 100 training samples, demonstrating a significant reduction in training data and inference cost.
Achieve near-optimal waveform optimization with 98.8% spectral efficiency using a 5-layer, AutoML-tuned unrolled proximal gradient descent network trained on just 100 samples.
This study explores the combination of automated machine learning (AutoML) with model-based deep unfolding (DU) for optimizing wireless beamforming and waveforms. We convert the iterative proximal gradient descent (PGD) algorithm into a deep neural network, wherein the parameters of each layer are learned instead of being predetermined. Additionally, we enhance the architecture by incorporating a hybrid layer that performs a learnable linear gradient transformation prior to the proximal projection. By utilizing AutoGluon with a tree-structured parzen estimator (TPE) for hyperparameter optimization (HPO) across an expanded search space, which includes network depth, step-size initialization, optimizer, learning rate scheduler, layer type, and post-gradient activation, the proposed auto-unrolled PGD (Auto-PGD) achieves 98.8% of the spectral efficiency of a traditional 200-iteration PGD solver using only five unrolled layers, while requiring only 100 training samples. We also address a gradient normalization issue to ensure consistent performance during training and evaluation, and we illustrate per-layer sum-rate logging as a tool for transparency. These contributions highlight a notable reduction in the amount of training data and inference cost required, while maintaining high interpretability compared to conventional black-box architectures.
