[논문 리뷰] Second order models for optimal transport and cubic splines on the Wasserstein space

Jean‐David Benamou, Thomas Gallouët외 1명|arXiv (Cornell University)|2018. 01. 12.

Geometric Analysis and Curvature Flows참고 문헌 33인용 수 24

한 줄 요약

이 논문은 확률측도의 워셔슈타인 공간으로 삼차 스퍼인플 기법을 확장하여 2차 모델을 위한 최적 운반 이론을 제안한다. 문제를 엔트로피 정규화와 반연속적 방법을 통한 다중모멘트 최적 운반 문제의 완화로 공식화함으로써, 수치적 계산이 가능해지고 수렴 보장이 되며, 1차원 및 2차원 예제에서 성능이 입증된다.

ABSTRACT

On the space of probability densities, we extend the Wasserstein geodesics to the case of higher-order interpolation such as cubic spline interpolation. After presenting the natural extension of cubic splines to the Wasserstein space, we propose a simpler approach based on the relaxation of the variational problem on the path space. We explore two different numerical approaches, one based on multi-marginal optimal transport and entropic regularization and the other based on semi-discrete optimal transport.

연구 동기 및 목표

• 리만다이프 기하구조에서의 삼차 스퍼인플 보간을 확률측도의 워셔슈타인 공간으로 확장하기 위해.
• 워셔슈타인 거리에서 경로의 가속도를 최소화하는 변분형식을 기반으로 한 고차 스퍼인플의 제작을 위해.
• 다중모멘트 최적 운반을 이용한 워셔슈타인 스퍼인플을 수치적으로 계산할 수 있는 방법 개발을 위해.
• 엔트로피 정규화와 싱크호른 알고리즘을 사용하는 방법과 반연속적 최적 운반을 이용한 라귀어르 세포 방법을 비교하고 구현하기 위해.
• 1차원 및 2차원 데이터에서 방법을 검증하여 수렴성과 계산 가능성 확인을 위해.

제안 방법

• 확률측도 공간에서의 경로의 제곱가속도 최소화를 통해 워셔슈타인 공간 내 삼차 스퍼인플을 공식화한다.
• 변분 문제를 이산 시간점에서 위치와 속도의 귀속 조건을 갖는 다중모멘트 최적 운반 문제로 완화한다.
• 다중모멘트 최적 운반 문제에 엔트로피 정규화를 적용하여 엄밀히 볼록 최적화 문제로 전환하고 싱크호른 알고리즘을 통해 해결한다.
• 계산 비용을 줄이기 위해 텐서형 커널 구조를 활용하여 2차원에서 N개의 시간 단계와 Nx개의 격자점에 대해 O(N Nx⁴) 복잡도를 달성한다.
• 다른 방법으로는 공간 영역을 이산화하고 라귀어르 세포를 계산하여 반연속적 최적 운반 문제를 해결한다.
• 좌표 기반 해밀토니안 공식화와 크리스톼프 기호를 활용하여 기하학적 지오데식 및 삼차 스퍼인플 역학의 오일러-라그랑주 방정식 유도.

실험 결과

연구 질문

• RQ1 삼차 스퍼인플 보간은 유한차원 리만다이프 기하구조에서 무한차원 워셔슈타인 공간의 확률측도로 일반화될 수 있는가?
• RQ2 워셔슈타인 공간 내 고차 스퍼인플의 적절한 변분형식은 무엇이며, 이를 수치 계산 가능하도록 어떻게 완화할 수 있는가?
• RQ3 엔트로피 정규화와 반연속적 방법은 어떻게 워셔슈타인 스퍼인플을 효율적으로 계산할 수 있도록 조정될 수 있는가?
• RQ4 제안된 수치적 방법의 수렴 성질과 계산 복잡도는 어떠한가?
• RQ5 1차원 및 2차원 테스트 케이스에서 엔트로피와 반연속적 방법의 두 수치적 접근 방식은 정확도와 효율성 측면에서 어떻게 비교되는가?

주요 결과

• 제안된 워셔슈타인 공간 내 삼차 스퍼인플 문제의 완화는 연속적이고 엄밀히 볼록한 비용 함수를 갖는 다중모멘트 최적 운반 문제와 동치이다.
• 엔트로피 정규화 방법은 싱크호른 알고리즘을 통해 효율적인 계산이 가능하며, 텐서형 커널 구조를 활용하여 2차원에서 O(N Nx⁴) 복잡도를 달성한다.
• 반연속적 방법은 라귀어르 세포를 계산하여 운반 문제를 해결하며, 엔트로피 정규화와 다른 수치적 성질을 지닌 대안을 제공한다.
• 1차원 및 2차원에서의 수치 실험은 제안된 방법이 워셔슈타인 스퍼인플을 계산하는 데 있어 실현 가능성과 수렴성을 입증한다.
• 이론적 분석을 통해 약한 조건 하에서 완화된 문제가 원래 변분 문제와 동일한 해를 달성함을 확인하였으며, 최소화자의 존재성과 유일성도 입증하였다.
• 이 방법은 변분 스퍼인플을 워셔슈타인 공간으로 일반화하여 영상 및 형태 분석에서 시간 순서 데이터의 보간을 위한 새로운 프레임워크를 제공한다.