rakaarao17/edge_inference_benchmark_release

0

stars

4

commits

Python

primary language

Feb 4, 2026

updated

README

Edge Inference Benchmark

Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark

License DOI

A benchmark framework evaluating accuracy-efficiency tradeoffs of time series foundation models on consumer GPUs (6GB VRAM).

πŸ“¦ Models Benchmarked

ModelParametersTypeExpected VRAM
XGBoost~1MTree-based0 GB (CPU)
ARIMA-Statistical0 GB (CPU)
DLinear98.5KEfficient DL~0.1 GB
PatchTST1.2MEfficient DL~0.5 GB
Chronos-Tiny8MFoundation~1 GB
Chronos-Mini20MFoundation~2 GB
Chronos-Small46MFoundation~3 GB
Moirai-Small14MFoundation~1.5 GB

πŸ“Š Datasets

DatasetDomainFrequencySeries
ETTh1EnergyHourly7
ExchangeFinanceDaily8
TrafficTransportHourly862
WeatherClimate10-min21

πŸš€ Quick Start

# 1. Create virtual environment
python -m venv venv
source venv/bin/activate  # Linux/Mac
# OR: venv\Scripts\activate  # Windows

# 2. Install PyTorch with CUDA first
pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cu118

# 3. Install dependencies
pip install -r requirements.txt

# 4. Download datasets
python -m src.data.download

# 5. Run benchmark
python experiments/run_baselines.py --model xgboost --dataset etth1 --runs 5
python experiments/run_foundation.py --model chronos --size small --dataset etth1

πŸ“ Project Structure

edge_inference_benchmark/
β”œβ”€β”€ configs/           # Model and dataset configurations
β”œβ”€β”€ src/               # Source code
β”‚   β”œβ”€β”€ data/          # Data loading
β”‚   β”œβ”€β”€ models/        # Model implementations
β”‚   β”œβ”€β”€ evaluation/    # Metrics and efficiency
β”‚   └── visualization/ # Plotting utilities
β”œβ”€β”€ experiments/       # Experiment scripts
β”œβ”€β”€ results/
β”‚   β”œβ”€β”€ figures/              # Visualization outputs
β”‚   └── benchmark_summary.json # All metrics summary
└── requirements.txt

πŸ“ˆ Key Results

Accuracy vs Latency Tradeoff

Pareto Frontier

Cold-Start Performance

Cold Start

πŸ–₯️ Hardware

Tested on:

  • GPU: NVIDIA RTX 3050 Mobile (6GB VRAM)
  • CUDA: 11.8
  • Python: 3.12

πŸ“ Citation

@article{author2026edge,
  title={Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark},
  author={Author, Your Name},
  journal={Transactions on Machine Learning Research},
  year={2026}
}

πŸ“„ License

MIT License - see LICENSE for details.

Contributors

rakaarao17

4 commits

rakaarao17/edge_inference_benchmark_release

0

stars

4

commits

Python

primary language

Feb 4, 2026

updated

README

Edge Inference Benchmark

Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark

License DOI

A benchmark framework evaluating accuracy-efficiency tradeoffs of time series foundation models on consumer GPUs (6GB VRAM).

πŸ“¦ Models Benchmarked

ModelParametersTypeExpected VRAM
XGBoost~1MTree-based0 GB (CPU)
ARIMA-Statistical0 GB (CPU)
DLinear98.5KEfficient DL~0.1 GB
PatchTST1.2MEfficient DL~0.5 GB
Chronos-Tiny8MFoundation~1 GB
Chronos-Mini20MFoundation~2 GB
Chronos-Small46MFoundation~3 GB
Moirai-Small14MFoundation~1.5 GB

πŸ“Š Datasets

DatasetDomainFrequencySeries
ETTh1EnergyHourly7
ExchangeFinanceDaily8
TrafficTransportHourly862
WeatherClimate10-min21

πŸš€ Quick Start

# 1. Create virtual environment
python -m venv venv
source venv/bin/activate  # Linux/Mac
# OR: venv\Scripts\activate  # Windows

# 2. Install PyTorch with CUDA first
pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cu118

# 3. Install dependencies
pip install -r requirements.txt

# 4. Download datasets
python -m src.data.download

# 5. Run benchmark
python experiments/run_baselines.py --model xgboost --dataset etth1 --runs 5
python experiments/run_foundation.py --model chronos --size small --dataset etth1

πŸ“ Project Structure

edge_inference_benchmark/
β”œβ”€β”€ configs/           # Model and dataset configurations
β”œβ”€β”€ src/               # Source code
β”‚   β”œβ”€β”€ data/          # Data loading
β”‚   β”œβ”€β”€ models/        # Model implementations
β”‚   β”œβ”€β”€ evaluation/    # Metrics and efficiency
β”‚   └── visualization/ # Plotting utilities
β”œβ”€β”€ experiments/       # Experiment scripts
β”œβ”€β”€ results/
β”‚   β”œβ”€β”€ figures/              # Visualization outputs
β”‚   └── benchmark_summary.json # All metrics summary
└── requirements.txt

πŸ“ˆ Key Results

Accuracy vs Latency Tradeoff

Pareto Frontier

Cold-Start Performance

Cold Start

πŸ–₯️ Hardware

Tested on:

  • GPU: NVIDIA RTX 3050 Mobile (6GB VRAM)
  • CUDA: 11.8
  • Python: 3.12

πŸ“ Citation

@article{author2026edge,
  title={Time Series Foundation Models on Consumer GPUs: A 6GB VRAM Benchmark},
  author={Author, Your Name},
  journal={Transactions on Machine Learning Research},
  year={2026}
}

πŸ“„ License

MIT License - see LICENSE for details.

Contributors

rakaarao17

4 commits

Languages

Python

100.0%