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
Explorando la Generalización Composicional (en COGS/ReCOGS_pos) por Transformers usando Procesamiento de Secuencias con Acceso Restringido (RASP)
Los humanos pueden comprender nuevas combinaciones de vocabulario identificado en diferentes contextos, una capacidad denominada generalización composicional (Compositional Generalization). El conjunto de pruebas COGS reporta que los modelos Transformer tienen una precisión del 0% en ciertas generalizaciones estructurales. Este artículo utiliza el lenguaje RASP (Procesamiento de Secuencias con Acceso Restringido) para demostrar que los codificadores-decodificadores Transformer pueden ejecutar sistemática y composicionalmente las tareas COGS y semánticamente equivalentes ReCOGS_pos: el modelo RASP logra puntuaciones casi perfectas en la división de generalización estructural. La investigación demuestra que las tareas (Re)COGS no requieren soluciones jerárquicas o de estructura arbórea, sino que utilizan 19 reglas de coincidencia de patrones planos compatibles con cabezas de atención, identificando patrones gramaticales mediante el enmascaramiento de frases preposicionales y cláusulas.
El problema central que aborda esta investigación es la limitación de capacidad de los modelos Transformer en tareas de generalización composicional, particularmente el desempeño deficiente en el conjunto de pruebas COGS (Desafío de Generalización Composicional basado en Interpretación Semántica).
Significado Teórico: La generalización composicional es una capacidad central de la comprensión del lenguaje humano; comprender cómo las redes neuronales implementan esta capacidad es crucial para avanzar en la comprensión del lenguaje de la IA
Significado Práctico: La precisión cercana al 0% de los modelos Transformer actuales en tareas de generalización estructural indica limitaciones fundamentales que requieren soluciones
Limitaciones de Redes Superficiales: El Codificador-Decodificador de 2 capas utilizado por Kim y Linzen (2020) tiene un desempeño extremadamente deficiente en generalización estructural
Hipótesis Jerárquica Incorrecta: La investigación existente asume que se requieren estructuras arbóreas o representaciones jerárquicas para resolver tareas COGS
Inefectividad de la Profundidad: Petty et al. (2024) descubrieron que incluso aumentar a 32 capas no mejora el desempeño del Transformer en generalización estructural COGS
El autor se inspiró en Zhou et al. (2023) utilizando RASP para analizar la capacidad de generalización de Transformer, intentando demostrar mediante prueba constructiva que Transformer puede teóricamente resolver tareas COGS y analizar por qué los modelos existentes fallan.
Prueba Constructiva: Utiliza el lenguaje RASP para demostrar que el Codificador-Decodificador Transformer puede teóricamente resolver sistemáticamente tareas COGS y ReCOGS_pos
Solución Plana: Propone una solución no jerárquica basada en 19 reglas de coincidencia de patrones planos, sin necesidad de reglas de estructura arbórea recursiva
Análisis de Errores: Predice y verifica patrones de error específicos del Transformer de línea base mediante la teoría de "errores de atracción"
Avance de Rendimiento: El modelo RASP logra 99.89% de coincidencia exacta de cadenas en COGS y 99.63% de coincidencia semántica exacta en ReCOGS_pos
Nueva División de Generalización: Descubre y verifica la nueva división de generalización difícil "v_dat_p2_pp_moved_to_recipient"
Mediante el mecanismo de enmascaramiento, evita que los nombres en frases preposicionales "atraigan" relaciones gramaticales incorrectas:
Error: The cake on the plate burned → theme(burn, plate) # Error de atracción
Correcto: The cake on the plate burned → theme(burn, cake) # Después del enmascaramiento
Viabilidad Teórica: Transformer puede teóricamente resolver tareas COGS mediante coincidencia de patrones planos, sin necesidad de representaciones jerárquicas
Mecanismo Clave: El enmascaramiento de nombres de frases preposicionales es crucial para evitar errores de atracción
Problema de Aprendizaje: El fracaso del Transformer actual es un problema de aprendizaje, no una limitación de capacidad
Errores Predecibles: Los errores específicos del modelo de línea base pueden predecirse con precisión basándose en la hipótesis de procesamiento plano
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.
Este artículo, mediante análisis teórico riguroso y verificación empírica, proporciona perspectivas importantes para comprender las capacidades y limitaciones de Transformer en tareas de generalización composicional. Aunque existen algunas limitaciones de practicidad, sus contribuciones teóricas e innovaciones metodológicas tienen valor importante para avanzar la investigación relacionada.