MCPcopy Index your code
hub / github.com/Anoise/WTFlib

github.com/Anoise/WTFlib @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
562 symbols 1,256 edges 51 files 73 documented · 13%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

Awesome Wireless Traffic Forecasting Library

Python 3.10 PyTorch 1.12 License CC BY-NC-SA

Intoroduction

  • This is an open-source library for wireless traffic forecasting (WTF-Lib), especially for deep wireless traffic analysis.
  • In this library, we provide a variety of state-of-the-art short- and long-term wireless traffic forecasting methods.
  • We provide a neat code base to evaluate advanced deep time series models or develop your model, which given a strong benchmark.

Leaderboard for WTF

Till November 2023, the top three models for short- and long-term WTF tasks are:

| Model

Ranking | Long-term

Forecasting | Short-term

Forecasting | Runtime | | ---------------- |---------------------------------------------------| ------------------------------------------------------------ |----------------------------------------------------| | 🥇 1st | PSLinear | PSLinear | DLinear | | 🥈 2nd | PatchTST | PatchTST | PSLinear | | 🥉 3rd | GWNet | FEDformer | STID |

Compared models of this leaderboard. ☑ means that their codes have already been included in this repo.

  • [x] PSLinear - Progressively Supervision based on Label Decomposition: Towards Long-Term Wireless Traffic Forecasting on Large-Scale Graphs. arXiv 2025, Code.

  • [x] PatchTST - A Time Series is Worth 64 Words: Long-term Forecasting with Transformers. [ICLR 2023] [Code].

  • [x] DLinear - Are Transformers Effective for Time Series Forecasting? [AAAI 2023] [Code].

  • [x] GWNet - Graph WaveNet for Deep Spatial-Temporal Graph Modeling. [arXiv 2019][Code].

  • [x] STID - Spatial-Temporal Identity: A Simple yet Effective Baseline for Multivariate Time Series Forecasting. [arXiv 2023][Code].

  • [x] DyDgcrn - Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solution. [[IEEE TKDE 2021]] (https://ieeexplore.ieee.org/document/9625773) [Code].

  • [x] Mvstgn - MVSTGN: A Multi-View Spatial-Temporal Graph Network for Cellular Traffic Prediction. [[IEEE TMC 2021]] (https://ieeexplore.ieee.org/document/9625773) [Code].

  • [x] Periodformer - Does Long-Term Series Forecasting Need Complex Attention and Extra Long Inputs? [[arXiv 2023]] (https://arxiv.org/abs/2306.05035) [Code].

  • [x] FEDformer - FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecasting. [ICML 2022] [Code].

  • [x] Pyraformer - Pyraformer: Low-complexity Pyramidal Attention for Long-range Time Series Modeling and Forecasting. [ICLR 2022] [Code].

  • [x] Autoformer - Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting. [NeurIPS 2021] [Code].

  • [x] Informer - Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecasting. [AAAI 2021] [Code].

  • [x] Reformer - Reformer: The Efficient Transformer. [ICLR 2020] [Code].

  • [x] STCNet - Deep Transfer Learning for Intelligent Cellular Traffic Prediction Based on Cross-Domain Big Data. [IEEE JSAC 2019] [Code]

  • [x] Transformer - Attention is All You Need. [NeurIPS 2017] [Code].

Long-term Wireless Traffic Forecasting (LWTF) Benchmark

Methods Metric C2TM Milano-All CBSD
4 5 6 7 8 24 36 48 72 24 36 48 72
PSLinear-MS MSE 9.057 9.161 9.298 9.431 9.389 0.633 0.775 0.907 1.184 1.641 1.772 1.822 1.955
MAE 0.171 0.174 0.171 0.317 0.169 0.249 0.269 0.285 0.301 0.625 0.644 0.646 0.660
PSLinear-STL MSE 9.084 9.183 9.328 9.447 9.400 0.708 0.870 1.062 1.366 1.644 1.759 1.821 1.965
MAE 0.174 0.172 0.172 0.169 0.167 0.275 0.294 0.315 0.343 0.633 0.649 0.653 0.669
PatchTST MSE 9.226 9.34 9.509 9.504 9.561 0.662 0.832 0.929 1.222 1.87 2.036 2.113 2.084
MAE 0.195 0.197 0.201 0.182 0.192 0.254 0.273 0.282 0.302 0.706 0.736 0.735 0.697
STID MSE 9.213

Core symbols most depended-on inside this repo

abs
called by 24
LDPS_Graph/layers/utils.py
cuda
called by 12
LDPS_Graph/layers/utils.py
cpu
called by 10
LDPS_Graph/layers/utils.py
encoder
called by 8
LDPS_Graph/models/dydcrnn_arch/dydcrnn.py
step
called by 7
LDPS_Graph/models/dydgcrn_arch/dydgcrn.py
decoder
called by 6
LDPS_Graph/models/dydcrnn_arch/dydcrnn.py
time_features
called by 5
LDPS_Graph/utils/timefeatures.py
transform
called by 4
LDPS_Graph/utils/tools.py

Shape

Method 369
Class 146
Function 47

Languages

Python100%

Modules by API surface

LDPS_Graph/models/Mvstgn.py74 symbols
LDPS_Graph/layers/PatchTST_backbone.py34 symbols
LDPS_Graph/layers/Embed.py30 symbols
LDPS_Graph/layers/utils.py29 symbols
LDPS_Graph/layers/Autoformer_EncDec.py24 symbols
LDPS_Graph/data_provider/data_loader.py24 symbols
LDPS_Graph/utils/timefeatures.py22 symbols
LDPS_Graph/layers/MultiWaveletCorrelation.py20 symbols
LDPS_Graph/models/dydcrnn_arch/dcrnn_utils.py19 symbols
LDPS_Graph/models/Stat_models.py19 symbols
LDPS_Graph/models/dydcrnn_arch/dydcrnn.py15 symbols
LDPS_Graph/models/dcrnn_arch/dcrnn_arch.py15 symbols

For agents

$ claude mcp add WTFlib \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact