← Search

Yangchao Wu

3 accepted papers

2025

STree: Speculative Tree Decoding for Hybrid State Space Models

NeurIPS 2025poster

Speculative decoding is a technique to leverage hardware concurrency in order to enable multiple steps of token generation in a single forward pass, thus improving the efficiency of large-scale autoregressive (AR) Transformer models. State-space models (SSMs) are already more efficient than AR Trans…

Cited by 0SourcecodeScholar
2024

RSA: Resolving Scale Ambiguities in Monocular Depth Estimators through Language Descriptions

NeurIPS 2024poster

We propose a method for metric-scale monocular depth estimation. Inferring depth from a single image is an ill-posed problem due to the loss of scale from perspective projection during the image formation process. Any scale chosen is a bias, typically stemming from training on a dataset; hence, exis…

2024

Sub-token ViT Embedding via Stochastic Resonance Transformers

ICML 2024poster

Vision Transformer (ViT) architectures represent images as collections of high-dimensional vectorized tokens, each corresponding to a rectangular non-overlapping patch. This representation trades spatial granularity for embedding dimensionality, and results in semantically rich but spatially coarsel…