Exploring Compositional Generalization (in COGS/ReCOGS_pos) by Transformers using Restricted Access Sequence Processing (RASP)
Bruns
Humans understand new combinations of words encountered if they are combinations of words recognized from different contexts, an ability called Compositional Generalization. The COGS benchmark (Kim and Linzen, 2020) arXiv:2010.05465 reports 0% accuracy for Transformer models on some structural generalizations. We use (Weiss et al., 2021) arXiv:2106.06981's Restricted Access Sequence Processing (RASP), a Transformer-equivalent programming language, to demonstrate that a Transformer Encoder-Decoder can perform COGS and the semantically equivalent ReCOGS_pos (Wu et al., 2024) arXiv:2303.13716 systematically and compositionally: Our RASP models attain near perfect scores on structural generalization splits on COGS (exact match) and ReCOGS_pos (semantic exact match). Our RASP models show the (Re)COGS tasks do not require a hierarchical or tree-structured solution (contrary to (Kim and Linzen, 2020) arXiv:2010.05465, (Yao and Koller, 2022) arXiv:2210.13050, (Murty et al., 2022) arXiv:2211.01288, (Liu et al., 2021) arXiv:2107.06516): we use word-level tokens with an "embedding" layer that tags with possible part of speech, applying just once per encoder pass 19 attention-head compatible flat pattern-matching rules (easily identified with specific training examples), shown using grammar coverage (Zeller et al., 2023) to cover the non-recursive aspects of the input grammar, plus masking out prepositional phrases ("pp noun") and/or sentential complements (cp) when recognizing grammar patterns and extracting nouns related to the main verb in the sentence, and output the next logical form (LF) token (repeating until the LF is complete). The models do not apply recursive, tree-structured rules like "np_det pp np -> np_pp -> np", but score near perfect semantic and string exact match on both COGS and ReCOGS pp recursion, cp recursion using the decoder loop.
academic
Exploration de la Généralisation Compositionnelle (dans COGS/ReCOGS_pos) par les Transformers utilisant le Traitement de Séquences à Accès Restreint (RASP)
Les humains peuvent comprendre de nouvelles combinaisons de vocabulaire identifiées dans des contextes différents, capacité appelée généralisation compositionnelle. L'évaluation COGS rapporte que les modèles Transformer atteignent une précision de 0% sur certaines généralisations structurelles. Cet article utilise le langage RASP (Traitement de Séquences à Accès Restreint) pour démontrer que les encodeurs-décodeurs Transformer peuvent exécuter systématiquement et compositionnellement les tâches COGS et ReCOGS_pos sémantiquement équivalentes : le modèle RASP atteint des scores quasi-parfaits sur les divisions de généralisation structurelle. La recherche montre que les tâches (Re)COGS ne nécessitent pas de solutions hiérarchiques ou arborescentes, mais utilisent plutôt 19 règles de correspondance de motifs plats compatibles avec les têtes d'attention, en masquant les syntagmes prépositionnels et les propositions pour identifier les motifs grammaticaux.
Le problème fondamental que cette recherche vise à résoudre est la limitation des capacités des modèles Transformer dans les tâches de généralisation compositionnelle, en particulier les mauvaises performances sur l'évaluation COGS (Défi de Généralisation Compositionnelle basé sur l'Interprétation Sémantique).
Signification théorique : La généralisation compositionnelle est une capacité fondamentale de la compréhension du langage humain ; comprendre comment les réseaux de neurones réalisent cette capacité est crucial pour faire progresser la compréhension du langage par l'IA
Signification pratique : La précision quasi-nulle des modèles Transformer actuels sur les tâches de généralisation structurelle indique des limitations fondamentales qui nécessitent des solutions
Limitation des réseaux peu profonds : L'encodeur-décodeur à 2 couches utilisé par Kim et Linzen (2020) montre des performances extrêmement faibles sur la généralisation structurelle
Hypothèse hiérarchique erronée : Les recherches existantes supposent que des structures arborescentes ou des représentations hiérarchiques sont nécessaires pour résoudre les tâches COGS
Inefficacité de la profondeur : Petty et al. (2024) constatent que même avec 32 couches, le Transformer n'améliore pas la généralisation structurelle COGS
L'auteur s'inspire de Zhou et al. (2023) utilisant RASP pour analyser les capacités de généralisation des Transformers, tentant de démontrer par preuve constructive que les encodeurs-décodeurs Transformer peuvent théoriquement résoudre les tâches COGS et d'analyser les raisons des échecs des modèles existants.
Preuve constructive : Utilisation du langage RASP pour démontrer que les encodeurs-décodeurs Transformer peuvent théoriquement résoudre systématiquement les tâches COGS et ReCOGS_pos
Solution plate : Proposition d'une solution non-hiérarchique basée sur 19 règles de correspondance de motifs plats, sans nécessiter de règles arborescentes récursives
Analyse des erreurs : Prédiction et vérification des motifs d'erreurs spécifiques des Transformers de base par la théorie des « erreurs d'attraction »
Percée de performance : Le modèle RASP atteint 99,89% de correspondance exacte de chaîne sur COGS et 99,63% de correspondance sémantique exacte sur ReCOGS_pos
Nouvelle division de généralisation : Découverte et vérification d'une nouvelle division de généralisation difficile « v_dat_p2_pp_moved_to_recipient »
Prévention des noms de syntagmes prépositionnels « attirant » des relations grammaticales incorrectes par le mécanisme de masquage :
Erreur : The cake on the plate burned → theme(burn, plate) # Erreur d'attraction
Correct : The cake on the plate burned → theme(burn, cake) # Après masquage
Les structures récursives sont traitées par la boucle de décodeur, supportant l'imbrication arbitraire de syntagmes prépositionnels et de propositions.
L'augmentation du nombre de couches Transformer (3-4 couches) n'améliore pas la performance sur obj_pp_to_subj_pp, confirmant les découvertes de Petty et al. (2024).
Évaluation COGS : Kim et Linzen (2020) proposent l'évaluation, rapportant une précision de généralisation structurelle quasi-nulle pour les Transformers
Amélioration ReCOGS : Wu et al. (2024) obtiennent une précision non-nulle mais toujours faible par correspondance sémantique exacte
Approches hiérarchiques : Liu et al. (2021), Weißenhorn et al. (2022) utilisent des structures arborescentes explicites pour obtenir des performances élevées
Faisabilité théorique : Les Transformers peuvent théoriquement résoudre les tâches COGS par correspondance de motifs plats, sans nécessiter de représentations hiérarchiques
Mécanisme clé : Le masquage des noms de syntagmes prépositionnels est crucial pour éviter les erreurs d'attraction
Problème d'apprentissage : L'échec des Transformers actuels est un problème d'apprentissage plutôt qu'une limitation de capacité
Erreurs prévisibles : Les erreurs spécifiques du modèle de base peuvent être prédites avec précision basées sur l'hypothèse de traitement plat
Construction manuelle : Le modèle RASP est conçu manuellement, non appris
Limitation de vocabulaire : Suppose que le mappage des parties du discours et des types de verbes est connu, ne résout pas la généralisation de vocabulaire
Spécificité linguistique : Limité à l'anglais, l'applicabilité à d'autres langues est inconnue
Spécificité de tâche : Le modèle est conçu spécifiquement pour COGS, pas un modèle de langage généraliste
Kim, N., & Linzen, T. (2020). COGS: A compositional generalization challenge based on semantic interpretation. EMNLP 2020.
Wu, Z., Manning, C. D., & Potts, C. (2024). ReCOGS: How incidental details of a logical form overshadow an evaluation of semantic interpretation. TACL.
Weiss, G., Goldberg, Y., & Yahav, E. (2021). Thinking like transformers. NeurIPS 2021.
Zhou, H., et al. (2023). What algorithms can transformers learn? A study in length generalization. arXiv preprint.
Zeller, A., et al. (2023). Grammar coverage. In The Fuzzing Book.
Cet article fournit des perspectives importantes pour comprendre les capacités et les limitations des Transformers dans les tâches de généralisation compositionnelle par une analyse théorique rigoureuse et une vérification empirique. Bien qu'il présente certaines limitations pratiques, ses contributions théoriques et innovations méthodologiques ont une valeur importante pour faire progresser la recherche connexe.