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
Erforschung der Kompositionalen Generalisierung (in COGS/ReCOGS_pos) durch Transformer mittels Restricted Access Sequence Processing (RASP)
Menschen können neue Kombinationen von Wörtern verstehen, die in unterschiedlichen Kontexten erkannt werden – eine Fähigkeit, die als kompositionalen Generalisierung bezeichnet wird. Der COGS-Benchmark meldet, dass Transformer-Modelle bei bestimmten strukturellen Generalisierungen eine Genauigkeit von 0% aufweisen. Dieses Paper verwendet die RASP-Sprache (Restricted Access Sequence Processing), um zu beweisen, dass Transformer-Encoder-Decoder die COGS- und semantisch äquivalenten ReCOGS_pos-Aufgaben systematisch und kompositionell ausführen können: Das RASP-Modell erreicht nahezu perfekte Ergebnisse bei der strukturellen Generalisierungsteilung. Die Forschung zeigt, dass (Re)COGS-Aufgaben keine hierarchischen oder Baumstruktur-Lösungen erfordern, sondern stattdessen 19 Aufmerksamkeitsköpfe mit kompatiblen flachen Musterabgleichsregeln verwenden, die grammatikalische Muster durch Maskierung von Präpositionalphrasen und Nebensätzen identifizieren.
Das Kernproblem dieser Forschung ist die Einschränkung der Fähigkeiten von Transformer-Modellen bei Aufgaben der kompositionalen Generalisierung, insbesondere die schlechte Leistung beim COGS-Benchmark (Compositional Generalization Challenge Based on Semantic Interpretation).
Theoretische Bedeutung: Kompositionalen Generalisierung ist eine Kernfähigkeit des menschlichen Sprachverständnisses. Das Verständnis, wie neuronale Netze diese Fähigkeit implementieren, ist entscheidend für die Förderung des Sprachverständnisses in der KI.
Praktische Bedeutung: Die nahezu 0%ige Genauigkeit aktueller Transformer-Modelle bei strukturellen Generalisierungsaufgaben deutet auf grundlegende Einschränkungen hin, die Lösungen erfordern.
Flache Netzwerk-Einschränkungen: Der von Kim und Linzen (2020) verwendete 2-schichtige Encoder-Decoder zeigt extrem schlechte Leistung bei struktureller Generalisierung.
Fehlerhafte Hierarchie-Annahme: Bestehende Forschung geht davon aus, dass Baumstrukturen oder hierarchische Darstellungen erforderlich sind, um COGS-Aufgaben zu lösen.
Ineffektivität der Tiefe: Petty et al. (2024) stellten fest, dass selbst eine Erhöhung auf 32 Schichten die Leistung des Transformer bei COGS-Strukturgeneralisierung nicht verbessert.
Der Autor wurde von Zhou et al. (2023) inspiriert, die RASP zur Analyse der Generalisierungsfähigkeiten von Transformern verwendeten. Er versucht, durch konstruktive Beweise zu zeigen, dass Transformer theoretisch COGS-Aufgaben lösen können, und analysiert, warum bestehende Modelle fehlschlagen.
Konstruktiver Beweis: Verwendung der RASP-Sprache zum Beweis, dass Transformer-Encoder-Decoder theoretisch COGS- und ReCOGS_pos-Aufgaben systematisch lösen können.
Flache Lösung: Vorschlag einer nicht-hierarchischen Lösung basierend auf 19 flachen Musterabgleichsregeln ohne rekursive Baumstruktur-Regeln.
Fehleranalyse: Vorhersage und Validierung spezifischer Fehlermuster des Basis-Transformers durch die Theorie der "Anziehungsfehler" (attraction errors).
Leistungsdurchbruch: RASP-Modell erreicht 99,89% String-Exact-Match auf COGS und 99,63% semantisches Exact-Match auf ReCOGS_pos.
Neue Generalisierungsteilung: Entdeckung und Validierung einer neuen schwierigen Generalisierungsteilung "v_dat_p2_pp_moved_to_recipient".
Durch Maskierungsmechanismen werden Fehler vermieden, bei denen Nomen in Präpositionalphrasen "falsche" grammatikalische Beziehungen "anziehen":
Fehler: The cake on the plate burned → theme(burn, plate) # Anziehungsfehler
Korrekt: The cake on the plate burned → theme(burn, cake) # Nach Maskierung
Rekursive Strukturen werden durch Decoder-Schleifen verarbeitet und unterstützen beliebig tiefe Verschachtelungen von Präpositionalphrasen und Nebensätzen.
Theoretische Machbarkeit: Transformer können COGS-Aufgaben theoretisch durch flache Musterabgleichung lösen, ohne hierarchische Darstellungen zu benötigen.
Schlüsselmechanismus: Maskierung von Präpositionalphrasen-Nomen ist entscheidend zur Vermeidung von Anziehungsfehlern.
Lernproblem: Das Versagen aktueller Transformer ist ein Lernproblem, nicht eine Fähigkeitsbegrenzung.
Fehler vorhersagbar: Spezifische Fehlermuster von Basis-Modellen können basierend auf der flachen Verarbeitungshypothese genau vorhergesagt werden.
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.
Dieses Paper trägt durch strenge theoretische Analysen und empirische Validierung wichtige Erkenntnisse zum Verständnis der Fähigkeiten und Einschränkungen von Transformern bei Aufgaben der kompositionalen Generalisierung bei. Obwohl es praktische Einschränkungen gibt, haben seine theoretischen Beiträge und methodischen Innovationen wichtigen Wert für die Förderung verwandter Forschung.