WorldParticle: Transformer를 통한 라그랑주 입자 역학 (Lagrangian Particle Dynamics)의 통합

요약

WorldParticle은 단일 Transformer 아키텍처를 사용하여 천, 유체, 고체 등 다양한 물리 현상을 통합적으로 모델링하는 학습 기반 입자 시뮬레이터입니다. 예측-수정(prediction-correction) 설계를 통해 입자 간 상호작용을 정밀하게 예측하며, 슈퍼 토큰(super token) 기술을 활용해 계산 효율성을 높였습니다. 이 모델은 학습되지 않은 재료나 경계 조건에서도 뛰어난 일반화 성능을 보여줍니다.

핵심 포인트

• 단일 Transformer 아키텍처로 6가지 이상의 다양한 물리 역학 범주를 통합 모델링 가능
• 예측기(Predictor)와 학습된 수정기(Learner Corrector)를 결합한 예측-수정 설계 채택
• 슈퍼 토큰 인코더와 점진적 토큰 병합(Progressive token merging)을 통한 어텐션 비용 감소 및 효율적 통신
• 학습되지 않은 재료, 경계 구성, 초기 조건 및 외력에 대한 높은 일반화 능력 입증
• 상호작용 제어 및 역설계(inverse design) 등 다운스트림 작업으로의 확장성 확인