← Search

Ziyi Wang

48 accepted papers

2026

CoT Vectors: Transferring and Probing the Reasoning Mechanisms of LLMs

ICLR 2026poster

Chain-of-Thought (CoT) prompting has emerged as a powerful approach to enhancing the reasoning capabilities of Large Language Models (LLMs). However, existing implementations, such as in-context learning and fine-tuning, remain costly and inefficient. To improve CoT reasoning at a lower cost, and in…

Cited by 0SourceScholar
2026

DAWN: Distributed LLM Multi-Agent Workflow Synthesis

AAAI 2026technical

Large language models (LLMs) have recently empowered multi-agent systems (MAS) to achieve remarkable advances in collaborative reasoning and complex task automation. The effectiveness of these systems fundamentally depends on the design of adaptive communication graphs—the underlying workflows that

Cited by 0SourcePDFScholar
2026

EGG-SR: Embedding Symbolic Equivalence into Symbolic Regression via Equality Graph

ICLR 2026poster

Symbolic regression seeks to uncover physical laws from experimental data by searching for closed-form expressions, which is an important task in AI-driven scientific discovery. Yet the exponential growth of the search space of expression renders the task computationally challenging. A promising yet…

Cited by 0SourcecodeScholar
2026

Eigen-1: Scientific Reasoning through Adaptive Multi-Agent Refinement and Monitor-based RAG

ICLR 2026poster

Large language models (LLMs) have recently shown strong progress on scientific reasoning, yet two major bottlenecks remain. First, explicit retrieval fragments reasoning, imposing a hidden tool tax of extra tokens and steps. Second, multi-agent pipelines often dilute strong solutions by averaging ac…

Cited by 0SourcecodeScholar
2026

ExpoCM: Exposure-Aware One-Step Generative Single-Image HDR Reconstruction

CVPR 2026

Single-image HDR reconstruction aims to recover high dynamic range radiance from a single low dynamic range (LDR) input, but remains highly ill-posed due to detail saturation in over-exposed regions and noise amplification in under-exposed areas. While recent diffusion-based approaches offer powerfu

Cited by 0SourcecodeScholar
2026

Learning PDE Solvers with Physics and Data: A Unifying View of Physics-Informed Neural Networks and Neural Operators

IJCAI 2026

Partial differential equations (PDEs) are central to scientific modeling. Nowadays, modern workflows increasingly rely on learning-based components to support model reuse, inference, and integration across large computational processes. Despite the emergence of various physics-aware data-driven appr

Cited by 0Scholar
2026

MOC: Multi-Order Communication in LLM-based Multi-Agent Systems

ICML 2026poster

Despite the remarkable progress of Large Language Model (LLM) based Multi-Agent Systems, most research focuses on optimizing coordination topology while largely underexploring the equally critical problem: how to transmit and optimize messages among agents effectively? Current communication schemes …

Cited by 0SourceScholar
2026

MP1: MeanFlow Tames Policy Learning in 1-step for Robotic Manipulation

AAAI 2026technical

In robot manipulation, robot learning has become a prevailing approach. However, generative models within this field face a fundamental trade-off between the slow, iterative sampling of diffusion models and the architectural constraints of faster Flow-based methods, which often rely on explicit cons

Cited by 0SourcePDFScholar
2026

MindCopilot: Towards Formalizing and Evaluating Granular Human-LLM Co-Writing

IJCAI 2026

Recent writing assistants are increasingly shifting from passive, prompt-driven interaction to proactive, suggestion-based completion, which integrates localized continuations into the writing flow and reduces coordination burden. However, existing evaluations simply focus on output quality, failing

Cited by 0Scholar
2026

Mitigating Hallucinations in Large Language Models via Causal Reasoning

AAAI 2026technical

Large language models (LLMs) exhibit logically inconsistent hallucinations that appear coherent yet violate reasoning principles, with recent research suggesting an inverse relationship between causal reasoning capabilities and such hallucinations. However, existing reasoning approaches in LLMs, suc

Cited by 0SourcePDFScholar
2026

RAG-TP: A General Framework for Vehicle Trajectory Prediction via Retrieval-Augmented Generation

CVPR 2026

Vehicle trajectory prediction is critical for safe and efficient autonomous driving. However, its generalization and scalability are hindered by heavy reliance on real-time, online priors. To break this bottleneck, we introduce RAG-TP, a framework reframing the problem from relying on uncertain onli

Cited by 0SourceScholar
2026

Shop-R1: Rewarding LLMs to Simulate Human Behavior in Online Shopping via Reinforcement Learning

ICLR 2026poster

Large Language Models (LLMs) have recently demonstrated strong potential in generating ‘believable human-like’ behavior in web environments. Prior work has explored augmenting training data with LLM-synthesized rationales and applying supervised fine-tuning (SFT) to enhance reasoning ability, which…

Cited by 0SourcecodeScholar
2026

Speech-Aware Long Context Pruning and Integration for Contextualized Automatic Speech Recognition

AAAI 2026technical

Automatic speech recognition (ASR) systems have achieved remarkable performance in common conditions but often struggle to leverage long-context information in contextualized scenarios that require domain-specific knowledge, such as conference presentations. This challenge arises primarily due to co

Cited by 0SourcePDFScholar
2026

Superman: Unifying Skeleton and Vision for Human Motion Perception and Generation

CVPR 2026

Human motion analysis tasks, such as temporal 3D pose estimation, motion prediction, and motion in-betweening, play an essential role in computer vision. However, current paradigms suffer from severe fragmentation. First, the field is split between "perception" models that understand motion from vid

Cited by 0SourcecodeScholar
2026

TaTToo: Tool-Grounded Thinking PRM for Test-Time Scaling in Tabular Reasoning

ICLR 2026poster

Process Reward Models (PRMs) have recently emerged as a powerful framework for enhancing the reasoning capabilities of large reasoning models (LRMs), particularly in the context of test-time scaling (TTS). However, their potential for supervising LRMs on tabular reasoning domains remains underexplor…

Cited by 0SourceScholar
2026

Towards Privacy-Protected Generalized Gaze Estimation Using Diffusion Models and Domain Stability Adaptation Framework

AAAI 2026technical

Modern gaze estimation models can accurately predict human gaze from facial images. However, due to privacy concerns and intricate data collection procedures, gaze estimation datasets are typically smaller and less diverse compared to those for other vision tasks, which directly leads to poor genera

Cited by 0SourcePDFScholar
2026

Training-Free Hierarchical Working Memory for Small Language Model Agents

ICML 2026poster

Small language models (SLMs) are attractive for agent deployment, but they struggle to reliably retain and reuse decision-relevant state information over long interactions. This issue is exacerbated when working memory is maintained via unstructured natural-language summarization. Some recent work a…

Cited by 0SourceScholar
2026

Universal Skeleton Understanding via Differentiable Rendering and MLLMs

ICML 2026poster

Multimodal large language models (MLLMs) exhibit strong visual-language reasoning, yet remain confined to their native modalities and cannot directly process structured, non-visual data such as human skeletons. Existing methods either compress skeleton dynamics into lossy feature vectors for text al…

Cited by 0SourceScholar
2026

VUDG: A Dataset for Video Understanding Domain Generalization

ICLR 2026poster

Video understanding has made remarkable progress in recent years, largely driven by advances in deep models and the availability of large-scale annotated datasets. However, the robustness of these models to domain shifts encountered in real-world video applications remains a critical yet underexplor…

Cited by 0SourceScholar
2025

Are Your LLMs Capable of Stable Reasoning?

ACL 2025finding

The rapid advancement of large language models (LLMs) has shown remarkable progress in complex reasoning tasks. However, a significant disparity exists between benchmark performances and real-world applications. We attribute this gap primarily to current evaluation protocols and metrics, which inade…

2025

LLM Safety Alignment is Divergence Estimation in Disguise

NeurIPS 2025poster

We present a theoretical framework showing that popular LLM alignment methods—including RLHF and its variants—can be understood as divergence estimators between aligned (safe or preferred) and unaligned (harmful or less-preferred) distributions. This perspective explains the emergence of separation…

Cited by 0SourcecodeScholar
2025

Linear Differential Vision Transformer: Learning Visual Contrasts via Pairwise Differentials

NeurIPS 2025poster

Vision Transformers (ViTs) have become a universal backbone for both image recognition and image generation. Yet their Multi–Head Self–Attention (MHSA) layer still performs a quadratic query–key interaction for \emph{every} token pair, spending the bulk of computation on visually weak or redundant…

Cited by 0SourcecodeScholar
2025

Recognizing Actions from Robotic View for Natural Human-Robot Interaction

ICCV 2025poster

Natural Human-Robot Interaction (N-HRI) requires robots to recognize human actions at varying distances and states, regardless of whether the robot itself is in motion or stationary. This setup is more flexible and practical than conventional human action recognition tasks. However, existing benchma…

2025

SemGeoMo: Dynamic Contextual Human Motion Generation with Semantic and Geometric Guidance

CVPR 2025poster

Generating reasonable and high-quality human interactive motions in a given dynamic environment is crucial for understanding, modeling, transferring, and applying human behaviors to both virtual and physical robots. In this paper, we introduce an effective method, SemGeoMo, for dynamic contextual hu…

Cited by 0SourcePDFScholar
2025

UniPre3D: Unified Pre-training of 3D Point Cloud Models with Cross-Modal Gaussian Splatting

CVPR 2025poster

The scale diversity of point cloud data presents significant challenges in developing unified representation learning techniques for 3D vision. Currently, there are few unified 3D models, and no existing pre-training method is equally effective for both object- and scene-level point clouds. In this…

2024

Demystify Mamba in Vision: A Linear Attention Perspective

NeurIPS 2024poster

Mamba is an effective state space model with linear computation complexity. It has recently shown impressive efficiency in dealing with high-resolution inputs across various vision tasks. In this paper, we reveal that the powerful Mamba model shares surprising similarities with linear attention Tran…

2024

RecDiffusion: Rectangling for Image Stitching with Diffusion Models

CVPR 2024poster

Image stitching from different captures often results in non-rectangular boundaries which is often considered unappealing. To solve non-rectangular boundaries current solutions involve cropping which discards image content inpainting which can introduce unrelated content or warping which can distort…

2024

SelectIT: Selective Instruction Tuning for LLMs via Uncertainty-Aware Self-Reflection

NeurIPS 2024poster

Instruction tuning (IT) is crucial to tailoring large language models (LLMs) towards human-centric interactions. Recent advancements have shown that the careful selection of a small, high-quality subset of IT data can significantly enhance the performance of LLMs. Despite this, common approaches of…

2024

Self-Training with Pseudo-Label Scorer for Aspect Sentiment Quad Prediction

ACL 2024long

Aspect Sentiment Quad Prediction (ASQP) aims to predict all quads (aspect term, aspect category, opinion term, sentiment polarity) for a given review, which is the most representative and challenging task in aspect-based sentiment analysis. A key challenge in the ASQP task is the scarcity of labeled…

2024

XMask3D: Cross-modal Mask Reasoning for Open Vocabulary 3D Semantic Segmentation

NeurIPS 2024poster

Existing methodologies in open vocabulary 3D semantic segmentation primarily concentrate on establishing a unified feature space encompassing 3D, 2D, and textual modalities. Nevertheless, traditional techniques such as global feature alignment or vision-language model distillation tend to impose onl…

2023

Take-A-Photo: 3D-to-2D Generative Pre-training of Point Cloud Models

ICCV 2023poster

With the overwhelming trend of mask image modeling led by MAE, generative pre-training has shown a remarkable potential to boost the performance of fundamental models in 2D vision. However, in 3D vision, the over-reliance on Transformer-based backbones and the unordered nature of point clouds have r…

Cited by 23PDFcodeScholar
2022

Learning Laparoscope Actions via Video Features for Proactive Robotic Field-of-View Control

RA-L 2022

Smart laparoscope motion control for adjusting surgical field-of-view is an increasingly hot topic in robot-assisted surgery. Previous off-the-shelf methods have been conducted in reactive ways which heavily rely on human input signals, e.g., gaze or voice, thus cannot avoid cognitive burdens to sur

Cited by 18SourceScholar
2022

P2P: Tuning Pre-trained Image Models for Point Cloud Analysis with Point-to-Pixel Prompting

NeurIPS 2022accept

Nowadays, pre-training big models on large-scale datasets has become a crucial topic in deep learning. The pre-trained models with high representation ability and transferability achieve a great success and dominate many downstream tasks in natural language processing and 2D vision. However, it is n…

2022

SemAffiNet: Semantic-Affine Transformation for Point Cloud Segmentation

CVPR 2022poster

Conventional point cloud semantic segmentation methods usually employ an encoder-decoder architecture, where mid-level features are locally aggregated to extract geometric information. However, the over-reliance on these class-agnostic local geometric representations may raise confusion between loca…

Cited by 20PDFcodeScholar
2021

Learning in Nonzero-Sum Stochastic Games with Potentials

ICML 2021spotlight

Multi-agent reinforcement learning (MARL) has become effective in tackling discrete cooperative game scenarios. However, MARL has yet to penetrate settings beyond those modelled by team and zero-sum games, confining it to a small subset of multi-agent systems. In this paper, we introduce a new gener…

Cited by 59SourcePDFScholar
2021

NOVAS: Non-convex Optimization via Adaptive Stochastic Search for End-to-end Learning and Control

ICLR 2021poster

In this work we propose the use of adaptive stochastic search as a building block for general, non-convex optimization operations within deep neural network architectures. Specifically, for an objective function located at some layer in the network and parameterized by some network parameters, we em…

Cited by 6SourcePDFScholar
2021

PV-RAFT: Point-Voxel Correlation Fields for Scene Flow Estimation of Point Clouds

CVPR 2021poster

In this paper, we propose a Point-Voxel Recurrent All-Pairs Field Transforms (PV-RAFT) method to estimate scene flow from point clouds. Since point clouds are irregular and unordered, it is challenging to efficiently extract features from all-pairs fields in the 3D space, where all-pairs correlation…

Cited by 118PDFcodeScholar
2021

PoinTr: Diverse Point Cloud Completion With Geometry-Aware Transformers

ICCV 2021poster

Point clouds captured in real-world applications are often incomplete due to the limited sensor resolution, single viewpoint, and occlusion. Therefore, recovering the complete point clouds from partial ones becomes an indispensable task in many practical applications. In this paper, we present a new…

Cited by 608PDFcodeScholar
2021

Towards Interpretable Deep Metric Learning With Structural Matching

ICCV 2021poster

How do the neural networks distinguish two images? It is of critical importance to understand the matching mechanism of deep models for developing reliable intelligent systems for many risky visual applications such as surveillance and access control. However, most existing deep metric learning meth…

Cited by 61PDFcodeScholar
2020

Constrained Sampling-based Trajectory Optimization using Stochastic Approximation

ICRA 2020poster

We propose a sampling-based trajectory optimization methodology for constrained problems. We extend recent works on stochastic search to deal with box control constraints, as well as nonlinear state constraints for discrete dynamical systems. Regarding the former, our strategy is to optimize over tr…

Cited by 13SourceScholar
2020

Safe Optimal Control Using Stochastic Barrier Functions and Deep Forward-Backward SDEs

CoRL 2020

This paper introduces a new formulation for stochastic optimal control and stochastic dynamic optimization that ensures safety with respect to state and control constraints. The proposed methodology brings together concepts such as Forward-Backward Stochastic Differential Equations, Stochastic Barri

Cited by 0SourcePDFScholar
2019

Learning Deep Stochastic Optimal Control Policies Using Forward-Backward SDEs

RSS 2019poster

In this paper we propose a new methodology for decision-making under uncertainty using recent advancements in the areas of nonlinear stochastic optimal control theory, applied mathematics, and machine learning. Grounded on the fundamental relation between certain nonlinear partial differential equat…

Cited by 56SourcePDFScholar