약한 로그‑볼록성에서 역확산의 와셔스테인 전파: 메트릭 불일치를 활용한 단일 스위치 라우팅

본 논문은 스코어 기반 확산 모델의 역시간 샘플링 과정에서 발생하는 오류 전파를 보다 정밀하게 분석하고, 기존의 전역적인 유클리드 거리(W₂) 전파 방식이 약한 로그‑볼록성(weak log‑concavity) 상황에서 과도하게 보수적일 수 있음을 지적한다. 약한 로그‑볼록성은 목표 분포가 전역적으로는 수축성을 가지지만, 국소적으로는 비볼록적인 형태를 유지한다는 특성을 갖는다. 이러한 특성 하에서는 Gaussian smoothing이 큰 거리에서는 강한 수축을 제공하지만, 원점 근처에서는 여전히 일방향 단조성(negative monotonicity)만 존재해 Euclidean dissipativity가 부족하다. 저자는 이 현상을 정량화하기 위해 ‘방사형 하한 프로파일(κₛ(r))’을 도입한다. κₛ(r)는 학습된 역드리프트 b̂ₜ의 거리‑의존적 일방향 마진을 나타내며, r→0 일 때는 −b(s) (근거리 부정적 마진), r→∞ 일 때는 m(s) (원거리 양의 마진) 로 수렴한다. 즉, 동일한 역드리프트가 거리 스케일에 따라 서로 다른 기하학적 역할을 수행한다는 점을 명확히 드러낸다. 이러한 스케일 의존성을 활용하기 위해 저자는 ‘반사 커플링(reflection coupling)’을 적용한다. 반사 커플링은 두 복제 프로세스 Yₜ와 Ȳₜ의 거리 rₜ를 1차원 확산으로 축소시켜, 거리‑의존적 드리프트 κₛ(r)와 노이즈 항을 동시에 고려한다. 이때 rₜ의 동역학은 drₜ ≲ −κₛ₀(rₜ) rₜ dt + 2g(T−t) dWₜ 형태를 갖는다. 여기서 κₛ₀는 스위치 시점 s₀ 이전 구간에 대한 하한 프로파일이며, 스위치 전후의 기하학적 차이를 반영한다. 초기 구간(스위치 이전)에서는 ‘볼록 전송 메트릭 φₛ₀(r)’을 설계한다. φₛ₀는 두 부분으로 구성되는데, 0≤r≤R_sw(s₀) 구간에서는 φₛ₀(r)=∫₀ʳ e^{−λₛ₀ u²} du 로 정의되어, κₛ₀(r) 가 아직 음수일 수 있는 근거리에서 2차 미분 항 2g(s₀)² φ'' 가 부정적 마진을 보상한다. r≥R_sw(s₀) 구간에서는 φₛ₀(r)=φₛ₀(R_sw)+aₛ₀(r−R_sw) 로 선형 연장되어, 원거리에서 양의 마진 m(s) 를 그대로 활용한다. 이 메트릭은 반사 커플링 하에서 Lₛ₀ φₛ₀(r) = 2g(s₀)² φ'' − κₛ₀(r) r φ' ≤ −c(s₀) φₛ₀(r) 를 만족하도록 설계돼, φₛ₀(rₜ) 가 지수적으로 감소함을 보장한다. 결과적으로 초기 구간에서 ‘볼록 전송 메트릭’이 실제로 거리‑의존적 수축을 제공한다는 의미이며, 전통적인 W₂ 전파보다 더 강력한 경계를 만든다. 스위치 시점 s₀에서는 φₛ₀ 메트릭을 다시 최종 목표 메트릭인 W₂ 로 복귀해야 한다. φₛ₀는 전역적으로 선형 하한 aₛ₀ r ≥ φₛ₀(r) 를 갖기 때문에, W₁(µ,ν) ≤ aₛ₀^{−1} W_{φₛ₀}(µ,ν) 가 성립한다. 복귀 비용을 정량화하기 위해 p‑모멘트 보존을 가정하고, Hölder 불평등을 이용해 W₂(ℒ(Z_{t_s}), p_{s₀}) ≤ C_sw(p₀)·