English

Hyper-parameter Optimization for Wireless Network Traffic Prediction Models with A Novel Meta-Learning Framework

Networking and Internet Architecture 2025-05-06 v3

Abstract

This paper proposes a novel meta-learning based hyper-parameter optimization framework for wireless network traffic prediction (NTP) models. The primary objective is to accumulate and leverage the acquired hyper-parameter optimization experience, enabling the rapid determination of optimal hyper-parameters for new tasks. In this paper, an attention-based deep neural network (ADNN) is employed as the base-learner to address specific NTP tasks. The meta-learner is an innovative framework that integrates meta-learning with the k-nearest neighbor algorithm (KNN), genetic algorithm (GA), and gated residual network (GRN). Specifically, KNN is utilized to identify a set of candidate hyper-parameter selection strategies for a new task, which then serves as the initial population for GA, while a GRN-based chromosome screening module accelerates the validation of offspring chromosomes, ultimately determining the optimal hyper-parameters. Experimental results demonstrate that, compared to traditional methods such as Bayesian optimization (BO), GA, and particle swarm optimization (PSO), the proposed framework determines optimal hyper-parameters more rapidly, significantly reduces optimization time, and enhances the performance of the base-learner. It achieves an optimal balance between optimization efficiency and prediction accuracy.

Keywords

Cite

@article{arxiv.2409.14535,
  title  = {Hyper-parameter Optimization for Wireless Network Traffic Prediction Models with A Novel Meta-Learning Framework},
  author = {Liangzhi Wang and Jie Zhang and Yuan Gao and Jiliang Zhang and Guiyi Wei and Haibo Zhou and Bin Zhuge and Zitian Zhang},
  journal= {arXiv preprint arXiv:2409.14535},
  year   = {2025}
}

Comments

This work has been submitted to the IEEE for possible publication

R2 v1 2026-06-28T18:53:01.010Z