Article Dans Une Revue Transactions on Machine Learning Research Journal Année : 2025

Sparser, Better, Faster, Stronger: Sparsity Detection for Efficient Automatic Differentiation

Résumé

From implicit differentiation to probabilistic modeling, Jacobian and Hessian matrices have many potential use cases in Machine Learning (ML), but they are viewed as computationally prohibitive. Fortunately, these matrices often exhibit sparsity, which can be leveraged to speed up the process of Automatic Differentiation (AD). This paper presents advances in sparsity detection, previously the performance bottleneck of Automatic Sparse Differentiation (ASD). Our implementation of sparsity detection is based on operator overloading, able to detect both local and global sparsity patterns, and supports flexible index set representations. It is fully automatic and requires no modification of user code, making it compatible with existing ML codebases. Most importantly, it is highly performant, unlocking Jacobians and Hessians at scales where they were considered too expensive to compute. On real-world problems from scientific ML, graph neural networks and optimization, we show significant speed-ups of up to three orders of magnitude. Notably, using our sparsity detection system, ASD outperforms standard AD for one-off computations, without amortization of either sparsity detection or matrix coloring.

Fichier principal
Vignette du fichier
sparser_better_faster_stronger.pdf (866.57 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence

Dates et versions

hal-05116348 , version 1 (17-06-2025)

Licence

Identifiants

Citer

Adrian Hill, Guillaume Dalle. Sparser, Better, Faster, Stronger: Sparsity Detection for Efficient Automatic Differentiation. Transactions on Machine Learning Research Journal, 2025. ⟨hal-05116348⟩
55 Consultations
153 Téléchargements

Altmetric

Partager

  • More