← Search

Yihan Wang

31 accepted papers

2026

Mantis: A Versatile Vision-Language-Action Model with Disentangled Visual Foresight

CVPR 2026

Recent advances in Vision-Language-Action (VLA) models demonstrate that visual signals can effectively complement sparse action supervisions. However, letting VLA directly predict high-dimensional visual states can distribute model capacity and incur prohibitive training cost, while compressing visu

Cited by 0SourcecodeScholar
2026

NavSpace: How Navigation Agents Follow Spatial Intelligence Instructions

ICRA 2026poster

Instruction-following navigation is a key step toward embodied intelligence. Prior benchmarks mainly focus on semantic understanding but overlook systematically evaluating navigation agents' spatial perception and reasoning capabilities. In this work, we introduce the NavSpace benchmark, which conta…

2026

ReWatch-R1: Boosting Complex Video Reasoning in Large Vision-Language Models through Agentic Data Synthesis

ICLR 2026poster

While Reinforcement Learning with Verifiable Reward (RLVR) significantly advances image reasoning in Large Vision-Language Models (LVLMs), its application to complex video reasoning remains underdeveloped. This gap stems primarily from a critical data bottleneck: existing datasets lack the challengi…

Cited by 0SourcecodeScholar
2026

ResMAS: Resilience Optimization in LLM-based Multi-agent Systems

AAAI 2026technical

Large Language Model-based Multi-Agent Systems (LLM-based MAS), where multiple LLM agents collaborate to solve complex tasks, have shown impressive performance in many areas. However, MAS are typically distributed across different devices or environments, making them vulnerable to perturbations such

Cited by 0SourcePDFScholar
2025

BridgePure: Limited Protection Leakage Can Break Black-Box Data Protection

NeurIPS 2025poster

Availability attacks, or unlearnable examples, are defensive techniques that allow data owners to modify their datasets in ways that prevent unauthorized machine learning models from learning effectively while maintaining the data's intended functionality. It has led to the release of popular black-…

Cited by 0SourceScholar
2025

LinkAlign: Scalable Schema Linking for Real-World Large-Scale Multi-Database Text-to-SQL

EMNLP 2025

Schema linking is a critical bottleneck in applying existing Text-to-SQL models to real-world, large-scale, multi-database environments. Through error analysis, we identify two major challenges in schema linking: (1) Database Retrieval: accurately selecting the target database from a large schema po

2025

On the Loss of Context Awareness in General Instruction Fine-tuning

NeurIPS 2025poster

Pre-trained Large Language Models (LLMs) require post-training methods such as supervised fine-tuning (SFT) on instruction-response pairs to enable instruction following. However, this process can cause forgetting in capabilities learned during pre-training. In this paper, we investigate the loss of…

Cited by 0SourcecodeScholar
2025

Provable Robust Overfitting Mitigation in Wasserstein Distributionally Robust Optimization

ICLR 2025poster

Wasserstein distributionally robust optimization (WDRO) optimizes against worst-case distributional shifts within a specified uncertainty set, leading to enhanced generalization on unseen adversarial examples, compared to standard adversarial training which focuses on pointwise adversarial perturbat…

2025

QiMeng-NeuComBack: Self-Evolving Translation from IR to Assembly Code

NeurIPS 2025poster

Compilers, while essential, are notoriously complex systems that demand prohibitively expensive human expertise to develop and maintain. The recent advancements in Large Language Models (LLMs) offer a compelling new paradigm: Neural Compilation, which could potentially simplify compiler development…

Cited by 0SourceScholar
2025

Structure-from-Motion with a Non-Parametric Camera Model

CVPR 2025highlight

In this paper, we present a new generic Structure-from-Motion pipeline, GenSfM, that uses a non-parametric camera projection model. The model is self-calibrated during the reconstruction process and can fit a wide variety of cameras, ranging from simple low-distortion pinhole cameras to more extreme…

2024

Defending LLMs against Jailbreaking Attacks via Backtranslation

ACL 2024findings

Although many large language models (LLMs) have been trained to refuse harmful requests, they are still vulnerable to jailbreaking attacks which rewrite the original prompt to conceal its harmful intent. In this paper, we propose a new method for defending LLMs against jailbreaking attacks by “backt…

2024

Efficient Availability Attacks against Supervised and Contrastive Learning Simultaneously

NeurIPS 2024poster

Availability attacks provide a tool to prevent the unauthorized use of private data and commercial datasets by generating imperceptible noise and crafting unlearnable examples before release. Ideally, the obtained unlearnability can prevent algorithms from training usable models. When supervised l…

2024

Two-stage LLM Fine-tuning with Less Specialization and More Generalization

ICLR 2024poster

Pretrained large language models (LLMs) are general purpose problem solvers applicable to a diverse set of tasks with prompts. They can be further improved towards a specific task by fine-tuning on a specialized dataset. However, fine-tuning usually makes the model narrowly specialized on this datas…

Cited by 6SourcePDFScholar
2023

Infinite Photorealistic Worlds Using Procedural Generation

CVPR 2023poster

We introduce Infinigen, a procedural generator of photorealistic 3D scenes of the natural world. Infinigen is entirely procedural: every asset, from shape to texture, is generated from scratch via randomized mathematical rules, using no external source and allowing infinite variation and composition…

2022

A Branch and Bound Framework for Stronger Adversarial Attacks of ReLU Networks

ICML 2022spotlight

Strong adversarial attacks are important for evaluating the true robustness of deep neural networks. Most existing attacks search in the input space, e.g., using gradient descent, and may miss adversarial examples due to non-convexity. In this work, we systematically search adversarial examples in t…

2022

Efficiently Computing Local Lipschitz Constants of Neural Networks via Bound Propagation

NeurIPS 2022accept

Lipschitz constants are connected to many properties of neural networks, such as robustness, fairness, and generalization. Existing methods for computing Lipschitz constants either produce relatively loose upper bounds or are limited to small networks. In this paper, we develop an efficient framewo…

2022

Lite Pose: Efficient Architecture Design for 2D Human Pose Estimation

CVPR 2022poster

Pose estimation plays a critical role in human-centered vision applications. However, it is difficult to deploy state-of-the-art HRNet-based pose estimation models on resource-constrained edge devices due to the high computational cost (more than 150 GMACs per frame). In this paper, we study efficie…

Cited by 139PDFcodeScholar
2022

On the Convergence of Certified Robust Training with Interval Bound Propagation

ICLR 2022poster

Interval Bound Propagation (IBP) is so far the base of state-of-the-art methods for training neural networks with certifiable robustness guarantees when potential adversarial perturbations present, while the convergence of IBP training remains unknown in existing literature. In this paper, we presen…

Cited by 10SourcePDFScholar
2022

RLPrompt: Optimizing Discrete Text Prompts with Reinforcement Learning

EMNLP 2022main

Prompting has shown impressive success in enabling large pre-trained language models (LMs) to perform diverse NLP tasks, especially with only few downstream data. Automatically finding the optimal prompt for each task, however, is challenging. Most existing work resorts to tuning *soft* prompts (e.g…

2021

DOP: Off-Policy Multi-Agent Decomposed Policy Gradients

ICLR 2021poster

Multi-agent policy gradient (MAPG) methods recently witness vigorous progress. However, there is a significant performance discrepancy between MAPG methods and state-of-the-art multi-agent value-based approaches. In this paper, we investigate causes that hinder the performance of MAPG algorithms and…

Cited by 158SourcePDFScholar
2021

Fast Certified Robust Training with Short Warmup

NeurIPS 2021poster

Recently, bound propagation based certified robust training methods have been proposed for training neural networks with certifiable robustness guarantees. Despite that state-of-the-art (SOTA) methods including interval bound propagation (IBP) and CROWN-IBP have per-batch training complexity similar…

2021

Fast and Complete: Enabling Complete Neural Network Verification with Rapid and Massively Parallel Incomplete Verifiers

ICLR 2021poster

Formal verification of neural networks (NNs) is a challenging and important problem. Existing efficient complete solvers typically require the branch-and-bound (BaB) process, which splits the problem domain into sub-domains and solves each sub-domain using faster but weaker incomplete verifiers, suc…

2020

Automatic Perturbation Analysis for Scalable Certified Robustness and Beyond

NeurIPS 2020poster

Linear relaxation based perturbation analysis (LiRPA) for neural networks, which computes provable linear bounds of output neurons given a certain amount of input perturbation, has become a core component in robustness verification and certified defense. The majority of LiRPA-based methods focus on…

2020

On Lp-norm Robustness of Ensemble Decision Stumps and Trees

ICML 2020poster

Recent papers have demonstrated that ensemble stumps and trees could be vulnerable to small input perturbations, so robustness verification and defense for those models have become an important research problem. However, due to the structure of decision trees, where each node makes decision purely b…