한 줄 요약
SANA-Video 2.0은 5B 및 14B 규모의 하이브리드 어텐션 비디오 생성 모델로, 720p 영상 생성 시 3.2× 가속된 DiT 성능과 84.30의 VBench 점수를 달성한다.
핵심 기여도
- **Hybrid Linear-Softmax Attention**: 3:1 비율로 주기적으로 삽입된 softmax anchor를 통해 75% linear attention, 25% softmax attention을 결합.
- **Block Attention Residuals (AttnRes)**: 블록 요약 정보를 후속 linear layer로 전달하여 effective rank를 약 12% 향상.
- **From-scratch Training**: 25% softmax anchor 비율은 proxy study를 통해 결정, pretraining 없이 직접 학습.
- **Sol-Engine 최적화**: kernel fusion, caching, sparse attention을 통해 3.58×의 추가 가속.
핵심 아이디어
SANA-Video 2.0은 기존 full-softmax 어텐션의 O(N²) 비용 문제를 해결하기 위해 linear attention의 O(N) 확장성을 활용하면서도, 정확한 토큰 상호작용을 복원하기 위해 주기적으로 softmax anchor를 삽입하는 하이브리드 구조를 제안한다. 이는 3:1의 비율로 softmax anchor를 배치하여, linear attention의 효율성과 softmax의 표현력을 결합하는 방식이다. 또한, AttnRes를 통해 anchor에서 생성된 정보를 후속 레이어로 전달함으로써, deep layer의 표현력이 약 12% 향상된다는 점이 핵심이다. 이는 기존 linear attention이 토큰 간 상호작용을 제한적으로 표현한다는 한계를 극복하기 위한 전략이다.
기술적 접근법
- **Hybrid Linear-Softmax Attention**: 75% gated linear attention + 25% periodic softmax anchors (3:1 ratio).
- **Block Attention Residuals (AttnRes)**: 블록 요약 정보를 후속 linear layer로 전달하여 anchor 정보 재사용.
- **Model Architecture**: 5B 모델은 40-step sampling 시 480×832×81 해상도에서 13.2초에 VBench 84.30 달성.
- **Training**: 25% softmax anchor 비율은 proxy study를 통해 결정, full model은 multi-stage pipeline에서 from-scratch로 학습.
- **Optimization**: Sol-Engine을 통해 kernel fusion, caching, sparse attention 적용, 5B 모델은 720p/5s에서 13.06초 소요.
주요 결과
- **VBench 점수**: 40-step sampling 시 480p에서 84.30 (H100 기준 13.2초).
- **DiT Forward Speed**: 720p/60s에서 full-softmax 대비 3.2× 빠름.
- **Sol-Engine 최적화**: 5B 모델은 720p/5s에서 13.06초, Wan 2.2-A14B 대비 120× 빠름.
- **Effective Rank 향상**: AttnRes 사용 시 deep layer의 effective rank 약 12% 증가.
의의 및 한계
SANA-Video 2.0은 full-softmax 기반 모델과 유사한 생성 품질을 유지하면서도, O(N) 확장성을 통해 장시간, 고해상도 영상 생성에 유리한 성능을 보인다. 특히, 720p/60s에서 3.2×, 720p/5s에서 3.58×의 추가 가속을 달성한 Sol-Engine 최적화는 실용적 가치가 크다. 그러나, anchor 비율은 proxy study를 통해 결정되었으며, 가장 긴 영상 결과는 텐서 형태의 프로파일링에 의존한다는 점에서 한계가 있다. 또한, 모델의 최적화는 특정 하드웨어 (H100, B200)에 의존적이며, 다른 플랫폼에서의 성능은 추가 연구가 필요하다.
실용적 활용
SANA-Video 2.0은 단일 GPU에서 720p 영상 생성이 가능하며, 3.2× 가속된 DiT 성능을 통해 실시간 또는 저비용 영상 생성에 적합하다. 특히, Sol-Engine 최적화를 통해 120× 빠른 처리가 가능한 점은 클라우드 기반 영상 생성 서비스나 모바일 장치에서의 활용 가능성을 높인다. 또한, AttnRes와 하이브리드 어텐션 구조는 장시간 영상 생성 및 스트리밍 시스템에 적용 가능하다.