Bag of Design Choices for Inference of High-Resolution Masked Generative Transformer

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Shao, Shitong, Zhou, Zikai, Ye, Tian, Bai, Lichen, Xu, Zhiqiang, Xie, Zeke
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913709371162624
author Shao, Shitong
Zhou, Zikai
Ye, Tian
Bai, Lichen
Xu, Zhiqiang
Xie, Zeke
author_facet Shao, Shitong
Zhou, Zikai
Ye, Tian
Bai, Lichen
Xu, Zhiqiang
Xie, Zeke
contents Text-to-image diffusion models (DMs) develop at an unprecedented pace, supported by thorough theoretical exploration and empirical analysis. Unfortunately, the discrepancy between DMs and autoregressive models (ARMs) complicates the path toward achieving the goal of unified vision and language generation. Recently, the masked generative Transformer (MGT) serves as a promising intermediary between DM and ARM by predicting randomly masked image tokens (i.e., masked image modeling), combining the efficiency of DM with the discrete token nature of ARM. However, we find that the comprehensive analyses regarding the inference for MGT are virtually non-existent, and thus we aim to present positive design choices to fill this gap. We propose and redesign a set of enhanced inference techniques tailored for MGT, providing a detailed analysis of their performance. Additionally, we explore several DM-based approaches aimed at accelerating the sampling process on MGT. Extensive experiments and empirical analyses on the recent SOTA MGT, such as MaskGIT and Meissonic lead to concrete and effective design choices, and these design choices can be merged to achieve further performance gains. For instance, in terms of enhanced inference, we achieve winning rates of approximately 70% compared to vanilla sampling on HPS v2 with Meissonic-1024x1024.
format Preprint
id arxiv_https___arxiv_org_abs_2411_10781
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bag of Design Choices for Inference of High-Resolution Masked Generative Transformer
Shao, Shitong
Zhou, Zikai
Ye, Tian
Bai, Lichen
Xu, Zhiqiang
Xie, Zeke
Computer Vision and Pattern Recognition
Machine Learning
Text-to-image diffusion models (DMs) develop at an unprecedented pace, supported by thorough theoretical exploration and empirical analysis. Unfortunately, the discrepancy between DMs and autoregressive models (ARMs) complicates the path toward achieving the goal of unified vision and language generation. Recently, the masked generative Transformer (MGT) serves as a promising intermediary between DM and ARM by predicting randomly masked image tokens (i.e., masked image modeling), combining the efficiency of DM with the discrete token nature of ARM. However, we find that the comprehensive analyses regarding the inference for MGT are virtually non-existent, and thus we aim to present positive design choices to fill this gap. We propose and redesign a set of enhanced inference techniques tailored for MGT, providing a detailed analysis of their performance. Additionally, we explore several DM-based approaches aimed at accelerating the sampling process on MGT. Extensive experiments and empirical analyses on the recent SOTA MGT, such as MaskGIT and Meissonic lead to concrete and effective design choices, and these design choices can be merged to achieve further performance gains. For instance, in terms of enhanced inference, we achieve winning rates of approximately 70% compared to vanilla sampling on HPS v2 with Meissonic-1024x1024.
title Bag of Design Choices for Inference of High-Resolution Masked Generative Transformer
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2411.10781