[논문 리뷰] SymX: Energy-based Simulation from Symbolic Expressions

José Antonio Fernández-Fernández, Fabian Löschner외 3명|arXiv (Cornell University)|2023. 02. 22.

Modeling and Simulation Systems인용 수 4

한 줄 요약

SymX는 물리기반 시뮬레이션에서 에너지 함수의 1차 및 2차 도함수를 자동으로 계산하고, 실시간으로 최적화된 벡터화된 코드를 컴파일하며, 병렬로 전역 도함수를 조립하는 기호 미분 프레임워크이다. 사용자 노력 최소화로 고성능, 생산용 시뮬레이션을 가능하게 하며, 고차수 유한요소법 시뮬레이션에서 최신 기술 프레임워크 대비 최대 두 배수의 성능 향상을 달성한다.

ABSTRACT

Optimization time integrators are effective at solving complex multi-physics problems including deformable solids with non-linear material models, contact with friction, strain limiting, etc. For challenging problems, Newton-type optimizers are often used, which necessitates first- and second-order derivatives of the global non-linear objective function. Manually differentiating, implementing, testing, optimizing, and maintaining the resulting code is extremely time-consuming, error-prone, and precludes quick changes to the model, even when using tools that assist with parts of such pipeline. We present SymX, an open source framework that computes the required derivatives of the different energy contributions by symbolic differentiation, generates optimized code, compiles it on-the-fly, and performs the global assembly. The user only has to provide the symbolic expression of each energy for a single representative element in its corresponding discretization and our system will determine the assembled derivatives for the whole simulation. We demonstrate the versatility of SymX in complex simulations featuring different non-linear materials, high-order finite elements, rigid body systems, adaptive discretizations, frictional contact, and coupling of multiple interacting physical systems. SymX's derivatives offer performance on par with SymPy, an established off-the-shelf symbolic engine, and produces simulations at least one order of magnitude faster than TinyAD, an alternative state-of-the-art integral solution.

연구 동기 및 목표

• 복잡한 물리 시뮬레이션에서 최적화 기반 시간 적분기법에 필요한 1차 및 2차 도함수의 유도 및 효율적 구현을 자동화하는 것.
• 다중물리계에서 비선형 에너지 함수의 수작업, 실수 유발 가능성이 높은 기호적 및 코드 수준의 미분이 필요 없도록 하는 것.
• 빠른 컴파일과 도함수의 고성능 평가를 제공함으로써 시뮬레이션 개발의 빠른 프로토타이핑과 반복을 가능하게 하는 것.
• 사용자 정의 데이터 유형과 솔버를 지원하여 기존 시뮬레이션 파이프라인에 탄력적으로 통합할 수 있도록 하는 것.
• 접촉, 균열, 적응적 메esh 정밀도 향상 등으로 인해 변화하는 희소성 패턴을 가진 동적 시스템을 처리하는 것.

제안 방법

• 프레임워크는 유한요소나 제약 조건과 같은 개별 요소의 국소 에너지 함수를 기호 표현으로 표현한다.
• 기호적 미분을 수행하여 자유도에 대한 에너지의 해석적 1차 및 2차 도함수를 자동으로 유도한다.
• 생성된 도함수 코드는 고성능을 위해 실시간으로 JIT 컴파일 파이프라인을 사용해 벡터화하고 컴파일한다.
• 시스템은 병렬로 요소 수준 기여를 전역 잔차 행렬과 헤시안 행렬로 자동 조립한다.
• 임의의 데이터 유형을 지원하며, 기존 최적화 솔버 및 시뮬레이션 백엔드와 원활하게 통합된다.
• 접촉이나 적응적 메쉬 정밀도 향상으로 인한 시스템 구조 및 희소성의 동적 변화를 처리한다.

실험 결과

연구 질문

• RQ1 기호적 미분과 실시간 코드 생성이 복잡한 물리 시뮬레이션에서 수작업 최적화된 코드와 유사한 성능을 달성할 수 있는가?
• RQ2 자동 도함수 계산이 다중물리 시뮬레이션 시스템의 개발 시간과 실수율을 얼마나 줄일 수 있는가?
• RQ3 물질 모델, 요소 유형, 시스템 복잡성의 변화에 따라 프레임워크는 어떻게 스케일링되는가?
• RQ4 접촉이나 구조 변화로 인한 희소성 패턴이 변화하는 동적 시스템을 효율적으로 처리할 수 있는가?
• RQ5 고차수 유한요소법에서 기존 자동 미분 도구에 비해 어떤 성능 향상이 달성 가능한가?

주요 결과

• SymX는 수작업 최적화된 코드와 유사한 성능을 달성하며, 고차수 유한요소 시뮬레이션에서 최신 기술 프레임워크 대비 최대 두 배수의 성능 향상을 보였다.
• 프레임워크는 복잡한 에너지 함수를 그 수학적 표현에 가깝게 기술하는 데 최소한의 코드로 시뮬레이션 개발을 빠르게 가능하게 했다.
• 5차 비틀림 요소의 경우, SymX는 23,824개의 도함수 항과 51,565개의 연산을 생성했으며, 최적화된 코드 1002 kB로 컴파일되어 높은 효율성을 보였다.
• 9종의 다른 에너지 유형이 포함된 복잡한 자동차 드리ft 시뮬레이션에서, SymX는 강체, 탄성 고체, 접촉 제약 조건 간의 전역 도함수 조립을 성공적으로 관리했다.
• 적응적 천공 및 트레부체트 시뮬레이션에서 시스템은 동적 변화와 복잡한 결합을 일관된 성능으로 처리하며 뛰어난 안정성을 보였다.
• 비미분 가능 중간 표현(예: 노름 또는 제곱근)을 다루는 데에는 제한이 있지만, SymX는 경고를 제공하고 수치적 불안정성을 완화하기 위한 안정적인 근사치를 지원한다.