Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Fast Autoregressive Video Generation with Diagonal Decoding

Created by
  • Haebom
Category
Empty

저자

Yang Ye, Junliang Guo, Haoyu Wu, Tianyu He, Tim Pearce, Tabish Rashid, Katja Hofmann, Jiang Bian

개요

자기회귀 변환기 모델은 비디오 생성에서 인상적인 성능을 보여주었지만, 토큰 단위의 순차적 디코딩 과정은 수만 개의 토큰으로 표현되는 긴 비디오의 경우 주요 병목 현상을 야기합니다. 본 논문에서는 비디오의 공간적 및 시간적 상관관계를 활용하는 학습이 필요 없는 자기회귀 사전 훈련 모델을 위한 추론 가속 알고리즘인 대각선 디코딩(DiagD)을 제안합니다. DiagD는 공간-시간 토큰 그리드에서 대각선 경로를 따라 토큰을 생성하여 각 프레임 내에서 병렬 디코딩을 가능하게 하고, 연속적인 프레임 간에 부분적으로 중복되도록 합니다. 제안된 알고리즘은 다양한 생성 모델과 작업에 적용 가능하며, 추론 속도와 화질 간의 절충을 유연하게 제어할 수 있습니다. 또한, 소규모 모델에서의 학습-추론 간격을 더욱 완화하기 위해 모델의 어텐션 패턴을 디코딩 순서와 정렬하는 비용 효율적인 미세 조정 전략을 제안합니다. 여러 자기회귀 비디오 생성 모델과 데이터셋에 대한 실험 결과, DiagD는 일반적인 순차적 디코딩에 비해 최대 10배의 속도 향상을 달성하면서 비슷한 화질을 유지함을 보여줍니다.

시사점, 한계점

시사점:
자기회귀 비디오 생성 모델의 추론 속도를 최대 10배까지 향상시킬 수 있는 효과적인 알고리즘(DiagD)을 제시.
다양한 모델과 작업에 적용 가능한 범용적인 알고리즘.
추론 속도와 화질 간의 유연한 절충 제어 가능.
비용 효율적인 미세 조정 전략을 통해 학습-추론 간격 완화.
한계점:
제안된 알고리즘의 성능은 사용하는 모델과 데이터셋에 따라 달라질 수 있음.
대규모 모델에 대한 성능 평가가 부족할 수 있음. (논문에서 언급된 소규모 모델에 대한 미세조정 전략의 효과는 대규모 모델에서는 다를 수 있음)
DiagD의 효율성은 비디오의 특징(예: 프레임 크기, 길이)에 따라 영향을 받을 수 있음.
👍