A Unified and Fast-Sampling Diffusion Bridge Framework via Stochastic Optimal Control
Created by
Haebom
Category
Empty
저자
Mokai Pan, Kaizhen Zhu, Yuexin Ma, Yanwei Fu, Jingyi Yu, Jingya Wang, Ye Shi
개요
UniDB는 확률적 최적 제어(SOC)를 기반으로 하는 확산 브리지 모델을 위한 통합적이고 빠른 샘플링 프레임워크입니다. Doob의 $h$-변환을 사용하는 기존 확산 브리지의 한계를 개선하기 위해, SOC 기반 최적화를 통해 문제를 재구성하고, 튜닝 가능한 터미널 페널티 계수를 도입하여 디테일 보존 및 출력 품질을 향상시켰습니다. 또한, 훈련 없는 가속 알고리즘과 데이터 예측 모델 및 SDE-Corrector 메커니즘을 통해 계산 효율성을 높였습니다.
시사점, 한계점
•
시사점:
◦
SOC 기반 접근 방식을 통해 확산 브리지 모델의 이론적 기반을 강화하고, 기존 방식의 한계를 극복했습니다.
◦
튜닝 가능한 터미널 페널티 계수를 통해 디테일 보존과 출력 품질을 개선했습니다.
◦
훈련 없는 가속 알고리즘, 데이터 예측 모델, SDE-Corrector 메커니즘을 통해 계산 효율성을 향상시켰습니다.
