RigidFormer: 트랜스포머를 이용한 강체 역학 학습

요약

RigidFormer는 다중 객체 강체 역학 시뮬레이션의 어려움(불연속적 접촉, 오차 누적 등)을 해결하기 위해 제안된 트랜스포머 기반 모델입니다. 이 모델은 메시 프리 표현으로부터 고충실도 강체 역학을 효율적으로 학습하며, 객체 수준에서 추론하고 압축된 앵커를 통해 각 객체를 전진시킵니다. 특히 Anchor-based RoPE와 미분 가능한 Kabsch 정렬을 사용하여 기하학적 일관성과 강성을 유지하면서 높은 성능과 일반화 능력을 보여줍니다.

핵심 포인트

• RigidFormer는 메시 프리(mesh-free) 표현을 활용하여 다중 객체 강체 역학 시뮬레이션을 수행합니다.
• 객체 수준 추론 및 압축된 앵커를 사용하여 계산 효율성을 높이고, 접촉 관련 기하학적 정보를 유지합니다.
• Anchor-based RoPE와 미분 가능한 Kabsch 정렬을 도입하여 순열 등변성(permutation-equivariance)과 강성(rigidity) 제약을 효과적으로 적용했습니다.
• 기존 메시 기반 방법론 대비 높은 성능, 빠른 실행 속도, 그리고 넓은 일반화 능력을 입증했습니다.