ComFree-Sim: A GPU-Parallelized Analytical Contact Physics Engine for Scalable Contact-Rich Robotics Simulation and Control

Resumo Técnico: ComFree-Sim

1. O Problema

A simulação física para robótica rica em contatos (manipulação e locomoção) enfrenta um gargalo fundamental na resolução de contatos.

• Limitação Atual: Os motores de física mainstream (como MuJoCo, Drake, e suas variantes GPU como MJWarp) geralmente impõem restrições de não-penetração e atrito de Coulomb através de restrições de complementaridade ou otimização com restrições.
• Custo Computacional: Isso exige soluções iterativas a cada passo de tempo, cujo custo cresce superlinearmente com a densidade de contatos.
• Consequência: Em cenários com muitos contatos simultâneos (ex: mãos robóticas manipulando objetos, pilhas de objetos caindo), o tempo de simulação aumenta drasticamente, tornando inviável o uso em tempo real para controle de alta frequência (MPC) ou treinamento de políticas em larga escala com paralelismo massivo.

2. Metodologia

O ComFree-Sim é um motor de física analítico e paralelizado em GPU que elimina a necessidade de solucionadores iterativos de complementaridade.

• Modelagem Livre de Complementaridade (ComFree):
  • Em vez de resolver um sistema de otimização, o método calcula os impulsos de contato em forma fechada (closed-form).
  • Utiliza um mecanismo de atualização de impedância no cone dual do atrito de Coulomb.
  • Princípio de Predição-Correção:
    1. Predição: Calcula a velocidade futura sob forças suaves (sem contato).
    2. Correção: Aplica uma correção baseada em impedância para resolver violações de restrições (não-penetração e viabilidade de atrito) no espaço do cone dual.
• Modelo de Contato Unificado 6D:
  • Estende o modelo pontual para capturar atrito tangencial, torsional (rotação em torno da normal) e rolante.
  • As restrições do cone dual são aproximadas por poliedros lineares, permitindo a separação dos cálculos por faces do cone.
• Paralelização em GPU (Warp):
  • A resolução de contato é desacoplada entre pares de contato e separável entre as faces do cone.
  • Isso mapeia naturalmente para kernels de GPU, permitindo que o custo computacional escale de forma quase linear com o número de contatos, em vez de superlinear.
• Heurística de Impedância Dual-Cone:
  • Introduz parâmetros de impedância (rigidez e amortecimento) que podem ser ajustados globalmente ou aprendidos, garantindo estabilidade numérica e compatibilidade com a interface do MuJoCo.

3. Principais Contribuições

1. Modelagem Analítica Unificada 6D: Extensão do modelo de contato livre de complementaridade para incluir atrito tangencial, torsional e de rolamento dentro de um framework analítico de cone dual.
2. Paralelização GPU com Heurísticas Estáveis: Implementação eficiente em GPU (via framework Warp) que mantém uma parametrização leve para o usuário, suportando também agendamento de impedância baseado em aprendizado.
3. Interface Drop-in e Avaliação Abrangente:
   • Implementado como um backend alternativo compatível com MuJoCo (substituindo o MJWarp).
   • Validação em benchmarks de simulação (penetração, atrito, estabilidade) e em hardware real (mão robótica LEAP).

4. Resultados

A. Benchmarks de Simulação (vs. MJWarp):

• Escalabilidade: O ComFree-Sim demonstra escalabilidade quase linear com o número de contatos, enquanto o MJWarp mostra crescimento não linear (superlinear).
• Throughput: Em cenas densas com muitos contatos, o ComFree-Sim atinge 2 a 3 vezes mais throughput (passos de simulação por segundo) que o MJWarp.
• Fidelidade Física:
  • Penetração: Com ajuste adequado de parâmetros de impedância ($k_{user}, d_{user}$), alcança profundidades de penetração comparáveis ou menores que o MJWarp.
  • Atrito: Captura corretamente a dissipação de energia em atrito torsional e de rolamento.
  • Estabilidade: Mantém estabilidade numérica em passos de tempo moderados, sem deriva espúria de velocidade.

B. Manipulação Dáctil em Tempo Real (Hardware LEAP Hand):

• Cenário: Controle MPC (Model Predictive Control) baseado em MPPI para reorientação de objetos com a mão robótica LEAP (multi-dedos) em tempo real.
• Desempenho:
  • Redução significativa no tempo de computação do passo do MPC (média de 2.4x mais rápido que o MJWarp).
  • Taxa de Sucesso: A latência reduzida permitiu taxas de controle de 35–72 Hz, resultando em uma taxa de sucesso em malha fechada 27 pontos percentuais maior em média em comparação ao MJWarp.
  • Precisão de posicionamento e orientação comparável à do MJWarp.

C. Retargeting de Movimento (Locomoção Ágil):

• Cenário: Retargeting de movimentos de um robô humanoide (Unitree G1) usando o algoritmo SPIDER.
• Resultado: O ComFree-Sim alcançou desempenho de rastreamento de tarefa comparável ao MJWarp, com uma pequena "lacuna sim-to-sim" (diferença entre simuladores), mas com tempo de otimização substancialmente menor.

5. Significado e Impacto

O ComFree-Sim representa um avanço significativo na infraestrutura de simulação robótica ao demonstrar que:

1. Resolução Analítica é Viável: Métodos analíticos desacoplados podem competir com solucionadores iterativos baseados em complementaridade em termos de fidelidade física, mas com custos computacionais drasticamente menores.
2. Habilitador de Controle de Alta Frequência: Ao eliminar o gargalo da resolução de contatos, o motor permite o uso de simulação em tempo real para tarefas ricas em contatos, como manipulação complexa e locomoção ágil, que antes eram limitadas pela latência do solver.
3. Escalabilidade para Aprendizado: A capacidade de paralelização massiva em GPU com escalabilidade linear torna o ComFree-Sim ideal para treinamento de políticas de RL em larga escala e geração de dados sintéticos, onde o número de ambientes paralelos e a densidade de contatos são críticos.

Em resumo, o ComFree-Sim oferece uma alternativa escalável, de baixa latência e fisicamente fiel para simulação robótica, superando as limitações de escalabilidade dos motores de física tradicionais em cenários de contato denso.