HyperAIHyperAI

Command Palette

Search for a command to run...

Passage de messages équivariant pour la prédiction de propriétés tensorielles et de spectres moléculaires

Kristof T. Schütt Oliver T. Unke Michael Gastegger

Résumé

Les réseaux neuronaux à passage de messages sont devenus une méthode de choix pour l'apprentissage sur les graphes, en particulier pour la prédiction de propriétés chimiques et l'accélération des études de dynamique moléculaire. Bien qu'ils se généralisent facilement à de grandes bases d'apprentissage, les approches précédentes se sont révélées moins efficaces en termes de données que les méthodes à noyaux. Nous identifions les limitations des représentations invariantes comme une cause majeure de cette faiblesse, et étendons la formulation à passage de messages aux représentations rotationnellement équivalentes. À partir de cette base, nous proposons le réseau neuronal d'interaction d'atomes polarisables (PaiNN), qui améliore les performances sur les benchmarks courants de molécules par rapport aux réseaux antérieurs, tout en réduisant la taille du modèle et le temps d'inférence. Nous exploitons les représentations atomiques équivalentes obtenues par PaiNN pour prédire des propriétés tensorielles. Enfin, nous appliquons cette approche à la simulation de spectres moléculaires, obtenant des accélérations de 4 à 5 ordres de grandeur par rapport à la référence de la théorie des structures électroniques.


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