Transformer를 통한 Lagrangian 입자 역학의 통합 시뮬레이션

요약

본 논문은 천, 유체, 고체 등 다양한 물리 현상을 단일 Transformer 아키텍처로 시뮬레이션할 수 있는 통합 입자 시뮬레이터를 제안합니다. 예측-수정(prediction-correction) 설계를 기반으로 하며, 슈퍼 토큰(super tokens) 개념을 도입하여 입자 간 상호작용을 효율적으로 계산하고 계산 비용을 절감합니다. 이를 통해 학습되지 않은 재료나 경계 조건에서도 높은 일반화 성능을 보여줍니다.

핵심 포인트

• 단일 Transformer 아키텍처로 6가지 이상의 다양한 역학 범주(유체, 고체, 분자 역학 등)를 통합 모델링 가능
• 예측기(Predictor)와 수정기(Corrector) 구조를 통해 외력에 의한 움직임과 입자 간 상호작용을 분리하여 처리
• 점진적 토큰 병합(Progressive token merging)과 슈퍼 토큰 인코더/디코더를 통해 주의 집중(attention) 비용 최적화
• 학습되지 않은 재료, 경계 구성, 초기 조건 및 외력에 대해 뛰어난 일반화 성능 입증
• 상호작용 제어, 역설계, 실제 조작 데이터 학습 등 다양한 다운스트림 작업 수행 가능