← Search

Yun He

7 accepted papers

2026

Generalized Parallel Scaling with Interdependent Generations

ICLR 2026poster

Parallel LLM inference scaling involves sampling a set of $N>1$ responses for a single input prompt. However, these $N$ parallel responses tend to be generated independently from each other, partitioning compute resources and leaving potentially useful information in one generation untapped by other…

Cited by 0SourceScholar
2026

Parallel-Probe: Towards Efficient Parallel Thinking via 2D Probing

ICML 2026poster

Parallel thinking has emerged as a promising paradigm for reasoning, yet it imposes significant computational burdens. Existing efficiency methods primarily rely on local, per-trajectory signals and lack principled mechanisms to exploit global dynamics across parallel branches. We introduce 2D probi…

Cited by 0SourceScholar
2025

Think Smarter not Harder: Adaptive Reasoning with Inference Aware Optimization

ICML 2025poster

Solving mathematics problems has been an intriguing capability of large language models, and many efforts have been made to improve reasoning by extending reasoning length, such as through self-correction and extensive long chain-of-thoughts. While promising in problem-solving, advanced long reasoni…

Cited by 7SourcePDFScholar
2023

Grad-PU: Arbitrary-Scale Point Cloud Upsampling via Gradient Descent With Learned Distance Functions

CVPR 2023poster

Most existing point cloud upsampling methods have roughly three steps: feature extraction, feature expansion and 3D coordinate prediction. However, they usually suffer from two critical issues: (1) fixed upsampling rate after one-time training, since the feature expansion unit is customized for each…

2023

PromptAttack: Probing Dialogue State Trackers with Adversarial Prompts

ACL 2023findings

A key component of modern conversational systems is the Dialogue State Tracker (or DST), which models a user’s goals and needs. Toward building more robust and reliable DSTs, we introduce a prompt-based learning approach to automatically generate effective adversarial examples to probe DST models. T…

2022

Density-Preserving Deep Point Cloud Compression

CVPR 2022poster

Local density of point clouds is crucial for representing local details, but has been overlooked by existing point cloud compression methods. To address this, we propose a novel deep point cloud compression method that preserves local density information. Our method works in an auto-encoder fashion:…

Cited by 70PDFcodeScholar
2022

HyperPrompt: Prompt-based Task-Conditioning of Transformers

ICML 2022spotlight

Prompt-Tuning is a new paradigm for finetuning pre-trained language models in a parameter efficient way. Here, we explore the use of HyperNetworks to generate hyper-prompts: we propose HyperPrompt, a novel architecture for prompt-based task-conditioning of self-attention in Transformers. The hyper-p…