Graph Set Transformer : une architecture pour l'apprentissage sur des ensembles de graphes

Le Graph Set Transformer (GST) est une architecture neuronale conçue pour traiter des ensembles de graphes, où chaque prédiction dépend à la fois du contexte global de l'ensemble et de la structure locale. Contrairement à DeepSets et SetTransformer, GST entrelace propagation de features au niveau des nœuds et modélisation contextuelle inter-graphes à chaque couche via un mécanisme de gating. Évalué sur des benchmarks synthétiques et réels (identification de centres de réaction, prédiction de rendement, classification d'images), GST surpasse les baselines à budget paramétrique équivalent.