부분 최적 수송을 위한 가속 Sinkhorn 알고리즘

초록

본 논문은 부분 최적 수송(POT) 문제에 대해 엔트로피 정규화된 Sinkhorn 방법을 가속화한 ASPOT 알고리즘을 제안한다. Nesterov식 모멘텀과 Greenkhorn 업데이트를 결합해 복잡도를 O(n^{7/3} ε^{-5/3})로 개선하고, 정규화 파라미터 γ의 선택이 기존 Sinkhorn의 ε‑의존성을 어떻게 완화시키는지 이론적으로 분석한다. 색상 변환과 점군 정합 실험을 통해 제안 방법이 수렴 속도와 해의 품질 모두에서 기존 베이스라인을 능가함을 입증한다.

상세 분석

본 연구는 부분 최적 수송(POT)이라는, 전체 질량이 동일하지 않거나 이상치가 존재할 때 일부 질량만을 이동시키는 문제에 초점을 맞춘다. 기존의 Sinkhorn 기반 방법은 엔트로피 정규화 덕분에 계산 효율성을 확보했지만, POT에 적용될 경우 복잡도 O(n^{2} ε^{-4})에 머물러 대규모 데이터에 한계가 있었다. 저자들은 두 가지 주요 기여를 제시한다. 첫 번째는 Nesterov 가속을 Greenkhorn 좌표 업데이트와 결합한 Accelerated Sinkhorn for POT(ASPO​T)이다. 알고리즘 2에서 보듯이, 매 반복마다 Nesterov 외삽 단계와 함께 가장 큰 잔차를 교정하는 Greenkhorn 스텝을 수행함으로써 모멘텀의 발산을 억제하고, 이론적으로는 δ_t ≤ O(1/t²) 수렴률을 확보한다. 이를 바탕으로 전체 연산 복잡도를 \tilde O(n^{7/3} ‖C‖{∞}^{4/3} ε^{-5/3}) 으로 낮춘다. 두 번째 기여는 정규화 파라미터 γ 의 선택이 Sinkhorn의 복잡도에 미치는 영향을 정밀히 분석한 것이다. 기존 연구는 γ를 고정하거나 경험적으로 튜닝했지만, 본 논문은 γ = (p² ε)/(49 H{min}(r,c)) 와 같은 형태로 설정하면 복잡도가 O(n² H_{min}^{1/p} ‖C‖_{∞}^{2} ε^{-3(p+1)/p}) + o(1) 으로 개선된다는 정리를 제시한다. 여기서 p∈