Bi-Anchor Interpolation Solver for Accelerating Generative Modeling

작성자

Haebom

카테고리

Empty

저자

Hongxu Chen, Hongxiang Li, Zhen Wang, Long Chen

💡 개요

본 논문은 고품질 생성 모델인 Flow Matching(FM) 모델의 주요 병목 현상인 느린 ODE 풀이 속도를 개선하기 위해 Bi-Anchor Interpolation Solver (BA-solver)를 제안합니다. BA-solver는 경량의 SideNet을 사용하여 두 개의 앵커 속도를 기반으로 중간 속도를 효율적으로 근사하여, 기존 훈련 불가 솔버의 성능 저하와 훈련 기반 방법의 높은 훈련 비용 문제를 해결합니다. 이를 통해 적은 Neural Function Evaluations (NFEs)에서도 높은 생성 품질을 달성하며 기존 파이프라인과의 호환성을 보장합니다.

🔑 시사점 및 한계

•

기존 Flow Matching 모델의 느린 생성 속도를 획기적으로 개선하여 실시간 생성에 가까운 속도를 제공합니다.

•

적은 NFE에서도 높은 생성 품질을 유지하므로, 계산 자원이 제한적인 환경에서도 효과적으로 활용될 수 있습니다.

•

기존 FM 모델에 쉽게 통합될 수 있어 이미지 편집 등 다양한 다운스트림 작업에 대한 적용 가능성을 높입니다.

•

SideNet의 학습 및 앵커 속도 간의 최적화된 보간 방식 설계에 대한 추가 연구가 필요할 수 있습니다.

PDF 보기

Made with Slashpage