[논문 리뷰] Iskra: A System for Inverse Geometry Processing

Ana Dodik, Ahmed H. Mahmoud외 1명|arXiv (Cornell University)|2026. 02. 12.

3D Shape Modeling and Analysis인용 수 0

한 줄 요약

iskra는 기하 처리 문제의 해를 미분 가능 파이프라인에 내재화함으로써 차별화된 해법들을 구하고, 희소 기하 연산자와 ML 프레임워크를 결합한 엔드 투 엔드 최적화를 가능하게 한다.

ABSTRACT

We propose a system for differentiating through solutions to geometry processing problems. Our system differentiates a broad class of geometric algorithms, exploiting existing fast problem-specific schemes common to geometry processing, including local-global and ADMM solvers. It is compatible with machine learning frameworks, opening doors to new classes of inverse geometry processing applications. We marry the scatter-gather approach to mesh processing with tensor-based workflows and rely on the adjoint method applied to user-specified imperative code to generate an efficient backward pass behind the scenes. We demonstrate our approach by differentiating through mean curvature flow, spectral conformal parameterization, geodesic distance computation, and as-rigid-as-possible deformation, examining usability and performance on these applications. Our system allows practitioners to differentiate through existing geometry processing algorithms without needing to reformulate them, resulting in low implementation effort, fast runtimes, and lower memory requirements than differentiable optimization tools not tailored to geometry processing.

연구 동기 및 목표

• 역 기하 처리의 필요성과 내부 기하 솔버를 미분해야 할 필요성을 제시한다.
• 희소 메쉬 계산과 미분 가능한 솔버를 지원하는 명령형 텐서 기반 시스템인 iskra를 제안한다.
• 선형 시스템, 고유값 문제, 고정점/ADMM 반복 등 광범위한 기하 처리 문제에 대한 미분을 가능하게 한다.
• 다양한 기하 작업에서의 사용성 및 성능을 입증하고 오픈 소스 구현을 제공한다.

제안 방법

• f(x; y)=0으로 정의된 해 g(x)에 대해 역전파하기 위한 adjoint 방법에 기반한 암묵적 미분 프레임워크를 도입한다.
• 수집-산란 프리미티브를 갖춘 희소 기하 워크플로에 대해 PyTorch 기반의 메시 인식형 텐서 API를 제공한다.
• 사용자가 순방향 솔버(선형, 고유값, 고정점, ADMM)를 정의하고 암묵적 관계를 통해 역전파를 자동으로 생성하도록 허용한다.
• 일반적인 구조에 대해 선형 암묵적 층(A y = b)과 스펙트럴 층(A U = U diag(λ))과 같은 특화된 암묵적 계층을 제공한다.
• 미분 가능한 외부 목적함수와 내부 기하 솔버를 결합하고 표준 기울기 기반 방법으로 외부 매개변수 θ를 최적화하는 것을 지원한다.

실험 결과

연구 질문

• RQ1 솔버를 재정리하지 않고도 광범위한 기하 처리 솔버를 미분할 수 있는가?
• RQ2 희소 메쉬 연산을 텐서 기반의 미분 가능 프로그래밍과 역 문제에 어떻게 통합할 수 있는가?
• RQ3 기하 솔버를 미분할 때 역전파 전략(고정점 vs. adjoint)의 트레이드오프는 무엇인가?
• RQ4 ARAP, 스펙트럴 파라메타라이제이션, 지오데식 거리 계산, ADMM 기반 변형을 통한 미분이 실제로 어떻게 수행되는가?
• RQ5 ML 파이프라인에 역기하 처리 임베딩의 상호 운용성 및 성능 영향은 무엇인가?

주요 결과

• iskra는 다양한 솔버에 대해 효율적인 adjoint-based 역방향 패스를 제공하여 선형 시스템, 고유값 문제, 로컬-글로벌 솔버, ADMM 반복을 통한 미분을 가능하게 한다.
• 이 시스템은 CPU/GPU 실행을 위해 PyTorch와 통합되며 gather-scatter 프리미티브를 통한 희소 메쉬 계산을 지원한다.
• 평균 곡률 흐름, 스펙트럴 컨포멀 파라메타라이제이션, 측지 거리 계산, ARAP 변형을 통한 미분을 시연한다.
• iskra는 메모리와 속도의 균형을 맞추기 위한 유연한 역전파 커스터마이제이션과 고정점 대 adjoint 기반 접근법을 제공한다.
• 오픈 소스 구현이 이용 가능하여 역 기하 처리 워크플로의 실용적 사용 및 벤치마킹을 가능하게 한다.