조건부 흐름 모델의 OOD 일반화를 위한 혼합 기반 최단경로 흐름 매칭

초록

SP‑FM은 조건에 따라 가변적인 가우시안 혼합 기반을 학습하고, 이를 최단경로 흐름 매칭과 결합해 조건부 생성 모델의 미지 조건 일반화를 크게 향상시킨다. 고정된 베이스 분포가 초래하는 학습 한계를 이론적으로 분석하고, 단일 가우시안 대비 다중 혼합이 최적 수송 문제를 잘 정의함을 증명한다. 단일세포 전사체와 고해상도 현미경 이미지 등 8개 데이터셋에서 기존 조건부 흐름 매칭 대비 일관된 성능 개선을 보여준다.

상세 분석

본 논문은 조건부 생성 모델이 미지의 조건(예: 새로운 약물, 회전 각도, 유전자 교란)에서 급격히 성능이 저하되는 근본 원인을 “베이스 분포 고정”이라는 설계 선택에 둔다. 기존 조건부 흐름 매칭은 표준 정규분포 N(0,I)를 모든 조건에 대해 동일하게 사용하고, 조건 y 만을 입력으로 받아 속도장 v 을 조정한다. 이는 각 조건을 독립적인 매핑 문제로 전환시켜, 새로운 조건이 등장했을 때 이전 조건과의 구조적 유사성을 활용할 여지를 없앤다.

SP‑FM은 이 문제를 두 단계로 해결한다. 첫째, 조건 y 에 따라 가변적인 가우시안 혼합 µ(y) 을 예측한다. 혼합 가중치, 평균, 공분산을 조건 인코더가 직접 출력하도록 설계했으며, 이를 통해 “조건 간 거리”가 베이스 분포의 거리와 직접 연결된다. 둘째, 조건부 속도장 vθ(x,t;y) 을 최단경로 흐름 매칭(Shortest‑Path Flow Matching) 방식으로 학습한다. 최단경로는 베이스 µ(y)와 목표 ρ(y)  사이의 최적 수송(OT) 경로를 의미하며, 이는 기존 흐름 매칭에서 사용되는 선형 보간보다 더 짧고 효율적인 경로를 제공한다.

이론적 분석에서는 베이스가 단일 가우시안일 때 OT의 쌍대문제가 불완전해져 해가 무수히 많아짐을 보이고, 이는 모델이 조건에 대한 일반화 능력을 상실하는 원인이라고 설명한다. 반면, K개의 혼합 컴포넌트를 도입하면 자유도가 충분히 늘어나 쌍대문제가 잘 정의되고, 베이스 복잡도 K, 목표 복잡도, 데이터 차원 D  사이에 오류 상한을 제시한다. 이러한 결과는 “조건 유사도 → 베이스 유사도 → 짧은 흐름”이라는 직관적 메커니즘을 수학적으로 뒷받침한다.

실험에서는 (1) 회전된 알파벳 이미지, (2) 고해상도 약물 처리 현미경 이미지, (3) 단일세포 전사체 데이터 등 8개의 벤치마크에서 기존 조건부 흐름 매칭, 조건부 확산 모델, 메타 흐름 매칭 등을 동일한 파라미터와 학습 예산으로 비교한다. 모든 경우에서 SP‑FM은 미지 조건에 대한 프리시전·리콜·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전·프레시전