Daily Arxiv

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

Hita: Holistic Tokenizer for Autoregressive Image Generation

Created by
  • Haebom

저자

Anlin Zheng, Haochen Wang, Yucheng Zhao, Weipeng Deng, Tiancai Wang, Xiangyu Zhang, Xiaojuan Qi

개요

Hita는 기존 자기회귀 이미지 생성 모델의 한계를 극복하기 위해 제안된 새로운 이미지 토크나이저입니다. 기존 모델들이 지역적 패치 기반의 토크나이징으로 전역 정보를 제한적으로 활용하는 문제를 해결하기 위해, 학습 가능한 전역 쿼리와 지역 패치 토큰을 사용하는 전역-지역 토크나이징 기법을 도입했습니다. 전역 토큰을 먼저 배치하고, 인과적 어텐션을 사용하여 이전 토큰 정보를 유지하며, 가벼운 융합 모듈을 통해 전역 토큰 정보의 우선순위를 제어함으로써 자기회귀 생성 과정과의 정합성을 높였습니다. ImageNet 벤치마크에서 FID 2.59, IS 281.9를 달성하며 기존 토크나이저 기반 모델보다 성능이 우수함을 보였고, 훈련 속도 또한 향상시켰습니다. 전역 표현은 질감, 재질, 형태와 같은 전역 이미지 특성을 포착하는 능력을 보였으며, 제로샷 스타일 전이 및 이미지 인페인팅에서도 효과적임을 확인했습니다. 코드는 깃허브에 공개되어 있습니다.

시사점, 한계점

시사점:
자기회귀 이미지 생성 모델의 성능 향상: ImageNet에서 SOTA 성능 달성 (FID 2.59, IS 281.9).
훈련 속도 향상.
전역 정보 활용을 통한 이미지 생성 품질 개선 (질감, 재질, 형태 등).
제로샷 스타일 전이 및 이미지 인페인팅과 같은 다양한 작업에 대한 효과적인 적용 가능성.
전역-지역 토크나이징 기법의 효용성 제시.
한계점:
논문에서 구체적인 한계점이 명시적으로 언급되지 않음. 향후 연구를 통해 추가적인 개선 여지가 있을 것으로 예상됨.
👍