Skip to main navigation Skip to search Skip to main content

Hybrid CNN-Transformer Architecture for RealTime UAV Fault Classification

  • Runqiang Yu*
  • , Haolin Jia
  • , Aotian Song
  • , Jian Hu
  • , Hui Wu
  • *Corresponding author for this work
  • School of Economics and Management, Harbin Institute of Technology Weihai

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

This paper proposes a hybrid CNN-Transformer architecture-based approach for unmanned aerial vehicle (UAV) fault classification, aiming to achieve efficient and real-time fault classification through time-series data. Addressing the limitations of traditional convolutional neural networks (CNNs) in capturing global dependencies and their deployment on edge devices, we design a model that integrates an attention mechanism. This model leverages CNNs to extract local features, employs a Transformer encoder to model long-term dependencies, and incorporates a lightweight design to optimize inference latency. Experiments are conducted using the 'RflyMAD' dataset. The results demonstrate that the proposed model achieves a classification accuracy of 98.74% on the test set, a macro-average F1-score of 97.89%, and an average inference latency of only 1.82 milliseconds, significantly outperforming conventional methods. This approach exhibits notable advantages in both performance improvement and realtime capability, offering reliable support for the safe operation of UAVs.

Original languageEnglish
Title of host publicationProceedings of 2025 International Conference on Intelligent Operation and Maintenance of Equipment, ICEIOM 2025
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages614-623
Number of pages10
ISBN (Electronic)9798331512347
DOIs
StatePublished - 2025
Externally publishedYes
Event2025 International Conference on Intelligent Operation and Maintenance of Equipment, ICEIOM 2025 - Urumqi, China
Duration: 1 Aug 20254 Aug 2025

Publication series

NameProceedings of 2025 International Conference on Intelligent Operation and Maintenance of Equipment, ICEIOM 2025

Conference

Conference2025 International Conference on Intelligent Operation and Maintenance of Equipment, ICEIOM 2025
Country/TerritoryChina
CityUrumqi
Period1/08/254/08/25

Keywords

  • attention mechanism
  • fault classification
  • hybrid CNN-Transformer
  • real-time inference

Fingerprint

Dive into the research topics of 'Hybrid CNN-Transformer Architecture for RealTime UAV Fault Classification'. Together they form a unique fingerprint.

Cite this