HyperAIHyperAI

Command Palette

Search for a command to run...

Minimisation de l'entropie graphique pour la classification de nœuds semi-supervisée

Yi Luo Guangchun Luo Ke Qin Aiguo Chen

Résumé

Les classificateurs de nœuds doivent réduire de manière globale les erreurs de prédiction, les ressources nécessaires à l’entraînement et la latence d’inférence dans l’industrie. Toutefois, la plupart des réseaux de neurones sur graphes (GNN) se concentrent uniquement sur une ou deux de ces dimensions. Les aspects compromis deviennent alors les maillons faibles du processus, freinant leur déploiement pratique pour des tâches à l’échelle industrielle. Ce travail propose une nouvelle méthode d’apprentissage semi-supervisé, nommée Minimisation de l’Entropie du Graphe (GEM), capable de résoudre simultanément ces trois problèmes. Grâce à une aggregation à un saut exploitant un grand nombre de nœuds non catégorisés, GEM atteint une précision de prédiction comparable à celle des GNN utilisant un passage de messages sur deux ou plusieurs sauts. De plus, GEM peut être décomposée pour permettre un entraînement stochastique par mini-batches d’échantillons d’arêtes indépendants, offrant ainsi un échantillonnage extrêmement rapide et une utilisation efficace de l’espace mémoire. Bien que son aggregation à un saut soit déjà plus rapide en inférence que celle des GNN profonds, GEM peut être encore accélérée de manière extrême grâce à la dérivation d’un classificateur sans saut via une distillation en temps réel des connaissances. Ainsi, GEM s’impose comme un choix pratique pour les services soumis à des contraintes de latence et sensibles aux erreurs, fonctionnant sur des matériels à ressources limitées. Le code est disponible à l’adresse suivante : https://github.com/cf020031308/GEM.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp