Volver a Blog

Aprendizaje RL vs. Imitación para la manipulación de robots: Guía para la decisión

Cuando utilizar el aprendizaje de refuerzo vs. aprendizaje de imitación para entrenar políticas de manipulación de robots <unk> tipo de tarea, disponibilidad de datos y presupuesto de cálculo.

Parte de: [Guía de aprendizaje por imitación]

[← Blog] T1)

La elección entre RL e IL no se trata de cuál es mejor se trata de cuál es apropiado para sus limitaciones específicas.

La regla de la decisión rápida

Antes del análisis detallado: si puede demostrar fácilmente la tarea con un sistema de teleoperación, utilice el aprendizaje de imitación. Si no puede demostrarla pero puede definir una clara señal de recompensa escalar, utilice RL. Si ninguna de las dos aplica, la tarea es demasiado compleja.

Esta regla es correcta para el 80% de los escenarios de manipulación práctica, mientras que el 20% restante requiere enfoques híbridos o un análisis cuidadoso de las compensaciones a continuación.

Aprendizaje por imitación: fortalezas y límites

Las ventajas de la IL que a menudo se subvaloran:

  • ** Ciclo de desarrollo rápido:** 100 demostraciones → política entrenada en 2-4 horas.
  • Seguro por construcción: La política aprende a mantenerse cerca de comportamientos demostrados.
  • No hay ingeniería de recompensas: Definir una función de recompensas para el ensamblaje de precisión es sorprendentemente difícil.
  • Costo predecible: El coste por mejoría de rendimiento es lineal y medible recoger 100 demostraciones más, esperar una mejora del N%.

Límites de IL:

  • Limita por la calidad del demostrador: La política no puede exceder la habilidad de los operadores que proporcionaron demostraciones.
  • Erro de composición: La clonación comportamental deriva de la distribución de la formación en horizontes de tiempo largos.
  • No puede descubrir soluciones no obvias: Si hay una estrategia que es difícil de demostrar pero fácil de ejecutar (por ejemplo, utilizando la dinámica inercial para tirar un objeto a una basura), IL nunca la encontrará.

Aprendizaje reforzado: fortalezas y límites

Ventajas de la RL:

  • Puede superar el rendimiento humano: RL descubrió estrategias sobrehumanas en los juegos, y ha encontrado estrategias de captura no intuitivas pero superiores para objetos específicos.
  • Hola comportamientos no demostrables: Las tareas en las que el humano no puede controlar fácilmente al robot (dinámica de alta velocidad, secuencias de contacto complejas) son objetivos naturales de RL.
  • ** Planificación a largo plazo:** Con recompensas bien formadas, RL puede manejar tareas de 20 a 50 pasos que son propensas a cometer errores en IL.

Los límites de RL que a menudo se subestiman:

  • Eficiencia de muestreo: La manipulación moderna de RL requiere de 500K-5M pasos de entorno para tareas simples.
  • Las recompensas de reducción (sólo el éxito/fracasos) rara vez funcionan sin una forma cuidadosa. Las funciones de recompensa densas para la manipulación son difíciles de especificar correctamente. Las recompensas mal formadas producen políticas que explotan la recompensa en lugar de resolver la tarea.
  • Dependencia de la simulación a la realidad: La simulación de la realidad en el mundo real de los robots físicos es posible, pero lenta y costosa.

Comparación de lado a lado

Dimension Aprendizaje por Imitación Reinforcement Learning
Sample efficiency 100-500 demos for most tasks 500K-5M sim steps (much more data)
Reward design needed? No (demonstrations are the reward) Yes (hardest part of the pipeline)
Simulation required? No (works on real robot directly) Practically yes (real-world RL is too slow)
Hardware access needed? Yes (physical robot + teleoperation) Only for validation (training is in sim)
GPU compute cost $5-50 per policy (2-12 hr training) $50-500 per policy (8-100 hr training)
Performance ceiling Bounded by demonstrator skill Can exceed human performance
Safety during training Inherently safe (follows demo distribution) Exploración can be dangerous on real hardware
Time to first policy 1-3 days (collect + train) 1-4 weeks (sim setup + training)
Sim-to-real gap? No (trained on real data) Yes (major challenge for deployment)
Multi-task scaling Linear cost per task (need demos for each) Reward re-engineering per task (often harder)

Diagrama de flujo de decisiones

Sigue esta secuencia de preguntas para llegar al enfoque correcto para su tarea específica:

  1. Si es así, IL es su punto de partida. Si no (por ejemplo, la tarea requiere velocidad sobrehumana, tiempo de reacción o grados de control de libertad), vaya al paso 3.
  2. ¿La tarea requiere precisión más allá de la capacidad del demostrador? Si la política de IL alcanza el 70-80% de éxito pero necesita 95%+, utilice IL calentador + refinación de RL (híbrido).
  3. ** ¿Puede definir una función de recompensa escalar para la tarea?** Si sí y tiene acceso a la simulación, utilice RL. Si la recompensa es ambigua o multi-objetivo, considere GAIL (utiliza demostraciones para aprender la recompensa) o dividir la tarea en subtareas con recompensas más claras.
  4. ¿Es posible transferir sim a real para su tarea? Si la tarea implica objetos rígidos con geometría conocida, sim a real es generalmente posible.
  5. Cuál es su presupuesto de computación? Si tiene acceso a las GPU A100/H100 y puede permitirse 10-100 horas de entrenamiento, RL es práctico.

Los métodos híbridos: el mejor de ambos

Approach Description Benefit When to Use
IL warm-start + RL fine-tuning Train IL policy first, use as RL initialization 10× more efficient RL When IL policy is 60-70% and you need 90%+
GAIL RL with reward from discriminator trained on demos No reward engineering When reward is hard to specify
DAgger IL with online data collection from expert Fixes compounding error When BC diverges at test time
Residual RL Base IL policy + RL residual corrector Safe exploration near IL behavior Precisión refinement of IL policies

Comparación de costes y líneas de tiempo

Más allá de los méritos técnicos, la decisión práctica entre RL e IL se reduce a menudo a la línea de tiempo y presupuesto del proyecto.

Resource IL (ACT, 500 demos) RL (PPO in Isaac Lab) Hybrid (IL + Residual RL)
Upfront engineering 1-2 weeks (pipeline setup) 2-4 weeks (sim setup + reward eng.) 3-5 weeks (both pipelines)
Data/experience collection 1-2 weeks (500 demos at 40/hr) 0 (automated in sim) 1 week (200 demos)
Training time 3-4 hours (RTX 4090) 8-24 hours (A100) 4 hr IL + 8 hr RL
GPU compute cost $5-15 $30-100 $35-80
Hardware access cost $800-2,000 (lease + operator) $0 (sim only until validation) $400-1,000
Total timeline 2-4 weeks 4-8 weeks 4-6 weeks
Typical success rate 80-90% 70-85% (after sim-to-real) 85-95%

La clave: la IL es más rápida y más barata para la mayoría de las tareas de manipulación. La RL se justifica cuando la IL no puede alcanzar el nivel de rendimiento requerido, cuando la tarea no es demostrable o cuando se necesita superar el rendimiento a nivel humano. El enfoque híbrido ofrece las tasas de éxito más altas, pero requiere inversión en ambas líneas de tubería.

Errores comunes en el método

Erro 1: Comienza con RL porque suena más sofisticado. Los equipos con antecedentes de ML a menudo usan RL por defecto porque es el enfoque más técnicamente interesante. Pero para la mayoría de las tareas prácticas de manipulación, IL produce mejores resultados más rápido.

Erro 2: Invertir poco en ingeniería de recompensas. Si eliges RL, presupuesta al menos el 40% de tu tiempo de ingeniería para el diseño y la iteración de funciones de recompensas. Una recompensa mal formada produce políticas que explotan la recompensa en lugar de resolver la tarea. El robot encuentra formas creativas de maximizar el número sin realizar el comportamiento previsto.

Erro 3: Ignorar la brecha de sim a real. Las políticas de RL entrenadas en simulación a menudo logran un 95% + de éxito en sim pero un 40-60% en el robot real sin aleatorización de dominio. Presupuesta tiempo y cálculo para la aleatorización de dominio, y siempre valida las políticas de sim entrenadas en hardware real antes de reclamar el éxito.

Erro 4: No recopilar datos de fallas durante el IL. Los operadores desechan las demostraciones fallidas por defecto. Pero como se discute en nuestra [guía de calidad de datos]T3), las demostraciones de fallas etiquetadas son valiosas para entrenar los comportamientos de recuperación y la detección de fallas.

Recomendaciones para el tipo de tarea

Task Type Recommended Approach Reason
Pick-and-place (varied poses) IL (ACT or Diffusion) Easily demonstrated, contact minimal
Precisión peg insertion IL or IL + Residual RL Demonstrable; RL refines final alignment
Dexterous in-hand manipulation RL (in sim) + IL for initialization Hard to demonstrate; RL finds solutions
Long-horizon assembly (10+ steps) Hierarchical: task planner + IL per step Neither alone handles long horizon well
Locomotion gait optimization RL Non-demonstrable, clear energy reward
Tool use (novel tools) IL for known tools, RL generalization Few-shot IL + sim RL exploration

El servicio de medio ambiente de RCSV [RL] T4) proporciona configuraciones de simulación para el entrenamiento de RL, y nuestro [servicios de datos] T5) maneja la colección de demostración de IL - a menudo utilizada en combinación para los enfoques híbridos descritos anteriormente.

Lectura relacionada

  • Empezando con NVIDIA Isaac Lab -- La plataforma principal para el entrenamiento RL acelerado por GPU
  • [Guía de LeRobot]T7) -- Formación de las políticas de IL con la política de ACT y difusión
  • [Políticas de robot de tiro cero vs. de tiro pocos] ((T8) -- Modelos de base como alternativa a tanto RL como IL desde cero
  • Costos de recogida de datos de robots -- Planificación presupuestaria para la recogida de datos de demostración de IL
  • [¿Qué hace buenos datos de formación de robots?]
  • Servicio de Medio Ambiente de la RL de la RCSV -- Simulación preconfigurada para la formación en RL
  • Servicios de datos -- Recopilación de demostraciones de IL gestionada para el lado IL de tuberías híbridas

¿Necesita ayuda para elegir el enfoque correcto?

RCSV proporciona tanto entornos de simulación de RL como recopilación de datos de IL. Podemos ayudar a diseñar la tubería adecuada para su tarea.

[Explora nuestros servicios]