Boost-and-Skip: A Simple Guidance-Free Diffusion for Minority Generation
Created by
Haebom
저자
Soobin Um, Beomsu Kim, Jong Chul Ye
개요
소수 샘플은 데이터 다양체의 저밀도 영역에 위치한 과소 표현된 인스턴스이며, 데이터 증강, 창의적인 콘텐츠 생성 등 많은 생성형 AI 애플리케이션에서 중요한 역할을 합니다. 기존의 확산 기반 소수 샘플 생성기는 소수 샘플 생성을 위한 계산 비용이 높은 안내(guidance)에 의존하는 경우가 많습니다. 이를 해결하기 위해 본 논문에서는 확산 모델을 사용하여 소수 샘플을 생성하는 간단하지만 강력한 안내 없는(guidance-free) 접근 방식인 Boost-and-Skip을 제시합니다. 이 프레임워크의 주요 장점은 표준 생성 프로세스에 두 가지 최소한의 변경만 필요하다는 것입니다. (i) 분산 강화된 초기화 (ii) 타임스텝 건너뛰기. 이러한 사소해 보이는 수정이 이론적 및 경험적 증거에 의해 뒷받침되어 과소 표현된 소수 특징의 출현을 효과적으로 촉진합니다. 광범위한 실험을 통해 Boost-and-Skip이 소수 샘플 생성 능력을 크게 향상시키고, 계산량이 훨씬 적으면서도 안내 기반 최첨단 접근 방식에 필적함을 보여줍니다. 코드는 https://github.com/soobin-um/BnS 에서 이용 가능합니다.