mlsanityData to 14 Oct 2026

Lightweight Transformer Models for On-Device Fault Detection: A Benchmark Study on Resource-Constrained Deployment

arXiv 2606.24173 ↗First posted 23 Jun 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

On-device fault detection enables real-time diagnostics without cloud dependency, but deploying machine learning models on resource-constrained hardware demands careful tradeoffs between accuracy, latency, and model size. We present a benchmark comparing traditional ML methods (Random Forest, XGBoost, SVM, Logistic Regression) against lightweight transformer architectures (DistilBERT, TinyBERT-6L, TinyBERT-4L, MobileBERT) for binary fault detection across three public datasets: NASA C-MAPSS turbofan degradation, SECOM semiconductor manufacturing, and UCI AI4I 2020 predictive maintenance. We evaluate classification performance (F1-score, AUC), model size, and CPU inference latency, and further assess INT8 dynamic quantization and a two-stage adaptive inference pipeline. Our results reveal that on well-separated sensor data (C-MAPSS), lightweight transformers match traditional ML at 87.8% F1 but at 100x the model size and 9000x the latency. TinyBERT-4L emerges as the most deployment-friendly transformer at 55 MB and 18 ms CPU latency. INT8 quantization reduces size by 25% while preserving 86.9% F1. Our adaptive pipeline, routing 97.9% of predictions through a quantized triage model and only 2.1% to a larger expert, achieves 87.6% F1 at 19.5 ms average latency. On severely imbalanced datasets (SECOM, UCI-PM), both traditional and transformer methods struggle significantly, highlighting fundamental limitations of current approaches for extreme class imbalance in fault detection. All code is publicly available.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 9 of its 20 references that are in the map. First posted on arXiv 23 Jun 2026.

Builds on

9 of 20 references.

References · 9

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  3. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  4. 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,676
  5. 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,593
  6. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
  7. 2020Efficient Transformers: A Surveycited by 790
  8. 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 482
  9. 2021Learning Graph Structures with Transformer for Multivariate Time Series Anomaly Detection in IoTcited by 104

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Lightweight Transformer Architectures for Edge Devices in Real-Time Applications
  2. 2025Energy-Efficient Transformer Inference: Optimization Strategies for Time Series Classification
  3. 2026A SISA-based Machine Unlearning Framework for Power Transformer Inter-Turn Short-Circuit Fault Localization
  4. 2025Automating Versatile Time-Series Analysis with Tiny Transformers on Embedded FPGAs
  5. 2025Dissecting Transformers: A CLEAR Perspective towards Green AI
  6. 2021hls4ml: An Open-Source Codesign Workflow to Empower Scientific Low-Power Machine Learning Devices
  7. 2025TWEO: Transformers Without Extreme Outliers Enables FP8 Training And Quantization For Dummies
  8. 2025EdgeFlex-Transformer: Transformer Inference for Edge Devices