Tree-Sliced Wasserstein Distance with Nonlinear Projection
Created by
Haebom
저자
Thanh Tran, Viet-Hoang Tran, Thanh Chu, Trang Pham, Laurent El Ghaoui, Tam Le, Tan M. Nguyen
개요
본 논문은 기존의 Sliced Wasserstein (SW) 거리의 대안으로 등장한 Tree-Sliced (TSW) 거리에 대한 새로운 비선형 투영 프레임워크를 제안한다. 기존 TSW 거리의 선형 투영을 일반적인 투영으로 대체하여 Radon 변환의 단사성을 보장하고 거리의 정의를 유지한다. 이를 통해 유클리드 공간과 구면상의 측도에 대한 효율적인 거리 측정을 가능하게 한다. 제안된 거리는 유클리드 및 구면 데이터셋에 대한 광범위한 수치 실험을 통해 검증되었으며, 기울기 흐름, 자기 지도 학습 및 생성 모델 등 다양한 응용 분야에서 기존 SW 및 TSW 변형보다 향상된 성능을 보여준다.