系统工程与电子技术 ›› 2024, Vol. 46 ›› Issue (8): 2721-2729.doi: 10.12305/j.issn.1001-506X.2024.08.19

• 系统工程 • 上一篇    

基于GPR模型的用户量预测优化方法

刘学浩1,2, 刘文学1, 杨超三1, 祝文晶1,2, 宋玉1,2, 李金海1,2,*   

  1. 1. 中国科学院微电子研究所通信与信息工程研发中心, 北京 100029
    2. 中国科学院大学集成电路学院, 北京 100049
  • 收稿日期:2024-01-10 出版日期:2024-07-25 发布日期:2024-08-07
  • 通讯作者: 李金海
  • 作者简介:刘学浩 (1999—), 男, 硕士研究生, 主要研究方向为通信与卫星导航技术、信息与信号处理
    刘文学 (1985—), 男, 高级工程师, 博士, 主要研究方向为卫星导航相关的软硬件设计、信号处理算法
    杨超三 (1989—), 男, 助理研究员, 博士, 主要研究方向为卫星通信与自组网通信
    祝文晶 (1999—), 女, 博士研究生, 主要研究方向为智能无线通信
    宋玉 (1999—), 女, 博士研究生, 主要研究方向为智能无线通信、自组网通信
    李金海 (1978—), 男, 高级工程师, 博士, 主要研究方向为卫星导航、宽带无线通信
  • 基金资助:
    地球观测与导航(2022YFB3903900);中国科学院“西部之光”交叉团队项目(xbzg-zdsys-202120)

Optimization method of user quantity prediction based on GPR model

Xuehao LIU1,2, Wenxue LIU1, Chaosan YANG1, Wenjing ZHU1,2, Yu SONG1,2, Jinhai LI1,2,*   

  1. 1. Communication and Information Engineering Research and Development Center, Institute of Microelectronics of the Chinese Academy of Sciences, Beijing 100029, China
    2. School of Integrated Circuits, University of Chinese Academy of Sciences, Beijing 100049, China
  • Received:2024-01-10 Online:2024-07-25 Published:2024-08-07
  • Contact: Jinhai LI

摘要:

高斯过程回归(Gaussian process regression, GPR)是一种基于高斯过程的非参数化贝叶斯回归方法, 其可以灵活适应不同类型数据, 用于建模和预测数据之间的复杂关系, 具有拟合能力强、泛化能力好等特点。针对海量用户场景下用户量实时预测问题, 提出一种基于GPR的用户量预测优化方法。在滑动窗口方法处理数据的基础上, 选择合适的核函数, 基于k折交叉验证得到最佳超参数组合以实现GPR模型训练, 完成在线用户量的实时预测并进行性能评估。实验结果表明, 相比于采用训练集中输出数据方差的50%作为信号噪声估计量的传统方案, 所提方法具有较高的预测准确度, 并且在测试集均方根误差(root mean square, RMS)、平均绝对误差(mean absolute error, MAE)、平均偏差(mean bias error, MBE)和决定系数R2这4个评估指标方面均有提升, 其中MBE至少提升了43.3%。

关键词: 高斯过程回归, 用户量预测, 滑动窗口, 交叉验证, 超参数优化

Abstract:

Gaussian process regression (GPR) is a non-parametric Bayesian regression method based on Gaussian processes. It is flexible in adapting to different types of data, and it is used to model and predict complex relationships between different types of data. It has strong fitting capabilities and good generalization abilities. A user quantity prediction optimization method based on GPR is proposed to tackle the problem of real-time user quantity prediction in the context of massive user scenario. Building upon the sliding window method for data processing, the method selects a suitable kernel function and uses k-fold cross-validation to determine the optimal hyperparameter combination for training the GPR model, which enables the real-time prediction of online user quantity. Finally, the performance of the model is evaluated. The experimental results demonstrate that compared with the traditional approach that uses half of the variance of the output data in the training set as the signal noise estimator, the proposed method has higher prediction accuracy and improvements in the four following evaluation metrics of root mean square (RMS), mean absolute error (MAE), mean bias error (MBE) and determination coefficient R2 on the test set. Specifically, the MBE shows an improvement of at least 43.3%.

Key words: Gaussian process regression (GPR), user quantity prediction, sliding window, cross-validation, hyperparameter optimization

中图分类号: