Command Palette
Search for a command to run...
Une procédure persistante de Weisfeiler–Lehman pour la classification des graphes
{Karsten Borgwardt Bastian Rieck Christian Bock}

Résumé
Le noyau de graphe Weisfeiler–Lehman présente des performances compétitives dans de nombreuses tâches de classification de graphes. Toutefois, ses caractéristiques basées sur les sous-arbres ne permettent pas de capturer les composantes connexes ni les cycles, des caractéristiques topologiques essentielles pour décrire les graphes. Pour extraire de telles informations, nous exploitons les informations de labels propagées aux nœuds et transformons les graphes non pondérés en graphes métriques. Cette transformation nous permet d’enrichir les caractéristiques basées sur les sous-arbres avec des informations topologiques obtenues grâce à l’homologie persistante, un concept issu de l’analyse topologique des données. Notre méthode, que nous formelisons comme une généralisation des caractéristiques de sous-arbres Weisfeiler–Lehman, présente une précision de classification favorable, et ses améliorations en termes de performance prédictive sont principalement dues à l’intégration d’informations relatives aux cycles.
Benchmarks
| Benchmark | Méthodologie | Métriques |
|---|---|---|
| graph-classification-on-mutag | P-WL-C | Mean Accuracy: 90.51 |
| graph-classification-on-proteins | P-WL-UC | Accuracy: 75.36% |
| graph-property-prediction-on-ogbg-molhiv | P-WL | Ext. data: No Number of params: 4600000 Test ROC-AUC: 0.8039 ± 0.0040 Validation ROC-AUC: 0.8279 ± 0.0059 |
Construire l'IA avec l'IA
De l'idée au lancement — accélérez votre développement IA avec du co-codage IA gratuit, un environnement prêt à l'emploi et les meilleurs prix GPU.