Grokking en el mundo real: Aumento de datos para el razonamiento de múltiples saltos con Transformers en escenarios prácticosGrokking in the Wild: Data Augmentation for Real-World Multi-Hop
Reasoning with Transformers
Los Transformers han logrado un gran éxito en numerosas tareas de PLN, pero siguen mostrando brechas notables en el razonamiento factual de múltiples pasos, especialmente cuando el conocimiento del mundo real es escaso. Los avances recientes en grokking han demostrado que las redes neuronales pueden transitar de memorizar a generalizar perfectamente una vez que detectan patrones lógicos subyacentes; sin embargo, estos estudios han utilizado principalmente tareas sintéticas pequeñas. En este artículo, por primera vez, extendemos el grokking a datos factuales del mundo real y abordamos el desafío de la escasez de datos aumentando grafos de conocimiento existentes con datos sintéticos cuidadosamente diseñados para elevar la proporción phi_r de hechos inferidos sobre hechos atómicos por encima del umbral requerido para el grokking. Sorprendentemente, encontramos que incluso datos sintéticos factualmente incorrectos pueden fortalecer los circuitos de razonamiento emergentes en lugar de degradar la precisión, ya que obligan al modelo a depender de la estructura relacional en lugar de la memorización. Al evaluar en benchmarks de razonamiento multi-hop, nuestro enfoque alcanza hasta un 95-100% de precisión en 2WikiMultiHopQA, mejorando sustancialmente sobre líneas base sólidas y equiparando o superando los resultados actuales del estado del arte. Además, proporcionamos un análisis en profundidad de cómo el aumento de phi_r impulsa la formación de circuitos generalizadores dentro de los Transformers. Nuestros hallazgos sugieren que la ampliación de datos basada en grokking puede desbloquear capacidades implícitas de razonamiento multi-hop, abriendo la puerta a un razonamiento factual más robusto e interpretable en modelos de lenguaje a gran escala.