HyperAIHyperAI

Command Palette

Search for a command to run...

Raisonnement par mémoire : embeddings de graphes de connaissances par plus proches voisins

Peng Wang Xin Xie Xiaohan Wang Ningyu Zhang

Résumé

Les approches précédentes d’embedding des graphes de connaissances consistent généralement à représenter les entités par des vecteurs et à utiliser des fonctions de score pour prédire les entités cibles, mais elles peinent souvent à raisonner sur des entités rares ou émergentes non vues auparavant. Dans cet article, nous proposons KNN-KGE, une nouvelle méthode d’embedding des graphes de connaissances basée sur des modèles linguistiques pré-entraînés, qui réalise une interpolation linéaire de la distribution des entités avec leurs k plus proches voisins. Nous calculons les voisins les plus proches en fonction de la distance dans l’espace d’embedding des entités, à partir du stock de connaissances. Notre approche permet aux entités rares ou émergentes d’être mémorisées de manière explicite, plutôt que de manière implicite dans les paramètres du modèle. Les résultats expérimentaux montrent que notre méthode améliore les performances en prédiction de liens inductive et transductive, tout en offrant de meilleures performances dans des scénarios à faible ressource, même avec seulement quelques triples, ce qui pourrait faciliter le raisonnement grâce à une mémoire explicite. Le code est disponible à l’adresse suivante : https://github.com/zjunlp/KNN-KG.


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