Training Graph Neural Networks Subject to a Tight Lipschitz Constraint - Inria - Institut national de recherche en sciences et technologies du numérique
Article Dans Une Revue Transactions on Machine Learning Research Journal Année : 2024

Training Graph Neural Networks Subject to a Tight Lipschitz Constraint

Résumé

We propose a strategy for training a wide range of graph neural networks (GNNs) under tight Lipschitz bound constraints. Specifically, by leveraging graph spectral theory, we derive computationally tractable expressions of a tight Lipschitz constant. This allows us to propose a constrained-optimization approach to control the constant, ensuring robustness to adversarial perturbations. Unlike the existing methods for controlling the Lipschitz constant, our approach reduces the size of the handled matrices by a factor equal to the square of the number of nodes in the graph. We employ a stochastic projected subgradient algorithm, which operates in a block-coordinate manner, with the projection step performed via an accelerated iterative proximal algorithm. We focus on defending against attacks that perturb features while keeping the topology of the graph constant. This contrasts with most of the existing defenses, which tackle perturbations of the graph structure. We report experiments on various datasets in the context of node classification tasks, showing the effectiveness of our constrained GNN model.
Fichier principal
Vignette du fichier
TMLR_revised_to_submit_HAL.pdf (1.05 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04828157 , version 1 (09-12-2024)

Licence

Identifiants

  • HAL Id : hal-04828157 , version 1

Citer

Simona Juvina, Ana-Antonia Neacșu, Jean-Christophe Pesquet, Burileanu Corneliu, Jérôme Rony, et al.. Training Graph Neural Networks Subject to a Tight Lipschitz Constraint. Transactions on Machine Learning Research Journal, 2024. ⟨hal-04828157⟩
0 Consultations
0 Téléchargements

Partager

More