← Search

Yukang Yang

7 accepted papers

2026

Mixture of States: Routing Token-Level Dynamics for Multimodal Generation

CVPR 2026

We introduce MoS (Mixture of States), a novel fusion paradigm for multimodal diffusion models that merges modalities using flexible, state-based interactions. The core of MoS is a learnable, token-wise router that creates denoising timestep- and input-dependent interactions between modalities' hidde

Cited by 0SourcecodeScholar
2025

Causal Head Gating: A Framework for Interpreting Roles of Attention Heads in Transformers

NeurIPS 2025poster

We present causal head gating (CHG), a scalable method for interpreting the functional roles of attention heads in transformer models. CHG learns soft gates over heads and assigns them a causal taxonomy—facilitating, interfering, or irrelevant—based on their impact on task performance. Unlike prior…

Cited by 0SourceScholar
2025

Emergent Symbolic Mechanisms Support Abstract Reasoning in Large Language Models

ICML 2025poster

Many recent studies have found evidence for emergent reasoning capabilities in large language models (LLMs), but debate persists concerning the robustness of these capabilities, and the extent to which they depend on structured reasoning mechanisms. To shed light on these issues, we study the intern…

2025

Training-Free Guidance Beyond Differentiability: Scalable Path Steering with Tree Search in Diffusion and Flow Models

NeurIPS 2025poster

Training-free guidance enables controlled generation in diffusion and flow models, but most methods rely on gradients and assume differentiable objectives. This work focuses on training-free guidance addressing challenges from non-differentiable objectives and discrete data distributions. We propose…

Cited by 0SourceScholar
2024

Gradient Guidance for Diffusion Models: An Optimization Perspective

NeurIPS 2024poster

Diffusion models have demonstrated empirical successes in various applications and can be adapted to task-specific needs via guidance. This paper studies a form of gradient guidance for adapting a pre-trained diffusion model towards optimizing user-specified objectives. We establish a mathematical f…

2023

GlyphControl: Glyph Conditional Control for Visual Text Generation

NeurIPS 2023poster

Recently, there has been an increasing interest in developing diffusion-based text-to-image generative models capable of generating coherent and well-formed visual text. In this paper, we propose a novel and efficient approach called GlyphControl to address this task. Unlike existing methods that re…

2023

Rank-DETR for High Quality Object Detection

NeurIPS 2023poster

Modern detection transformers (DETRs) use a set of object queries to predict a list of bounding boxes, sort them by their classification confidence scores, and select the top-ranked predictions as the final detection results for the given input image. A highly performant object detector requires acc…