arXiv · 2506.07871
Can Hessian-Based Insights Support Fault Diagnosis in Attention-based Models?
Abstract
As attention-based deep learning models scale in size and complexity, diagnosing their faults becomes increasingly challenging. In this work, we conduct an empirical study to evaluate the potential of Hessian-based analysis for diagnosing faults in attention-based models. Specifically, we use Hessian-derived insights to identify fragile regions (via curvature analysis) and parameter interdependencies (via parameter interaction analysis) within attention mechanisms. Through experiments on three diverse models (HAN, 3D-CNN, DistilBERT), we show that Hessian-based metrics can localize instability and pinpoint fault sources more effectively than gradients alone. Our empirical findings suggest that these metrics could significantly improve fault diagnosis in complex neural architectures, potentially improving software debugging practices.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Sigma Jahan, Mohammad Masudur Rahman. 2025-06-09. Can Hessian-Based Insights Support Fault Diagnosis in Attention-based Models?. https://arxiv.org/abs/2506.07871
Cite the original work for its findings. Save a collection to share your selection of sources.