Enhancing 6G mmWave Beam Prediction in V2I with Class Imbalance Mitigation

The increase in the use of autonomous vehicles has motivated a paradigm shift in the transportation domain as it redefines the boundaries of urban mobility by augmenting safety measures. This research paper explores an innovative approach to enhance 6G millimeter-wave (mmWave) beam prediction for vehicle-to-infrastructure (V2I) communications by using generative adversarial networks (GANs). By generating synthetic data samples effectively and balancing the real-world dataset, we improve the accuracy of beam prediction models significantly. Our proposed method of training random forests on synthetic data (RFGAN) to predict beam indices provides the solution for imbalanced class issues and significantly improves the predictive performance of mmWave beam selection, contributing to more reliable and efficient V2I communications. This work also performs comparative analysis with state-of-the-art models in top-K evaluation metrics, average power loss, and overhead savings related to adapting to the new approach.