Laguerre Geometry for Interpreting Large Language Models
Cet article introduit un cadre de géométrie de Laguerre qui redéfinit les concepts des LLM comme des régions spatiales plutôt que comme des points, permettant le développement d'outils sans entraînement tels que la Lentille Géométrique et l'Autoencodeur de Laguerre pour visualiser, mesurer et manipuler avec précision les trajectoires de raisonnement internes et les structures hiérarchiques des modèles de transformeurs.
Article original sous licence CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Ceci est une explication générée par l'IA de l'article ci-dessous. Elle n'a pas été rédigée ni approuvée par les auteurs. Pour une précision technique, consultez l'article original. Lire la clause de non-responsabilité complète
Imaginez un Grand Modèle de Langage (LLM) comme une immense bibliothèque à plusieurs étages où chaque livre représente un mot possible que le modèle pourrait dire ensuite. Pendant longtemps, les scientifiques ont pensé que chaque « idée » ou « concept » à l'intérieur de cette bibliothèque n'était qu'une simple épingle minuscule plantée sur la carte de la pièce. Si vous vouliez trouver le concept de « Chien », il vous suffirait de chercher une épingle spécifique.
Mais ce nouvel article suggère que cette vision est trop simpliste. Au lieu d'une simple épingle, les auteurs proposent qu'un concept est en réalité une pièce entière avec des murs, un sol et un plafond. Ils appellent cela une « cellule de Laguerre-Voronoi ». Imaginez cela comme un quartier dans une ville : « Chien » n'est pas seulement une maison ; c'est tout le pâté de maisons où vivent toutes les idées liées aux chiens, délimité par des clôtures invisibles qui le séparent de « Chat » ou d'« Oiseau ».
La carte qui a tout changé
Les auteurs, Chunwei Ma et Russell D. Wolfinger, ont utilisé une branche des mathématiques appelée Géométrie de Laguerre pour dessiner ces cartes. Dans cette géométrie, chaque concept a un centre (comme une place de village) et un « poids » spécial (comme un rayon d'influence).
Voici le grand tournant qu'ils ont découvert : le poids importe.
Si vous regardez seulement la proximité de deux mots l'un par rapport à l'autre (la distance), vous ne pouvez pas faire la différence entre un « chiot » (un type de chien) et un « mammifère » (une catégorie qui inclut les chiens). Ils pourraient être également proches sur une carte simple. Mais quand on ajoute le « poids » de la géométrie de Laguerre, la carte révèle une hiérarchie. Elle montre que « chiot » est à l'intérieur de la chambre « chien », et que « chien » est à l'intérieur de la chambre « mammifère ». Les mathématiques prouvent que le modèle organise naturellement ces idées comme des poupées russes, et que l'on ne peut voir cet emboîtement qu'en utilisant ce prisme géométrique spécifique.
Jeter un œil à l'intérieur de la machine
L'article s'attaque également à un mystère : à quoi pense le modèle au milieu d'une phrase, avant qu'il ne finisse de taper ?
Imaginez que vous demandiez au modèle : « La capitale de la France est... »
- Ancienne vision (Logit Lens) : Les scientifiques pensaient que le modèle dérivait lentement vers la réponse « Paris » dès le premier mot.
- Nouvelle vision (Geometric Lens) : Les auteurs ont construit un outil appelé « Geometric Lens » pour observer le « processus de pensée » du modèle en temps réel. Ils ont découvert que le chemin interne du modèle est beaucoup plus chaotique et intéressant.
Dans leurs expériences, ils ont observé les pensées cachées du modèle voyager à travers différents « quartiers » (régions) tandis qu'il traitait la phrase.
- Au début, le modèle considère de nombreuses options.
- Ensuite, il semble saisir le mot factuellement correct « Paris » (sans l'espace) dans les couches intermédiaires.
- Enfin, dans les toutes dernières étapes, il se corrige pour produire le mot grammaticalement correct « _Paris » (avec l'espace) avant de donner la réponse.
Le Geometric Lens est le seul outil qui a pu capturer tout ce voyage. Les autres outils soit se sont bloqués trop tôt, soit ont manqué les étapes intermédiaires.
Quand le modèle est confus
Les auteurs ont également testé ce qui se passe lorsque vous essayez de piéger le modèle. Ils lui ont donné une consigne du type : « Vous êtes dans un monde fictif où Marseille et Lyon ont échangé leurs noms. Le Louvre est à... »
Normalement, le modèle est perturbé par l'histoire fictive et pourrait dire « Lyon » ou « Marseille » au lieu de la vraie réponse, « Paris ».
Cependant, lorsque les auteurs ont utilisé leur Geometric Lens pour observer les pensées du modèle avant qu'il ne finisse de taper (spécifiquement à la couche 20), ils ont découvert quelque chose d'incroyable : le modèle connaissait la vérité depuis le début. Même si la réponse finale était fausse à cause du piège, le Geometric Lens a montré que la « pensée » interne du modèle avait brièvement visité la bonne réponse, « Paris », avant d'être emportée par l'histoire fictive.
Ce que cela signifie (et ce que cela ne signifie pas)
L'article suggère que nous devrions cesser de considérer les concepts comme des points uniques et commencer à les voir comme des régions entières avec des formes et des poids spécifiques. Ils ont prouvé mathématiquement que ces régions existent et sont linéairement séparables (ce qui signifie que l'on peut tracer une ligne droite entre deux quartiers de concepts différents).
Ils ont mesuré cela en utilisant de vrais modèles comme Phi-2 et Gemma-2. Dans des tests comparant différentes manières de définir les concepts, leur méthode « Laguerre » a permis de distinguer plus facilement les « Animaux » des « Plantes » que les méthodes précédentes. Par exemple, sur le modèle Phi-2, leur méthode présentait un taux d'erreur de 0,111 pour distinguer les animaux des plantes, ce qui est meilleur que le taux d'erreur de 0,121 de l'ancienne méthode de « Géométrie Catégorielle ».
Ils ont également démontré que leur « Geometric Lens » est plus efficace pour repérer le véritable raisonnement du modèle lors de consignes confuses que les autres outils. Dans le test d'interférence sur « Paris », le Geometric Lens a permis de récupérer le jeton (token) correct avec une précision de 0,56 sur Phi-2, battant les 0,55 de Patchscopes et les 0,05 du Logit Lens.
L'essentiel
Cet article ne prétend pas avoir résolu tous les mystères de l'IA. Il ne nous dit pas encore comment contrôler le modèle (c'est l'objet de travaux futurs). Mais il offre une nouvelle façon plus claire de voir ce qui se passe à l'intérieur de la boîte noire. Il suggère que l'intelligence du modèle n'est pas simplement une ligne droite de A vers B ; c'est un voyage complexe à travers une ville de quartiers, où le modèle visite différentes idées, les pèse, et change parfois d'avis avant de parler. Et grâce à la géométrie de Laguerre, nous avons enfin une carte qui nous montre tout le quartier, et pas seulement un point isolé.
Noyé(e) sous les articles dans votre domaine ?
Recevez des digests quotidiens des articles les plus récents correspondant à vos mots-clés de recherche — avec des résumés techniques, dans votre langue.