HyperAIHyperAI

Command Palette

Search for a command to run...

Vers l'entraînement de réseaux neuronaux graphiques à milliards de paramètres pour les simulations atomiques

Anuroop Sriram; Abhishek Das; Brandon M. Wood; Siddharth Goyal; C. Lawrence Zitnick

Résumé

Les progrès récents dans les Réseaux Neuronaux sur Graphes (GNNs) pour la modélisation des simulations atomiques ont le potentiel de révolutionner la découverte de catalyseurs, une étape clé pour réaliser les percées énergétiques nécessaires à la lutte contre le changement climatique. Cependant, les GNNs qui se sont avérés les plus efficaces pour cette tâche sont intensifs en mémoire car ils modélisent des interactions d'ordre supérieur dans les graphes, telles que celles entre des triplets ou quadruplets d'atomes, ce qui rend difficile l'évolutivité de ces modèles. Dans cet article, nous présentons le Parallélisme sur Graphes, une méthode permettant de distribuer les graphes d'entrée sur plusieurs GPU, nous permettant ainsi d'entraîner des GNNs très volumineux avec des centaines de millions ou des milliards de paramètres. Nous évaluons empiriquement notre méthode en augmentant le nombre de paramètres des modèles DimeNet++ et GemNet récemment proposés d'un facteur supérieur à un ordre de grandeur. Sur le grand ensemble de données Open Catalyst 2020 (OC20), ces modèles parallélisés sur graphes entraînent des améliorations relatives de 1) 15 % sur la métrique d'erreur absolue moyenne (MAE) des forces pour la tâche S2EF et 2) 21 % sur la métrique AFbT pour la tâche IS2RS, établissant ainsi de nouveaux résultats d'état de l'art.


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