VideoRAE: 표현 오토인코더를 활용한 비디오 파운데이션 모델 제어로 생성 모델링

본 논문은 전통적인 3차원 변분 오토인코더가 의미론적 및 시공간 구조를 포착하는 데 가진 한계를 극복하도록 설계된 새로운 표현 오토인코더 VideoRAE를 소개합니다. VideoRAE는 V-JEPA 2와 VideoMAEv2와 같은 동결된 비디오 파운데이션 모델의 다중 스케일 계층적 특징을 활용하여, 경량 1D 자기주목 projection 레이어를 통해 컴팩트한 잠재 공간으로 압축합니다. 이 압축된 표현은 Diffusion Transformers를 위한 연속 잠재 변수 생성을 위한 다차원 코드북 양자화나, 자기회귀 비디오 모델을 위한 이산 토큰 생성에 활용될 수 있습니다. 실험 결과는 VideoRAE가 UCF-101 데이터셋에서 최첨단 생성 품질을 달성하며, 전통적 베이스라인 방법 대비 수렴 속도가 5배 빠름을 보여줍니다. 20억 파라미터 규모의 텍스트-비디오 생성 작업에서는 LTX-VAE를 VideoRAE로 대체함으로써 더 빠른 수렴과 비교 가능한 생성 품질을 얻을 수 있습니다. 이러한 결과는 사전학습된 비디오 파운데이션 모델의 표현이 비디오 생성에 친화적인 보편적 잠재 공간으로 작용하며, 다양한 생성 모델링 작업에 대해 표현을 처음부터 학습할 필요 없는 통합 프레임워크를 제공한다는 가설을 검증합니다.

배경

비디오 생성 모델의 발전은 오랫동안 3차원 변분 오토인코더(3D-VAEs)가 학습한 잠재 공간에 의존해 왔습니다. 그러나 기존의 3D-VAE는 주로 픽셀 수준의 재구성에 최적화되어 있어, 자연 비디오 데이터에 내재된 고차원 의미 정보와 복잡한 시공간 구조를 포착하는 데 한계가 있었습니다. 이는 잠재 공간이 낮은 수준의 충실도에는 효과적이지만, 고품질 의미 생성에 필요한 견고함을 갖추지 못하게 하는 병목 현상을 초래했습니다. 산업계가 더 복잡한 비디오 합성 작업으로 나아감에 따라, 이러한 전통적인 인코더가 의미적 무결성을 보존하는 데 가진 부족함이 중요한 기술적 장벽으로 대두되었습니다.

동시에 V-JEPA 2와 VideoMAEv2와 같은 새로운 세대의 비디오 파운데이션 모델(VFMs)이 등장하여 비디오 이해 작업에서 탁월한 성능을 입증했습니다. 이러한 모델들은 방대한 데이터셋을 학습하여 저수준의 가장자리부터 고수준의 의미 개념에 이르기까지 풍부한 계층적 특징을 추출합니다. 그러나 기존 문헌과 엔지니어링 관행 사이에는 중요한 격차가 존재했는데, 바로 이러한 강력한 VFMs의 동결된 사전 학습 표현을 컴팩트하고 재구현 가능하며 생성 친화적인 잠재 공간으로 효과적으로 변환할 수 있는지 여부가 충분히 탐구되지 않았다는 점입니다. prevailing assumption은 생성 모델이 전역적으로 학습된 전용 인코더를 필요로 한다는 것이었지만, VideoRAE는 이러한 통념에 도전하며 사전 학습된 파운데이션 모델의 표현을 재사용하는 가능성을 제시합니다.

심층 분석

VideoRAE의 기술적 아키텍처는 동결된 비디오 파운데이션 모델의 활용을 극대화하면서 계산 오버헤드를 최소화하도록 설계되었습니다. 이 과정은 V-JEPA 2나 VideoMAEv2와 같은 강력한 VFM을 교사 인코더로 동결하는 것부터 시작됩니다. 이 인코더는 입력 비디오에서 기본 시각 원시 요소부터 복잡한 의미 구조에 이르기까지 포괄적인 정보를 포착하는 다중 스케일 계층적 특징을 추출합니다. 이러한 특징을 관리 가능한 잠재 공간으로 압축하기 위해 VideoRAE는 경량 1D 자기주목(projection) 레이어를 사용합니다. 이 설계 선택은 고차원 시공간 데이터를 처리하는 데 따른 계산 복잡성을 크게 줄이면서도 특징 간의 시간적 의존성을 유지하는 데 핵심적입니다.

VideoRAE의 주요 혁신 중 하나는 지역적 및 전역적 표현 정렬 목표를 활용하는 디코딩 메커니즘입니다. KL 발산 정규화에 의존하는 전통적인 VAE와 달리, VideoRAE는 디코더가 동결된 VFM 교사 특징과 의미적으로 일관된 특징을 재구성하도록 강제합니다. 이러한 정렬 메커니즘은 전통적인 VAE에서 흔히 발생하는 사후 붕괴(posterior collapse) 문제를 방지하고, 생성된 잠재 변수가 높은 의미 충실도를 유지하도록 보장합니다. 또한 모델은 다차원 코드북을 사용한 고차원 양자화 기술을 통합하여, 디퓨전 기반 생성을 위한 연속 잠재 공간과 자기회귀 생성을 위한 이산 토큰 공간 사이를 유연하게 전환할 수 있습니다.

이러한 이중 지원 능력은 비디오 생성 분야에서 서로 다른 아키텍처가 종종 서로 호환되지 않는 잠재 공간을 필요로 했던 주요 분열을 해결합니다. VideoRAE는 두 가지 모두에 작동하는 단일 통합 표현을 제공함으로써 개발 파이프라인을 단순화하고 다양한 생성 구성 요소의 상호 운용성을 향상시킵니다. 디퓨전 트랜스포머의 경우, 모델은 디노이징 과정을 높은 의미 정확도로 안내하는 연속 잠재 변수를 생성합니다. 반면 자기회귀 모델의 경우, 견고한 문맥 인식을 갖춘 다음 프레임 예측을 가능하게 하는 이산 토큰을 생성합니다.

산업 영향

VideoRAE의 실험 결과는 기존 방법 대비 그 우월성을 설득력 있게 입증합니다. UCF-101 비디오 생성 데이터셋 벤치마크에서 VideoRAE는 최첨단 생성 품질 지표를 달성했습니다. 구체적으로, 자기회귀(AR) 생성기와 결합했을 때 gFVD 점수 40을 기록했으며, 디퓨전 트랜스포머(DiT) 생성기와의 결합에서는 93의 점수를 얻었습니다. 이러한 수치는 현재 최고 수준의 성능을 나타내며, VideoRAE가 학습한 표현이 두 가지 생성 패러다임 모두에 매우 효과적임을 시사합니다. 더 중요한 것은 VideoRAE의 수렴 속도가 전통적인 베이스라인 오토인코더보다 약 5배 빠르다는 점입니다. 이는 연구자와 개발자에게 직접적인 계산 비용 절감과 빠른 반복 주기로 이어지는 훈련 효율성의 극적인 개선을 의미합니다.

보다 복잡하고 대규모인 애플리케이션에서 VideoRAE의 이점은 더욱 두드러집니다. 20억 파라미터 규모의 텍스트-비디오 생성 작업을 포함한 통제된 실험에서 연구자들은 전통적인 LTX-VAE를 VideoRAE로 교체했습니다. 그 결과, VideoRAE를 사용한 모델이 비교 가능한, 아니 더 우수한 생성 품질을 유지하면서 훨씬 더 빠르게 수렴하는 것으로 나타났습니다. 이는 기존 대규모 비디오 생성 모델이 VideoRAE를 통해 아키텍처의 완전한 재설계 없이도 성능을 향상시킬 수 있음을 시사하므로 산업계에 특히 중요합니다. 레거시 구성 요소를 VideoRAE로 교체할 수 있는 능력은 비디오 생성 능력을 향상시키려는 기업들에게 낮은 리스크와 높은 보상을 제공하는 경로를 제공합니다.

아블레이션 연구는 VideoRAE 성공의 핵심 구성 요소를 다시 한번 강조합니다. 지역적 및 전역적 표현 정렬 목표를 제거하면 의미적 일관성이 눈에 띄게 저하되어 고품질 출력을 유지하는 데 그 중요성이 부각되었습니다. 마찬가지로 1D 자기주목 projection 레이어는 압축 효율성과 정보 보존 사이의 균형을 맞추는 데 필수적인 것으로 확인되었습니다. 이러한 통찰력은 동결된 사전 학습 비디오 파운데이션 모델 표현이 보편적이고 생성 친화적인 잠재 공간으로 작용할 수 있다는 가설을 검증합니다.

전망

VideoRAE의 도입은 독립적이고 작업 특화적인 훈련에서 강력한 파운데이션 모델 표현의 재사용으로의 패러다임 전환을 의미하며, 비디오 생성 기술에 중대한 전환점을 마련했습니다. 이는 비디오 합성의 미래에 깊은 영향을 미칩니다. 동결된 VFM 특징이 생성을 위해 효과적으로 재사용될 수 있음을 입증함으로써, VideoRAE는 사전 학습된 모델에 내장된 방대한 지식을 활용하는 새로운 세대의 생성 아키텍처의 문을 엽니다. 이 접근법은 고품질 비디오 생성기 개발의 진입 장벽을 낮출 뿐만 아니라, 연구자들이 처음부터 시작하는 대신 확립된 파운데이션 모델 위에 구축할 수 있게 함으로써 혁신의 속도를 가속화합니다.

앞으로 연속 및 이산 잠재 공간에 대한 이중 지원은 VideoRAE를 진화하는 비디오 생성 환경에서 다재다능한 도구로 위치시킵니다. 산업계가 비디오 합성의 한계를 계속 탐구함에 따라 디퓨전과 자기회귀 프레임워크 모두와 원활하게 통합할 수 있는 능력은 점점 더 중요해질 것입니다. VideoRAE의 효율성과 품질 개선은 빠른 프로토타이핑과 높은 충실도 출력이 필요한 애플리케이션에서 비디오 생성 개발자의 도구 모음에서 표준 구성 요소가 될 것임을 시사합니다. 그 영향은 콘텐츠 제작 및 엔터테인먼트부터 시각 효과 및 시뮬레이션과 같은 특수 분야에 이르기까지 다양한 부문에서 느껴질 것입니다.

더 나아가 VideoRAE의 성공은 생성형 AI의 다른 분야에서도 유사한 접근 방식을 영감 줄 수 있습니다. 동결하고 파운데이션 모델 표현을 생성을 위해 재사용하는 원칙은 이미지, 오디오 및 멀티모달 합성으로 확장될 수 있습니다. 대규모 모델 학습에 있어 컴퓨팅 자원이 제한 요소가 됨에 따라 기존 사전 학습 표현을 활용하는 능력은 중요한 경쟁 우위가 될 것입니다. VideoRAE는 기존 모델의 더 스마트하고 효율적인 사용이 우수한 결과를 낳을 수 있음을 보여주며, 비디오 생성 공간에서 지속 가능하고 확장 가능한 AI 개발을 위한 새로운 기준을 설정합니다.

Sources