← Search

Kunyu Shi

4 accepted papers

2026

Beyond Scattered Acceptance: Fast and Coherent Inference for DLMs via Longest Stable Prefixes

ICLR 2026poster

Diffusion Language Models (DLMs) promise parallel generation via iterative denoising, yet their practical speed is often throttled by \emph{schedulers} that accept scattered high-confidence tokens, fragmenting KV caches and forcing repeated local repairs. We present \emph{Prefix Absorption}, a train…

Cited by 0SourceScholar
2024

Enhancing Vision-Language Pre-training with Rich Supervisions

CVPR 2024highlight

We propose Strongly Supervised pre-training with ScreenShots (S4) - a novel pre-training paradigm for Vision-Language Models using data from large-scale web screenshot rendering. Using web screenshots unlocks a treasure trove of visual and textual cues that are not present in using image-text pairs.…

Cited by 10SourcePDFScholar
2024

Non-autoregressive Sequence-to-Sequence Vision-Language Models

CVPR 2024poster

Sequence-to-sequence vision-language models are showing promise but their applicability is limited by their inference latency due to their autoregressive way of generating predictions. We propose a parallel decoding sequence-to-sequence vision-language model trained with a Query-CTC loss that margin…