Aceleración GPU para la teletransportación de pesos en RL

Fuentes: GPU-accelerated Weight Teleportation for Reinforcement Learning

El artículo técnico describe un paradigma de actualización de pesos optimizado para el aprendizaje por refuerzo (RL) a gran escala, diseñado para superar las limitaciones de la sincronización de pesos completa. A diferencia de la preentrenamiento o la inferencia, el RL requiere que los pesos del entrenador y del muestreador coincidan estrechamente, lo que genera un cuello de botella al transmitir terabytes de datos. La solución propuesta utiliza una compresión masivamente paralela en GPU y un pipeline de descompresión nativo para la arquitectura Blackwell de NVIDIA, logrando tiempos de sincronización de pesos inferiores a 4 segundos y pausas de muestreador menores a 0,23 segundos en configuraciones de producción reales.

El sistema se basa en la sincronización de deltas de pesos, donde solo se transmiten las modificaciones necesarias mediante máscaras comprimidas. Al aprovechar la naturaleza dispersa de las señales de entrenamiento, el payload comprimido es significativamente más pequeño que el peso completo. La implementación integra algoritmos de compresión como LZ4 y Snappy, optimizados para el hardware Blackwell, que permite una descompresión eficiente mediante el Motor de Descompresión (DE) dedicado. Este enfoque elimina la latencia asociada a la lectura desde disco y la recarga completa de los checkpoints, permitiendo que la descompresión sea el cuello de botella principal. Los resultados demuestran que esta metodología es más rápida que la comunicación P2P RDMA intra-clúster para la mayoría de las cargas de trabajo, ofreciendo una solución escalable para clústeres de GPU distribuidos.