CoolFace
Modelpublic

EnergyFM/energy-tspulse

sourceHugging Faceapache-2.0updated 2mo agoView on Hugging Face
0likes105downloads
Model Card

Energy-TSPulse

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.

Pretraining Dataset

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:

  • β€”76,217 residential and commercial buildings
  • β€”1.26 billion hourly electricity consumption readings
  • β€”Multiple countries and climate zones
  • β€”Diverse building types and operational patterns

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.

Available Variants

VariantDescription
mainPretrained on the full EnergyBench dataset containing real-world residential and commercial buildings. Recommended checkpoint for most use cases.
512-commPretrained exclusively on commercial building energy consumption data.
512-resPretrained exclusively on residential building energy consumption data.

Supported Tasks

Anomaly Detection

Energy-TSPulse detects point and contextual anomalies in smart meter data using joint time–frequency representations.

Classification

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.


πŸ”΅ Energy-TSPulse (Anomaly Detection / Reconstruction)

python
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)

🟣 Energy-TSPulse (Classification)

python
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)

πŸ“š Energy-TSPulse Recipies

NotebookOpen in Colab
⚑ Zero-Shot Energy Anomaly Detection with Energy-TSPulse<a target="blank" href="https://colab.research.google.com/github/energyfms/notebooks/blob/main/energytspulseanomalydetection_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/energytspulseanomalydetection_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/energytspulseclassificationfinetuning.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>

Resources

Energy Benchmark Leaderboard

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.


Limitations and Intended Use

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.


Citation

If you use EnergyFM in your work, please cite:

bibtex
@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}
}