mlsanityData to 14 Oct 2026

Hierarchical Fault Detection and Diagnosis for Transformer Architectures

arXiv 2604.28118 ↗First posted 30 Apr 2026cs.SE cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Transformers now underpin critical AI systems across industry and research. Yet their faults can silently alter model behavior without runtime errors, and existing techniques offer little support for tracing these failures to their component and root cause. Such faults evade detection because loss and numerical values stay normal, and the visible symptom rarely identifies the component responsible. We present DEFault++, a hierarchical learning-based technique that first detects a fault, then identifies the affected component, and finally the cause within it, helping developers effectively debug transformer models. DEFault++ organizes component-level runtime measurements with a Fault Propagation Graph (FPG), a structural prior over the architecture's dependency paths, and reports the evidence behind each diagnosis. To train and evaluate it, we construct DEFault-bench, a benchmark of 5,556 labeled runs from mutation testing across seven models, nine tasks, and both encoder and decoder architectures. DEFault++ improves fault detection over four prior techniques, reaching an F1 of 0.826--0.909, and in a developer study with 21 participants, it raises repair accuracy from 57.1% to 83.3%. These results show that transformer fault diagnosis benefits from component-level measurements and architecture-aware reasoning rather than model-level behavior alone, and DEFault-bench provides a foundation for further research on transformer fault diagnosis.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 53 of its 98 references that are in the map. First posted on arXiv 30 Apr 2026.

Builds on

53 of 98 references.

References · 53

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  4. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  5. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  6. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  7. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  8. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
  9. 2016Layer Normalizationcited by 5,669
  10. 2017On Calibration of Modern Neural Networkscited by 4,960
  11. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  12. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
38 more
  1. 2016Pointer Sentinel Mixture Modelscited by 3,096
  2. 2017Prototypical Networks for Few-shot Learningcited by 3,015
  3. 2020Supervised Contrastive Learningcited by 2,829
  4. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  5. 2018Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learningcited by 1,102
  6. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
  7. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  8. 2020On Layer Normalization in the Transformer Architecturecited by 922
  9. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
  10. 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 784
  11. 2017A systematic study of the class imbalance problem in convolutional neural networkscited by 513
  12. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
  13. 2020On the Stability of Fine-tuning BERT: Misconceptions, Explanations, and Strong Baselinescited by 255
  14. 2017DeepTest: Automated Testing of Deep-Neural-Network-driven Autonomous Carscited by 251
  15. 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 161
  16. 2024Keyformer: KV Cache Reduction through Key Tokens Selection for Efficient Generative Inferencecited by 126
  17. 2019Taxonomy of Real Faults in Deep Learning Systemscited by 90
  18. 2018DeepMutation: Mutation Testing of Deep Learning Systemscited by 67
  19. 2019A Comprehensive Study on Deep Learning Bug Characteristicscited by 65
  20. 2021Attention, please! A survey of Neural Attention Models in Deep Learningcited by 51
  21. 2021DeepLocalize: Fault Localization for Deep Neural Networkscited by 38
  22. 2019DeepFault: Fault Localization for Deep Neural Networkscited by 30
  23. 2020Piecewise linear activations substantially shape the loss surfaces of neural networkscited by 19
  24. 2021DeepDiagnosis: Automatically Diagnosing Faults and Recommending Actionable Fixes in Deep Learning Programscited by 19
  25. 2022DeepFD: Automated Fault Diagnosis and Localization for Deep Learning Programscited by 16
  26. 2024ATTNChecker: Highly-Optimized Fault Tolerant Attention for Large Language Model Trainingcited by 12
  27. 2023Keep the Faith: Faithful Explanations in Convolutional Neural Networks for Case-Based Reasoningcited by 10
  28. 2021Automatic Fault Detection for Deep Learning Programs Using Graph Transformationscited by 9
  29. 2024A Multi-Agent Approach to Fault Localization via Graph-Based Retrieval and Reflexioncited by 9
  30. 2024From Today's Code to Tomorrow's Symphony: The AI Transformation of Developer's Routine by 2030cited by 8
  31. 2025FT-Transformer: Resilient and Reliable Transformer with End-to-End Fault Tolerant Attentioncited by 8
  32. 2023DeepSeer: Interactive RNN Explanation and Debugging via State Abstractioncited by 6
  33. 2026When Agents Fail: A Comprehensive Study of Bugs in LLM Agents with Automated Labelingcited by 6
  34. 2023Mutation Testing of Deep Reinforcement Learning Based on Real Faultscited by 4
  35. 2024Fault Localization in Deep Learning-based Software: A System-level Approachcited by 4
  36. 2025Improved Detection and Diagnosis of Faults in Deep Neural Networks Using Hierarchical and Explainable Classificationcited by 3
  37. 2025Where's the Bug? Attention Probing for Scalable Fault Localizationcited by 3
  38. 2022Testing Feedforward Neural Networks Training Programscited by 2

50 most cited of 53 All 53 (JSON)

Built on by

Not cited yet.