paper-with-me

홈 › Papers

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

2024-11-16 · Shitong Shao, Zikai Zhou, Tian Ye, Lichen Bai, Zhiqiang Xu, Zeke Xie

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 modify and re-design a set of DM-based inference techniques for MGT and further elucidate their performance on MGT. We also discuss the approach to correcting token's distribution to enhance inference. Extensive experiments and empirical analyses 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 the recent SOTA MGT Meissonic. Our contributions have the potential to further enhance the capabilities and future development of MGTs.

📄 PDF Abstract BibTeX arXiv:2411.10781

Code (1)

xie-lab-ml/Meissonic-Inference 공식 구현 jax

Tasks

Text Generation

Methods 이 논문이 사용한 방법론

Attention 설명 없음
Adam 설명 없음
Residual Connection 설명 없음
BPE Byte Pair Encoding, or BPE, is a subword segmentation algorithm that encodes rare and unknown words as sequences of subword units. The intuition is that various word…
Linear Layer A Linear Layer is a projection $\mathbf{XW + b}$.
SET Dynamic Sparse Training method where weight mask is updated randomly periodically
Absolute Position Encodings Absolute Position Encodings are a type of position embeddings for [Transformer-based models] where positional encodings are…
Multi-Head Attention 설명 없음

Similar Papers 제목 키워드 기반

Customize Your Visual Autoregressive Recipe with Set Autoregressive Modeling

2024-10-14 · Wenze Liu, Le Zhuo, Yi Xin, Sheng Xia 외

We introduce a new paradigm for AutoRegressive (AR) image generation, termed Set AutoRegressive Modeling (SAR). SAR generalizes the conventional AR to the next-set setting, i.e., splitting the sequence into arbitrary set…

Image Generation

MarDini: Masked Autoregressive Diffusion for Video Generation at Scale

2024-10-26 · Haozhe Liu, Shikun Liu, Zijian Zhou, Mengmeng Xu 외

We introduce MarDini, a new family of video diffusion models that integrate the advantages of masked auto-regression (MAR) into a unified diffusion model (DM) framework. Here, MAR handles temporal planning, while DM focu…

Image to Video GenerationVideo Generation

Exploring Prediction Targets in Masked Pre-Training for Speech Foundation Models

2024-09-16 · Li-Wei Chen, Takuya Higuchi, He Bai, Ahmed Hussen Abdelaziz 외

Speech foundation models, such as HuBERT and its variants, are pre-trained on large amounts of unlabeled speech data and then used for a range of downstream tasks. These models use a masked prediction objective, where th…

DenoisingPrediction

HMAR: Efficient Hierarchical Masked Auto-Regressive Image Generation

2025-06-04 · CVPR 2025 1 · Hermann Kumbong, Xian Liu, Tsung-Yi Lin, Ming-Yu Liu 외

Visual Auto-Regressive modeling (VAR) has shown promise in bridging the speed and quality gap between autoregressive image models and diffusion models. VAR reformulates autoregressive modeling by decomposing an image int…

AllImage GenerationPrediction

Characterizing and Taming Resolution in Convolutional Neural Networks

2021-10-28 · Eddie Yan, Liang Luo, Luis Ceze

Image resolution has a significant effect on the accuracy and computational, storage, and bandwidth costs of computer vision model inference. These costs are exacerbated when scaling out models to large inference serving…