[3D Diffusion / GenAI] Flow Matching

 

Flow Matching

- 기존 Source 분포로부터 Target 분포로 데이터 분포를 학습하는 과정인 모델을 따르는 CNF (Continous Normalizing Flow)   
   모델을 개선한 것
   Flow Model

         - Source 분포를 Target 분포로 변환을 해줄 수 있는 모델을 의미함. ( diffeomorphism, 미분 가능, 역함수 존재 )

         - 하지만, Continous time 상황에서는 flow를 직접적으로 학습하기 어려움.

 

         - 이러한 문제를 해결하기 위해, velocity field 를 정의함. ( t에 따른 변화 ) 

            그러므로, velocity field를 정의하고, ODE를 통하여  Flow model을 찾아 활용하고자 함. ( Euler Solver) 

Flow Matching

- 기존 Source 분포로부터 Target 분포로 데이터 분포를 학습하는 과정인 모델을 따르는 CNF (Continous Normalizing Flow)   
   모델을 개선한 것

 

- Source 와 Target 분포를 이어지게 하고자, Probability path를 정의함.

이것을 활용하여 다음과 같이 알고리즘을 수행