← Search

Zhi Wang

72 accepted papers

2026

Block-wise Adaptive Caching for Accelerating Diffusion Policy

ICLR 2026poster

Diffusion Policy has demonstrated strong visuomotor modeling capabilities, but its high computational cost renders it impractical for real-time robotic control. Despite huge redundancy across repetitive denoising steps, existing diffusion acceleration techniques fail to generalize to Diffusion Polic…

Cited by 0SourcecodeScholar
2026

Characterizing, Evaluating, and Optimizing Complex Reasoning

ICML 2026oral

Large Reasoning Models (LRMs) increasingly rely on reasoning traces with complex internal structures. However, existing work lacks a unified answer to three fundamental questions: (1) what defines high-quality reasoning, (2) how to reliably evaluate long, implicitly structured reasoning traces, and …

Cited by 0SourceScholar
2026

Diversity-Incentivized Exploration for Versatile Reasoning

ICLR 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a crucial paradigm for incentivizing reasoning capabilities in Large Language Models (LLMs). Due to vast state-action spaces and reward sparsity in reasoning tasks, existing methods often struggle with deficient exploration and poo…

Cited by 0SourcecodeScholar
2026

ExGRPO: Learning to Reason from Prior Successes

ICLR 2026poster

Reinforcement learning from verifiable rewards (RLVR) is an emerging paradigm for improving the reasoning ability of large language models. However, standard on-policy training discards rollout experiences after a single update, leading to computational inefficiency and instability. While prior work…

Cited by 0SourcecodeScholar
2026

From Conflict to Consensus: Boosting Medical Reasoning via Multi-Round Agentic RAG

ICML 2026poster

Large Language Models (LLMs) exhibit high reasoning capacity in medical question-answering, but their tendency to produce hallucinations and outdated knowledge poses critical risks in healthcare fields. While Retrieval-Augmented Generation (RAG) mitigates these issues, existing methods rely on noisy…

Cited by 0SourceScholar
2026

HIGH QUALITY UNDERWATER IMAGE COMPRESSION WITH ADAPTIVE COLOR CORRECTION

ICASSP 2026oral

With the increasing exploration and exploitation of the underwater world, underwater images have become a critical medium for human interaction with marine environments, driving extensive research into their efficient transmission and storage. However, contemporary underwater image compression algor…

Cited by 0SourcePDFScholar
2026

IGen: Scalable Data Generation for Robot Learning from Open-World Images

CVPR 2026

The rise of generalist robotic policies has created an exponential demand for large-scale training data. However, on-robot data collection is labor-intensive and often limited to specific environments. In contrast, open-world images capture a vast diversity of real-world scenes that naturally align

Cited by 0SourceScholar
2026

Medical Vision–Language Pretraining with LLM-Guided Temporal Supervision

AAAI 2026technical

Medical vision–language pretraining typically relies on static image–text pairs, overlooking temporal cues vital for understanding clinical progression. This limits model sensitivity to evolving semantics and reduces their effectiveness in real-world clinical reasoning. To address this challenge, we

Cited by 0SourcePDFScholar
2026

MoETTA: Test-Time Adaptation Under Mixed Distribution Shifts with MoE-LayerNorm

AAAI 2026technical

Test-time adaptation (TTA) has proven effective in mitigating performance drops under single-domain distribution shifts by updating model parameters during inference. However, real-world deployments often involve mixed distribution shifts---where test samples are affected by diverse and potentially

Cited by 0SourcePDFScholar
2026

SIAM: Towards Generalizable Articulated Object Modeling via Single Robot-Object Interaction

AAAI 2026technical

Articulated object modeling, which represents interconnected rigid bodies with their geometry, part segmentation, articulation tree, and physical properties, is crucial for robotic perception and manipulation. Recently existing methods like SAGCI leverage Interactive Perception (IP) to refine models

Cited by 0SourcePDFScholar
2026

SP-VLA: A Joint Model Scheduling and Token Pruning Approach for VLA Model Acceleration

ICLR 2026poster

Vision-Language-Action (VLA) models have attracted increasing attention for their strong control capabilities. However, their high computational cost and low execution frequency hinder their suitability for real-time tasks such as robotic manipulation and autonomous navigation. Existing VLA accelera…

Cited by 0SourcecodeScholar
2026

Sparse ActionGen: Accelerating Diffusion Policy with Real-time Pruning

ICML 2026poster

Diffusion Policy has dominated action generation due to its strong capabilities for modeling multi-modal action distributions, but its multi-step denoising processes make it impractical for real-time visuomotor control. Existing caching-based acceleration methods typically rely on $\textit{static}$ …

Cited by 0SourcecodeScholar
2026

Thinking in Dynamics: How Multimodal Large Language Models Perceive, Track, and Reason Dynamics in Physical 4D World

CVPR 2026

Humans inhabit a physical 4D world, where spatial geometry and semantic content evolve over time, forming a dynamic reality. While current Multimodal Large Language Models (MLLMs) demonstrate strong capabilities in understanding static visual inputs, it remains unclear whether they can effectively "

Cited by 0SourcecodeScholar
2026

VVS: Accelerating Speculative Decoding for Visual Autoregressive Generation via Partial Verification Skipping

CVPR 2026

Visual autoregressive (AR) generation models have demonstrated strong potential for image generation, yet their next-token-prediction paradigm introduces considerable inference latency. Although speculative decoding (SD) has been proven effective for accelerating visual AR models, its "draft one ste

Cited by 0SourcecodeScholar
2025

Accelerating Parallel Diffusion Model Serving with Residual Compression

NeurIPS 2025poster

Diffusion models produce realistic images and videos but require substantial computational resources, necessitating multi-accelerator parallelism for real-time deployment. However, parallel inference introduces significant communication overhead from exchanging large activations between devices, lim…

Cited by 0SourcecodeScholar
2025

Beyond A Single AI Cluster: A Survey of Decentralized LLM Training

EMNLP 2025

The emergence of large language models (LLMs) has revolutionized AI development, yet their resource demands beyond a single cluster or even datacenter, limiting accessibility to well-resourced organizations. Decentralized training has emerged as a promising paradigm to leverage dispersed resources a

Cited by 0SourcePDFScholar
2025

CATP-LLM: Empowering Large Language Models for Cost-Aware Tool Planning

ICCV 2025poster

Utilizing large language models (LLMs) for tool planning has emerged as a promising avenue for developing general AI systems, where LLMs automatically schedule external tools (e.g., vision models) to tackle complex tasks based on task descriptions. To push this paradigm toward practical applications…

2025

COSMIC: Clique-Oriented Semantic Multi-space Integration for Robust CLIP Test-Time Adaptation

CVPR 2025poster

Recent vision-language models (VLMs) face significant challenges in test-time adaptation to novel domains. While cache-based methods show promise by leveraging historical information, they struggle with both caching unreliable feature-label pairs and indiscriminately using single-class information d…

2025

DICE: Staleness-Centric Optimizations for Parallel Diffusion MoE Inference

ICCV 2025poster

Mixture-of-Experts-based (MoE-based) diffusion models demonstrate remarkable scalability in high-fidelity image generation, yet their reliance on expert parallelism introduces critical communication bottlenecks. State-of-the-art methods alleviate such overhead in parallel diffusion inference through…

2025

Divide and Conquer: Grounding LLMs as Efficient Decision-Making Agents via Offline Hierarchical Reinforcement Learning

ICML 2025poster

While showing sophisticated reasoning abilities, large language models (LLMs) still struggle with long-horizon decision-making tasks due to deficient exploration and long-term credit assignment, especially in sparse-reward scenarios. Inspired by the divide-and-conquer principle, we propose an innova…

2025

DoorBot: Closed-Loop Task Planning and Manipulation for Door Opening in the Wild with Haptic Feedback

ICRA 2025

Robots operating in unstructured environments face significant challenges when interacting with everyday objects like doors. They particularly struggle to generalize across diverse door types and conditions. Existing vision-based and open-loop planning methods often lack the robustness to handle var

Cited by 1SourceScholar
2025

EVOS: Efficient Implicit Neural Training via EVOlutionary Selector

CVPR 2025poster

We propose EVOlutionary Selector (EVOS), an efficient training paradigm for accelerating Implicit Neural Representation (INR). Unlike conventional INR training that feeds all samples through the neural network in each iteration, our approach restricts training to strategically selected points, reduc…

2025

Enhancing Implicit Neural Representations via Symmetric Power Transformation

AAAI 2025technical

We propose symmetric power transformation to enhance the capacity of Implicit Neural Representation (INR) from the perspective of data transformation. Unlike prior work utilizing random permutation or index rearrangement, our method features a reversible operation that does not require additional st…

2025

Feature-Based Instance Neighbor Discovery: Advanced Stable Test-Time Adaptation in Dynamic World

NeurIPS 2025poster

Despite progress, deep neural networks still suffer performance declines under distribution shifts between training and test domains, leading to a substantial decrease in Quality of Experience (QoE) for applications. Existing test-time adaptation (TTA) methods are challenged by dynamic, multiple tes…

Cited by 0SourcecodeScholar
2025

JAQ: Joint Efficient Architecture Design and Low-Bit Quantization with Hardware-Software Co-Exploration

AAAI 2025technical

The co-design of neural network architectures, quantization precisions, and hardware accelerators offers a promising approach to achieving an optimal balance between performance and efficiency, particularly for model deployment on resource-constrained edge devices. In this work, we propose the JAQ F…

Cited by 0SourcePDFScholar
2025

Learning to Reason under Off-Policy Guidance

NeurIPS 2025poster

Recent advances in large reasoning models (LRMs) demonstrate that sophisticated behaviors such as multi-step reasoning and self-reflection can emerge via reinforcement learning with verifiable rewards~(RLVR). However, existing RLVR approaches are inherently ``on-policy'', limiting learning to a mod…

Cited by 0SourcecodeScholar
2025

Lungmix: A Mixup-Based Strategy for Generalization in Respiratory Sound Classification

ICASSP 2025accepted

Respiratory sound classification plays a pivotal role in diagnosing respiratory diseases. While deep learning models have succeeded with various respiratory sound datasets, our experiments indicate that models trained on one dataset often fail to generalize effectively to others, mainly due to data…

Cited by 0SourceScholar
2025

Mixture-of-Experts Meets In-Context Reinforcement Learning

NeurIPS 2025poster

In-context reinforcement learning (ICRL) has emerged as a promising paradigm for adapting RL agents to downstream tasks through prompt conditioning. However, two notable challenges remain in fully harnessing in-context learning within RL domains: the intrinsic multi-modality of the state-action-rewa…

Cited by 0SourcecodeScholar
2025

Music-Aligned Holistic 3D Dance Generation via Hierarchical Motion Modeling

ICCV 2025poster

Well-coordinated, music-aligned holistic dance enhances emotional expressiveness and audience engagement. However, generating such dances remains challenging due to the scarcity of holistic 3D dance datasets, the difficulty of achieving cross-modal alignment between music and dance, and the complexi…

2025

PAL: Sample-Efficient Personalized Reward Modeling for Pluralistic Alignment

ICLR 2025poster

Foundation models trained on internet-scale data benefit from extensive alignment to human preferences before deployment. However, existing methods typically assume a homogeneous preference shared by all individuals, overlooking the diversity inherent in human values. In this work, we propose a gene…

Cited by 2SourcePDFScholar
2025

PulmoScan: A Practical Pulmonary Disease Pre-Screening System

ICASSP 2025accepted

Automation of pulmonary disease identification has been a long-standing area of research and gained increased attention after the COVID-19 pandemic. However, existing respiratory sound classification algorithms exhibit significant limitations, including suboptimal performance, insufficient input rob…

Cited by 0SourceScholar
2025

Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers

CVPR 2025poster

Recent advancements in diffusion models, particularly the architectural transformation from UNet-based models to Diffusion Transformers (DiTs), significantly improve the quality and scalability of image and video generation. However, despite their impressive capabilities, the substantial computation…

2025

SAGENet: Binaural Echo-Based 3D Depth Estimation with Sparse Angular Queries and Refined Geometric Cues

IROS 2025

In this paper, we propose SAGENet that utilizes only binaural echoes (i.e., for scenarios when vision perception seriously degrades) for scene depth estimation. Unlike previous methods that implicitly learn spatial features from echoes, which may cause shape and scale drift, SAGENet explicitly extra

Cited by 0SourcecodeScholar
2025

Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision

NeurIPS 2025poster

Offline meta-RL usually tackles generalization by inferring task beliefs from high-quality samples or warmup explorations. The restricted form limits their generality and usability since these supervision signals are expensive and even infeasible to acquire in advance for unseen tasks. Learning dire…

Cited by 0SourcecodeScholar
2025

Understanding Bias Terms in Neural Representations

NeurIPS 2025poster

In this paper, we examine the impact and significance of bias terms in Implicit Neural Representations (INRs). While bias terms are known to enhance nonlinear capacity by shifting activations in typical neural networks, we discover their functionality differs markedly in neural representation networ…

Cited by 0SourceScholar
2024

Attention-Guided Contrastive Role Representations for Multi-agent Reinforcement Learning

ICLR 2024poster

Real-world multi-agent tasks usually involve dynamic team composition with the emergence of roles, which should also be a key to efficient cooperation in multi-agent reinforcement learning (MARL). Drawing inspiration from the correlation between roles and agent's behavior patterns, we propose a nove…

2024

Beyond task diversity: provable representation transfer for sequential multitask linear bandits

NeurIPS 2024poster

We study lifelong learning in linear bandits, where a learner interacts with a sequence of linear bandit tasks whose parameters lie in an $m$-dimensional subspace of $\mathbb{R}^d$, thereby sharing a low-rank representation. Current literature typically assumes that the tasks are diverse, i.e., thei…

2024

KOSMOS-E : Learning to Follow Instruction for Robotic Grasping

IROS 2024poster

Tuning on instruction-following data has been shown to enhance the capabilities and controllability of language models, but the idea is less explored in the robotic field. In this work, we introduce KOSMOS-E, a Multimodal Large Language Model (MLLM) that leverages instruction-following robotic grasp…

Cited by 1SourceScholar
2024

LCM: Locally Constrained Compact Point Cloud Model for Masked Point Modeling

NeurIPS 2024poster

The pre-trained point cloud model based on Masked Point Modeling (MPM) has exhibited substantial improvements across various tasks. However, these models heavily rely on the Transformer, leading to quadratic complexity and limited decoder, hindering their practice application. To address this limita…

2024

MesonGS: Post-training Compression of 3D Gaussians via Efficient Attribute Transformation

ECCV 2024poster

"3D Gaussian Splatting demonstrates excellent quality and speed in novel view synthesis. Nevertheless, the huge file size of the 3D Gaussians presents challenges for transmission and storage. Current works design compact models to replace the substantial volume and attributes of 3D Gaussians, along…

Cited by 12SourcePDFScholar
2024

Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglement

NeurIPS 2024poster

A longstanding goal of artificial general intelligence is highly capable generalists that can learn from diverse experiences and generalize to unseen tasks. The language and vision communities have seen remarkable progress toward this trend by scaling up transformer-based models trained on massive d…

2024

Nonconvex Multiview Subspace Clustering Framework with Efficient Method Designs and Theoretical Analysis

IJCAI 2024poster

Multi-view subspace clustering (MvSC) is one of the most effective methods for understanding and processing high-dimensional data. However, existing MvSC methods still have two shortcomings: (1) they adopt the nuclear norm as the low-rank constraint, which makes it impossible to fully exploit the mu…

2024

OVD-Explorer: Optimism Should Not Be the Sole Pursuit of Exploration in Noisy Environments

AAAI 2024technical

In reinforcement learning, the optimism in the face of uncertainty (OFU) is a mainstream principle for directing exploration towards less explored areas, characterized by higher uncertainty. However, in the presence of environmental stochasticity (noise), purely optimistic exploration may lead to ex…

2024

Procedural Level Generation with Diffusion Models from a Single Example

AAAI 2024technical

Level generation is a central focus of Procedural Content Generation (PCG), yet deep learning-based approaches are limited by scarce training data, i.e., human-designed levels. Despite being a dominant framework, Generative Adversarial Networks (GANs) exhibit a substantial quality gap between genera…

2024

Retraining-Free Model Quantization via One-Shot Weight-Coupling Learning

CVPR 2024poster

Quantization is of significance for compressing the over-parameterized deep neural models and deploying them on resource-limited devices. Fixed-precision quantization suffers from performance drop due to the limited numerical representation ability. Conversely mixed-precision quantization (MPQ) is a…

2024

Towards Compact 3D Representations via Point Feature Enhancement Masked Autoencoders

AAAI 2024technical

Learning 3D representation plays a critical role in masked autoencoder (MAE) based pre-training methods for point cloud, including single-modal and cross-modal based MAE. Specifically, although cross-modal MAE methods learn strong 3D representations via the auxiliary of other modal knowledge, they…

2024

Vision-Language Pre-training with Object Contrastive Learning for 3D Scene Understanding

AAAI 2024technical

In recent years, vision language pre-training frameworks have made significant progress in natural language processing and computer vision, achieving remarkable performance improvement on various downstream tasks. However, when extended to point cloud data, existing works mainly focus on building ta…

2023

BadTrack: A Poison-Only Backdoor Attack on Visual Object Tracking

NeurIPS 2023poster

Visual object tracking (VOT) is one of the most fundamental tasks in computer vision community. State-of-the-art VOT trackers extract positive and negative examples that are used to guide the tracker to distinguish the object from the background. In this paper, we show that this characteristic can b…

Cited by 6SourcePDFScholar
2023

Curriculum Multi-Negative Augmentation for Debiased Video Grounding

AAAI 2023technical

Video Grounding (VG) aims to locate the desired segment from a video given a sentence query. Recent studies have found that current VG models are prone to over-rely the groundtruth moment annotation distribution biases in the training set. To discourage the standard VG model's behavior of exploiting…

2023

ElasticViT: Conflict-aware Supernet Training for Deploying Fast Vision Transformer on Diverse Mobile Devices

ICCV 2023poster

Neural Architecture Search (NAS) has shown promising performance in the automatic design of vision transformers (ViT) exceeding 1G FLOPs. However, designing lightweight and low-latency ViT models for diverse mobile devices remains a big challenge. In this work, we propose ElasticViT, a two-stage NAS…

Cited by 25PDFcodeScholar
2023

FSR: A General Frequency-Oriented Framework to Accelerate Image Super-resolution Networks

AAAI 2023technical

Deep neural networks (DNNs) have witnessed remarkable achievement in image super-resolution (SR), and plenty of DNN-based SR models with elaborated network designs have recently been proposed. However, existing methods usually require substantial computations by operating in spatial domain. To addre…

2023

GIFD: A Generative Gradient Inversion Method with Feature Domain Optimization

ICCV 2023poster

Federated Learning (FL) has recently emerged as a promising distributed machine learning framework to preserve clients' privacy, by allowing multiple clients to upload the gradients calculated from their local data to a central server. Recent studies find that the exchanged gradients also take the r…

Cited by 40PDFcodeScholar
2023

Instance-aware Dynamic Prompt Tuning for Pre-trained Point Cloud Models

ICCV 2023poster

Pre-trained point cloud models have found extensive applications in 3D understanding tasks like object classification and part segmentation. However, the prevailing strategy of full fine-tuning in downstream tasks leads to large per-task storage overhead for model parameters, which limits the effici…

Cited by 48PDFcodeScholar
2023

Weakly-Supervised Semantic Segmentation for Histopathology Images Based on Dataset Synthesis and Feature Consistency Constraint

AAAI 2023technical

Tissue segmentation is a critical task in computational pathology due to its desirable ability to indicate the prognosis of cancer patients. Currently, numerous studies attempt to use image-level labels to achieve pixel-level segmentation to reduce the need for fine annotations. However, most of the…

2022

Mixed-Precision Neural Network Quantization via Learned Layer-Wise Importance

ECCV 2022poster

"The exponentially large discrete search space in mixed-precision quantization (MPQ) makes it hard to determine the optimal bit-width for each layer. Previous works usually resort to iterative search methods on the training set, which consume hundreds or even thousands of GPU-hours. In this study, w…

2022

Online Continual Adaptation with Active Self-Training

AISTATS 2022poster

Models trained with offline data often suffer from continual distribution shifts and expensive labeling in changing environments. This calls for a new online learning paradigm where the learner can continually adapt to changing environments with limited labels. In this paper, we propose a new online…

Cited by 13SourcePDFScholar
2022

Thompson Sampling for Robust Transfer in Multi-Task Bandits

ICML 2022spotlight

We study the problem of online multi-task learning where the tasks are performed within similar but not necessarily identical multi-armed bandit environments. In particular, we study how a learner can improve its overall performance across multiple related tasks through robust transfer of knowledge.…

2022

bert2BERT: Towards Reusable Pretrained Language Models

ACL 2022long

In recent years, researchers tend to pre-train ever-larger language models to explore the upper limit of deep models. However, large language model pre-training costs intensive computational resources, and most of the models are trained from scratch without reusing the existing pre-trained models, w…

Cited by 86SourcePDFScholar
2021

Multitask Bandit Learning Through Heterogeneous Feedback Aggregation

AISTATS 2021poster

In many real-world applications, multiple agents seek to learn how to perform highly related yet slightly different tasks in an online bandit learning protocol. We formulate this problem as the $\epsilon$-multi-player multi-armed bandit problem, in which a set of players concurrently interact with a…

2020

Texture and Shape Biased Two-Stream Networks for Clothing Classification and Attribute Recognition

CVPR 2020poster

Clothes category classification and attribute recognition have achieved distinguished success with the development of deep learning. People have found that landmark detection plays a positive role in these tasks. However, little research is committed to analyzing these tasks from the perspective of…

Cited by 66PDFScholar
2016

A fast direct source localization approach for acoustic sensor array

ICASSP 2016accepted

We present a novel Fast Direct Source Localization (FDSL) approach for acoustic isotropic sensor array applications. Unlike previous approximate maximum likelihood (AML) approach, the proposed FDSL focuses on the phase shift caused by time delays among sensors and obtains an analytical result withou…

Cited by 0SourceScholar