Interpretable attention-based prototype network for UAV fault diagnosis under small sample conditions

Shuang Liang, Jinsong Yu, Diyin Tang and Xu Ke

Reliability Engineering and System Safety, 2026, vol. 265, issue PB

Abstract: Fault diagnosis for Unmanned Aerial Vehicles (UAVs) is a typical †small data†problem due to the scarcity of failure data. However, existing few-shot methods face challenges in performing UAV fault diagnosis with explicit explanations in real-world flight scenarios. To address this challenge, we propose an interpretable attention-based prototype network (IA-PN) for both UAV few-shot fault classification and unknown fault identification, utilizing flight parameters measured by multiple sensors, while also automatically explaining the diagnostic results. The IA-PN framework constructs an interpretable network structure within the prototype network to mitigate overfitting with small samples. To enhance the model’s interpretability, a novel interpretable attention mechanism is employed to mine and locate the most crucial features in both the temporal and parameter domains. This mechanism provides local explanations in the form of parameter importance and identifies anomalous flight parameters for each fault category, offering insights into fault characteristics. Under unsupervised conditions, the proposed interpretable diagnostic framework can identify unknown faults based on relevant known fault data and provide potential explanations. Experimental results on two real UAV flight datasets, along with visualizations of interpretations in the parameter space, demonstrate that our method is both effective and interpretable in UAV fault diagnosis under small sample conditions.

Keywords: Fault diagnosis; Interpretable attention mechanism; Prototype network; Small samples; Unmanned aerial vehicle (search for similar items in EconPapers)
Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832025008014
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:265:y:2026:i:pb:s0951832025008014

DOI: 10.1016/j.ress.2025.111601

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().