Parameter Efficient Hybrid Transformer (PEHT) for Network Traffic Prediction via Dynamic Urban Congestion Integration

TL;DR

PEHT结合LoRA优化Transformer,利用城市交通与拥堵信息实现网络流量高精度预测。

cs.LG 🔴 高级 2026-06-27 68 次浏览
Abdolazim Rezaei Mehdi Sookhak Mahboobeh Haghparast
深度学习 Transformer 城市交通 网络预测 参数效率

核心发现

方法论

本文提出的PEHT架构通过将网络通信特征与城市迁移和拥堵信息分离,利用低秩适应(LoRA)技术对Transformer编码器进行参数高效微调,显著减少训练参数量。模型采用多模态融合策略,将外部迁移和拥堵特征注入解码器,增强预测能力。具体流程包括:• 采用基于图的虚拟基站聚类算法,将原始网格数据映射为具有代表性的虚拟基站,缓解数据稀疏问题;• 构建多模态特征向量,包括历史流量、时间特征、静态和动态网络特征,并进行嵌入编码;• 在Transformer编码器中引入LoRA机制,通过低秩矩阵逼近权重更新,减少参数数量超过90%;• 设计多模态融合策略,将迁移和拥堵信息在解码阶段融合,确保时间因果性;• 构建基于Encoder-Decoder的预测模型,利用多头自注意力机制捕获长距离依赖,最终输出未来网络流量。实验中,模型在Telecom Italia Milan数据集及多种合成拥堵场景下均优于SOTA方法,表现为RMSE、MAE和$R^2$指标的显著提升。

关键结果

  • 在Milan数据集上,PEHT在SMS、Call和Internet流量预测中,RMSE分别为18.42、34.85和122.10,较最优基线ST2T分别降低了14.6%、11.9%和11.4%,显示出优异的预测精度;MAE指标也显著优于对比模型,达到了12.15、20.60和68.45,整体模型的$R^2$值超过0.95,验证了模型在捕获复杂交通动态方面的有效性。
  • 在五个合成数据集中的不同交通场景下,PEHT模型通过引入LoRA和多模态融合,均实现了RMSE最低(最低值为8.85),且模型在不同拥堵和迁移条件下表现出良好的鲁棒性和泛化能力,特别是在高拥堵和边缘区域场景中,预测误差明显低于传统模型。
  • 消融实验显示,加入LoRA机制后,参数数量大幅减少(从19.1百万降至不到14万),但预测性能几乎不受影响,验证了参数高效性和模型的可扩展性。此外,融合迁移和拥堵信息显著提升模型对突发交通变化的响应能力,增强了模型的实际应用价值。

研究意义

该研究突破了网络流量预测中参数瓶颈的限制,结合城市迁移与拥堵信息,显著提升了预测精度,为智能网络资源调度提供了理论基础和技术支撑。其创新的多模态融合与参数高效调优,为未来大规模城市网络管理和智能调度系统奠定了基础。特别是在5G及未来6G网络中,动态环境下的高精度预测将极大改善网络性能和用户体验,推动智慧城市的智能化发展。

技术贡献

技术上,本文首次将LoRA引入Transformer编码器,用于高维空间的参数高效微调,解决了传统Transformer在大规模城市网络数据中的参数膨胀问题。模型采用多模态融合策略,将外部迁移和拥堵信息有效融入预测流程,确保时间因果性。通过虚拟基站聚类算法缓解数据稀疏,提升模型鲁棒性。整体架构结合Transformer的长距离依赖捕获能力与参数高效调优,为网络流量预测提供了新思路。

新颖性

本研究的创新点在于:首先,将LoRA技术首次引入城市网络流量预测Transformer中,实现参数大幅压缩;其次,提出多模态融合策略,将迁移和拥堵信息在解码阶段融合,避免信息泄露且增强模型理解能力;再次,通过虚拟基站聚类缓解数据稀疏问题,提升模型的泛化能力。这些创新共同推动了参数高效、外部信息丰富的网络预测模型的发展,填补了现有方法在参数效率和多源信息融合方面的空白。

局限性

  • 模型在极端突发事件或极端交通变化场景下的预测能力仍有待提升,尤其是在极端拥堵或突发事故发生时,模型的鲁棒性可能不足;
  • 当前模型主要依赖静态迁移和拥堵特征,未来应考虑引入更多实时动态信息(如天气、特殊事件)以增强预测能力;
  • 参数压缩虽显著,但在极大规模部署中仍需优化模型推理速度和硬件适应性,未来需结合边缘计算技术进行优化。

未来方向

未来将探索多模态信息的动态融合机制,结合实时传感器和视频监控数据,提升模型对突发事件的响应能力。同时,将引入联邦学习框架,实现跨区域数据协作,增强模型的泛化能力。此外,计划结合强化学习优化资源调度策略,实现网络性能的动态自适应,推动智慧城市网络的智能化升级。

AI 总览摘要

城市网络流量预测一直是通信领域的核心难题,随着5G和未来6G技术的发展,城市环境中的网络需求变得更加复杂多变。传统的预测模型如基于统计的方法或单一深度学习架构,难以同时兼顾高精度和参数效率,尤其在面对复杂的城市迁移和拥堵动态时表现不足。

为解决这一难题,本文提出了参数高效的混合Transformer架构PEHT,结合了低秩适应(LoRA)技术,显著减少了模型参数数量,同时保持甚至提升了预测准确性。该模型通过将输入特征分为网络通信特征和城市迁移、拥堵特征两大类,利用多模态融合策略在解码阶段注入外部信息,增强模型对城市动态的理解能力。

在技术实现上,PEHT采用了虚拟基站聚类算法,将原始高维空间中的网格数据映射为代表性更强的虚拟基站,缓解数据稀疏问题。模型中的Transformer编码器引入LoRA机制,通过低秩矩阵逼近,减少参数量超过90%,极大提升了模型的训练效率和可扩展性。多头自注意力机制确保模型能捕获长距离依赖关系,融合策略则确保信息流的时间因果性。

在实验部分,模型在Telecom Italia Milan数据集上取得了优异表现,预测误差明显低于现有最优模型。例如,在多源流量预测中,RMSE降低至18.42,较最优基线下降14.6%。在多种合成拥堵场景中,PEHT表现出强大的鲁棒性和泛化能力,验证了其在实际城市网络中的应用潜力。这些结果表明,结合外部迁移和拥堵信息的多模态Transformer架构,是未来城市网络智能预测的重要方向。

该研究不仅在模型参数效率方面实现突破,也为城市交通与网络管理的深度融合提供了技术方案。未来,模型将结合更多实时动态信息,优化边缘计算部署,并通过联邦学习实现区域间数据协作,推动智慧城市的网络智能化升级。尽管如此,模型在极端突发事件中的表现仍需改进,未来工作将聚焦于增强模型鲁棒性和实时性,推动城市网络的智能调度与管理。

深度解读

原文摘要

Accurate network traffic prediction is a critical element for efficient resource allocation in dynamic urban cellular networks. However, prediction remains challenging because network demand is influenced by complex mobility patterns, congestion dynamics, and heterogeneous user behavior. This paper introduces the Parameter-Efficient Hybrid Transformer (PEHT), a network traffic prediction framework that integrates urban mobility and congestion information into a Transformer-based architecture. PEHT separates primary network communication features from secondary urban mobility features and incorporates Low-Rank Adaptation (LoRA) into the Transformer encoder to reduce the number of trainable parameters while maintaining high predictive accuracy. A multimodal fusion strategy then injects external mobility and congestion features into the decoder to improve traffic forecasting. Experiments on the Telecom Italia Milan dataset and multiple synthetic congestion scenarios show that PEHT outperforms state-of-the-art baselines in terms of RMSE, MAE, and $R^2$. The implementation is available in the GitHub repository.

cs.LG cs.AI