UniDB: A Unified Diffusion Bridge Framework via Stochastic Optimal Control
Created by
Haebom
저자
Kaizhen Zhu, Mokai Pan, Yuexin Ma, Yanwei Fu, Jingyi Yu, Jingya Wang, Ye Shi
개요
본 논문은 기존 확산 브리지 모델의 한계점인 이미지 디테일 손실 문제를 해결하기 위해, 확률적 최적 제어(SOC) 기반의 통합 프레임워크인 UniDB를 제안합니다. UniDB는 SOC 기반 최적화를 통해 최적 제어기를 위한 폐쇄형 해를 도출하여 기존 Doob의 $h$-변환 기반 확산 브리지 모델들을 일반화합니다. 기존 모델들은 UniDB 프레임워크에서 종단 페널티 계수가 무한대로 갈 때의 특수한 경우로 해석되며, UniDB는 조정 가능한 종단 페널티 계수를 통해 제어 비용과 종단 페널티 간의 최적 균형을 달성하여 디테일 보존 및 출력 품질을 크게 향상시킵니다. 다양한 이미지 복원 작업에 대한 실험을 통해 UniDB의 우수성과 적응성을 검증하고, 기존 모델과의 호환성을 강조합니다.