Scaled and Inter-token Relation Enhanced Transformer for Sample-restricted Residential NILM

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Rahman, Minhajur, Arafat, Yasir
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909418989289472
author Rahman, Minhajur
Arafat, Yasir
author_facet Rahman, Minhajur
Arafat, Yasir
contents Transformers have demonstrated exceptional performance across various domains due to their self-attention mechanism, which captures complex relationships in data. However, training on smaller datasets poses challenges, as standard attention mechanisms can over-smooth attention scores and overly prioritize intra-token relationships, reducing the capture of meaningful inter-token dependencies critical for tasks like Non-Intrusive Load Monitoring (NILM). To address this, we propose a novel transformer architecture with two key innovations: inter-token relation enhancement and dynamic temperature tuning. The inter-token relation enhancement mechanism removes diagonal entries in the similarity matrix to improve attention focus on inter-token relations. The dynamic temperature tuning mechanism, a learnable parameter, adapts attention sharpness during training, preventing over-smoothing and enhancing sensitivity to token relationships. We validate our method on the REDD dataset and show that it outperforms the original transformer and state-of-the-art models by 10-15\% in F1 score across various appliance types, demonstrating its efficacy for training on smaller datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2410_12861
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Scaled and Inter-token Relation Enhanced Transformer for Sample-restricted Residential NILM
Rahman, Minhajur
Arafat, Yasir
Computation and Language
Artificial Intelligence
Transformers have demonstrated exceptional performance across various domains due to their self-attention mechanism, which captures complex relationships in data. However, training on smaller datasets poses challenges, as standard attention mechanisms can over-smooth attention scores and overly prioritize intra-token relationships, reducing the capture of meaningful inter-token dependencies critical for tasks like Non-Intrusive Load Monitoring (NILM). To address this, we propose a novel transformer architecture with two key innovations: inter-token relation enhancement and dynamic temperature tuning. The inter-token relation enhancement mechanism removes diagonal entries in the similarity matrix to improve attention focus on inter-token relations. The dynamic temperature tuning mechanism, a learnable parameter, adapts attention sharpness during training, preventing over-smoothing and enhancing sensitivity to token relationships. We validate our method on the REDD dataset and show that it outperforms the original transformer and state-of-the-art models by 10-15\% in F1 score across various appliance types, demonstrating its efficacy for training on smaller datasets.
title Scaled and Inter-token Relation Enhanced Transformer for Sample-restricted Residential NILM
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2410.12861