저자들은 확산 및 흐름 일치 샘플러에서 큰 가이던스 스케일에서 발생하는 과포화 및 불안정성 문제를 해결하기 위해 분류자 없는 가이던스(CFG)를 위한 터미널 피팅 복원을 제안합니다. 점근적 보존 관점에서 CFG를 분석함으로써, 표준 가이던스가 최소 시그마가 0에 접근할 때 결정론적 DDIM 단계가 발산함을 보여줍니다.

  • 복원은 가이던스 방향에서 표준 CFG 항 w(r-1)을 r^(1+w)-r로 대체하여 하나의 계수와 추가 네트워크 함수 평가 없이도 작동합니다.
  • 이 수정은 sigma_min 발산 폭발을 제거하고 sigma_min이 0으로 갈 때 정확한 유도 흐름에 대해 1차 정확도를 달성합니다.
  • 학습된 CIFAR-10 체크포인트 및 Stable Diffusion 1.5 DDIM에서 이 방법은 잔류 증폭과 포화를 줄이는 고가이던스 안정화제 역할을 합니다.
  • 이 접근법은 테스트된 그리드에서 표준 CFG보다 point-FID 성능을 향상시키면서 hard-cell 블록에서 분류자 프록시 대상 정확도를 유지합니다.

저자들은 이 복원이 보편적인 이미지 품질 향상이 아니라 추가 계산 비용 없이 고가이던스 샘플링을 안정화하는 데 특화되어 있다고 언급합니다.