Downloads · 30 days
31
8% of all-time downloads
EnergyFM/energy-tspulse
energy-tspulse is a machine learning model from EnergyFM. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as apache-2.0.
Energy-TSPulse is a domain-specific Time Series Foundation Model (TSFM) for energy meter data analytics, pretrained on large-scale real-world smart meter data from the EnergyBench corpus. Built upon IBM Research's TSP…
Downloads · 30 days
31
8% of all-time downloads
All-time downloads
369
Public
Parameters
168K
9.3 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors688 KB · 98%
From the Hugging Face model README
Energy-TSPulse is a domain-specific Time Series Foundation Model (TSFM) for energy meter data analytics, pretrained on large-scale real-world smart meter data from the EnergyBench corpus. Built upon IBM Research's TSPulse architecture, it learns rich temporal representations from diverse residential and commercial electricity consumption patterns.
The pretrained model is designed for zero-shot and transfer learning across heterogeneous buildings, regions, and operational contexts, while remaining lightweight and computationally efficient. It can be adapted to a variety of downstream energy analytics tasks, including anomaly detection, appliance classification, and missing value imputation.
EnergyFM is pretrained on EnergyBench, a large-scale real-world smart meter dataset available on Hugging Face:
👉 https://huggingface.co/datasets/ai-iot/EnergyBench
The dataset consists of:
The scale and diversity of EnergyBench enable EnergyFM to learn daily, weekly, and seasonal consumption patterns and to generalize robustly to unseen buildings and regions.
| Variant | Description |
|---|---|
| main | Pretrained on the full EnergyBench dataset containing real-world residential and commercial buildings. Recommended checkpoint for most use cases. |
| 512-comm | Pretrained exclusively on commercial building energy consumption data. |
| 512-res | Pretrained exclusively on residential building energy consumption data. |
Energy-TSPulse detects point and contextual anomalies in smart meter data using joint time–frequency representations.
Energy-TSPulse supports appliance usage classification using transfer learning on low-frequency smart meter data, achieving competitive or superior performance compared to strong feature-based classifiers.
import torch
from tsfm_public.models.tspulse import TSPulseForReconstruction
device = "cuda" if torch.cuda.is_available() else "cpu"
model = TSPulseForReconstruction.from_pretrained(
"EnergyFM/energy-tspulse",
revision="main", # Loads Energy-TSPulse weights
num_input_channels=1
).to(device)
import torch
from tsfm_public.models.tspulse import TSPulseForClassification
device = "cuda" if torch.cuda.is_available() else "cpu"
model = TSPulseForClassification.from_pretrained(
"EnergyFM/energy-tspulse",
revision="main" # Change to 512-res or 512-comm to access commercial or residential specific variant
).to(device)
| Notebook | Open in Colab |
|---|---|
| ⚡ Zero-Shot Energy Anomaly Detection with Energy-TSPulse | <a target="_blank" href="https://colab.research.google.com/github/energyfms/notebooks/blob/main/energy_tspulse_anomaly_detection_zeroshot.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
| ⚡ Fine-Tuning Energy Anomaly Detection with Energy-TSPulse | <a target="_blank" href="https://colab.research.google.com/github/energyfms/notebooks/blob/main/energy_tspulse_anomaly_detection_finetuning.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
| ⚡ Appliance Classification with Energy-TSPulse | <a target="_blank" href="https://colab.research.google.com/github/energyfms/notebooks/blob/main/energy_tspulse_classification_finetuning.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> |
GitHub Repository 👉 https://github.com/energyfms/
Pretraining Dataset (EnergyBench) 👉 https://huggingface.co/datasets/ai-iot/EnergyBench
To compare EnergyFM against other state-of-the-art Time Series Foundation Models for energy analytics tasks, please visit our public benchmark leaderboard:
👉 Energy Benchmark Leaderboard
https://huggingface.co/spaces/EnergyFM/Leaderboard
The leaderboard provides standardized evaluations across forecasting, anomaly detection, and classification tasks, enabling direct comparison under consistent experimental settings.
EnergyFM is intended for energy meter analytics and has been pretrained on electricity consumption data. Performance may degrade when applied to unrelated domains or data with significantly different temporal characteristics.
If you use EnergyFM in your work, please cite:
@inproceedings{energyfm2026,
author = {Arjunan, Pandarasamy and Srivastava, Naman and Kumar, Kajeeth and Jati, Arindam and Ekambaram, Vijay and Dayama, Pankaj},
title = {EnergyFM: Pretrained Models for Energy Meter Data Analytics},
year = {2026},
url = {https://doi.org/10.1145/3744255.3798119},
doi = {10.1145/3744255.3798119},
booktitle = {Proceedings of the 17th ACM International Conference on Future and Sustainable Energy Systems},
pages = {556–568},
series = {E-Energy '26}
}