← Search

Wenxiao Wang

45 accepted papers

2026

Differential Fine-Tuning Large Language Models Towards Better Diverse Reasoning Abilities

ICLR 2026poster

Reasoning abilities of large language models (LLMs) require explicit derivations compared to general question-answering, supervised fine-tuning (SFT) can empower multiple reasoning abilities in LLMs via learning from various datasets. However, neither training the datasets jointly (mix-up) nor conti…

Cited by 0SourcecodeScholar
2026

HEAPr: Hessian-based Efficient Atomic Expert Pruning in Output Space

ICLR 2026poster

Mixture-of-Experts (MoE) architectures in large language models (LLMs) deliver exceptional performance and reduced inference costs compared to dense LLMs. However, their large parameter counts result in prohibitive memory requirements, limiting practical deployment. While existing pruning methods pr…

Cited by 0SourcecodeScholar
2026

How Do Language Models Speak Languages? A Case Study on Unintended Code-Switching

ICML 2026poster

Unintended code-switching, which refers to the phenomenon where LLM unexpectedly switch languages, poses a fundamental challenge in the multilingual capabilities in LLMs. However, we still lack a mechanistic account of how this failure mode is implemented inside the model. For example, what internal…

Cited by 0SourceScholar
2026

MatchMask: Mask-Centric Generative Data Augmentation for Label-Scarce Semantic Segmentation

CVPR 2026

Current semantic segmentation models are very data-hungry and require massive costly pixel-wise human annotations. Generative data augmentation, which scales the train set using generative models, provides a potential remedy. In this paper, we propose MatchMask, a novel mask-centric generative data

Cited by 0SourceScholar
2026

ScalingCache: Extreme Acceleration of DiTs through Difference Scaling and Dynamic Interval Caching

ICLR 2026poster

Diffusion Transformers (DiTs) have emerged as powerful generative models, but their iterative denoising structure and deep transformer blocks incur substantial computational overhead, limiting the accessibility and practical deployment of high-quality video generation. To address this bottleneck, we…

Cited by 0SourcecodeScholar
2026

Uncovering the Gradient Geometry of Long CoT: A Spectral-guided Approach to Reasoning Distillation

ICML 2026poster

Large reasoning models (LRMs) achieve remarkable reasoning performance by generating long chains-of-thought (CoT). However, standard supervised fine-tuning (SFT) treats all tokens uniformly, indiscriminately minimizing loss across both essential reasoning steps and those that are noisy, redundant, o…

Cited by 0SourceScholar
2025

Concise and Organized Perception Facilitates Reasoning in Large Language Models

NAACL 2025findings

Exploiting large language models (LLMs) to tackle reasoning has garnered growing attention. It still remains highly challenging to achieve satisfactory results in complex logical problems, characterized by plenty of premises within the context and requiring multi-hop reasoning. In particular, the re…

2025

Controlling Thinking Speed in Reasoning Models

NeurIPS 2025spotlight

Human cognition is theorized to operate in two modes: fast, intuitive System 1 thinking and slow, deliberate System 2 thinking. While current Large Reasoning Models (LRMs) excel at System 2 thinking, their inability to perform fast thinking leads to high computational overhead and latency. In this w…

Cited by 0SourceScholar
2025

Don't Take Things Out of Context: Attention Intervention for Enhancing Chain-of-Thought Reasoning in Large Language Models

ICLR 2025poster

Few-shot Chain-of-Thought (CoT) significantly enhances the reasoning capabilities of large language models (LLMs), functioning as a whole to guide these models in generating reasoning steps toward final answers. However, we observe that isolated segments, words, or tokens within CoT demonstrations c…

Cited by 0SourcePDFScholar
2025

DyePack: Provably Flagging Test Set Contamination in LLMs Using Backdoors

EMNLP 2025

Open benchmarks are essential for evaluating and advancing large language models, offering reproducibility and transparency. However, their accessibility makes them likely targets of test set contamination. In this work, we introduce **DyePack**, a framework that leverages backdoor attacks to identi

2025

FlexCAD: Unified and Versatile Controllable CAD Generation with Fine-tuned Large Language Models

ICLR 2025poster

Recently, there is a growing interest in creating computer-aided design (CAD) models based on user intent, known as controllable CAD generation. Existing work offers limited controllability and needs separate models for different types of control, reducing efficiency and practicality. To achieve con…

2025

From Redundancy to Relevance: Information Flow in LVLMs Across Reasoning Tasks

NAACL 2025long

Large Vision Language Models (LVLMs) achieve great performance on visual-language reasoning tasks, however, the black-box nature of LVLMs hinders in-depth research on the reasoning mechanism. As all images need to be converted into image tokens to fit the input format of large language models (LLMs)…

2025

GeoCAD: Local Geometry-Controllable CAD Generation with Large Language Models

NeurIPS 2025poster

Local geometry-controllable computer-aided design (CAD) generation aims to modify local parts of CAD models automatically, enhancing design efficiency. It also ensures that the shapes of newly generated local parts follow user-specific geometric instructions (e.g., an isosceles right triangle or a…

Cited by 0SourceScholar
2025

Improving Complex Reasoning with Dynamic Prompt Corruption: A Soft Prompt Optimization Approach

ICLR 2025poster

Prompt Tuning (PT) has emerged as a promising Parameter-Efficient Fine-Tuning (PEFT) approach by appending trainable continuous prompt vectors to the input, maintaining competitive performance with significantly fewer trainable parameters. While PT has shown effectiveness in enhancing task performan…

Cited by 0SourcePDFScholar
2025

STraj: Self-training for Bridging the Cross-Geography Gap in Trajectory Prediction

AAAI 2025technical

Accurate trajectory prediction has prominent significance in autonomous driving scenarios. Most existing methods predict the trajectory of an agent by learning its interaction with other agents and the map within the scenario. However, the heterogeneous distribution of these elements across differe…

2025

Tool Preferences in Agentic LLMs are Unreliable

EMNLP 2025

Large language models (LLMs) can now access a wide range of external tools, thanks to the Model Context Protocol (MCP). This greatly expands their abilities as various agents. However, LLMs rely entirely on the text descriptions of tools to decide which ones to use—a process that is surprisingly fra

2024

Adapt2Reward: Adapting Video-Language Models to Generalizable Robotic Rewards via Failure Prompts

ECCV 2024poster

"For a general-purpose robot to operate in reality, executing a broad range of instructions across various environments is imperative. Central to the reinforcement learning and planning for such robotic agents is a generalizable reward function. Recent advances in vision-language models, such as CLI…

Cited by 2SourcePDFScholar
2024

DRSM: De-Randomized Smoothing on Malware Classifier Providing Certified Robustness

ICLR 2024poster

Machine Learning (ML) models have been utilized for malware detection for over two decades. Consequently, this ignited an ongoing arms race between malware authors and antivirus systems, compelling researchers to propose defenses for malware-detection models against evasion attacks. However, most if…

2024

Delving into the Reversal Curse: How Far Can Large Language Models Generalize?

NeurIPS 2024poster

While large language models (LLMs) showcase unprecedented capabilities, they also exhibit certain inherent limitations when facing seemingly trivial tasks. A prime example is the recently debated "reversal curse", which surfaces when models, having been trained on the fact "A is B", struggle to gen…

2024

EMVP: Embracing Visual Foundation Model for Visual Place Recognition with Centroid-Free Probing

NeurIPS 2024poster

Visual Place Recognition (VPR) is essential for mobile robots as it enables them to retrieve images from a database closest to their current location. The progress of Visual Foundation Models (VFMs) has significantly advanced VPR by capturing representative descriptors in images. However, existing f…

Cited by 0SourcePDFScholar
2024

Enhancing Multiple Dimensions of Trustworthiness in LLMs via Sparse Activation Control

NeurIPS 2024poster

As the development and application of Large Language Models (LLMs) continue to advance rapidly, enhancing their trustworthiness and aligning them with human preferences has become a critical area of research. Traditional methods rely heavily on extensive data for Reinforcement Learning from Human Fe…

Cited by 1SourcePDFScholar
2024

Few-shot Hybrid Domain Adaptation of Image Generator

ICLR 2024poster

Can a pre-trained generator be adapted to the hybrid of multiple target domains and generate images with integrated attributes of them? In this work, we introduce a new task -- Few-shot $\textit{Hybrid Domain Adaptation}$ (HDA). Given a source generator and several target domains, HDA aims to acquir…

Cited by 4SourcePDFScholar
2024

From Yes-Men to Truth-Tellers: Addressing Sycophancy in Large Language Models with Pinpoint Tuning

ICML 2024poster

Large Language Models (LLMs) tend to prioritize adherence to user prompts over providing veracious responses, leading to the sycophancy issue. When challenged by users, LLMs tend to admit mistakes and provide inaccurate responses even if they initially provided the correct answer. Recent works propo…

Cited by 10SourcePDFScholar
2024

G2LTraj: A Global-to-Local Generation Approach for Trajectory Prediction

IJCAI 2024poster

Predicting future trajectories of traffic agents accurately holds substantial importance in various applications such as autonomous driving. Previous methods commonly infer all future steps of an agent either recursively or simultaneously. However, the recursive strategy suffers from the accumulated…

2024

Instance-adaptive Zero-shot Chain-of-Thought Prompting

NeurIPS 2024poster

Zero-shot Chain-of-Thought (CoT) prompting emerges as a simple and effective strategy for enhancing the performance of large language models (LLMs) in real-world reasoning tasks. Nonetheless, the efficacy of a singular, task-level prompt uniformly applied across the whole of instances is inherently…

Cited by 5SourcePDFScholar
2024

Learning Occupancy for Monocular 3D Object Detection

CVPR 2024poster

Monocular 3D detection is a challenging task due to the lack of accurate 3D information. Existing approaches typically rely on geometry constraints and dense depth estimates to facilitate the learning but often fail to fully exploit the benefits of three-dimensional feature extraction in frustum and…

2024

Pseudo Label Refinery for Unsupervised Domain Adaptation on Cross-dataset 3D Object Detection

CVPR 2024poster

Recent self-training techniques have shown notable improvements in unsupervised domain adaptation for 3D object detection (3D UDA). These techniques typically select pseudo labels i.e. 3D boxes to supervise models for the target domain. However this selection process inevitably introduces unreliable…

2024

Regulating Intermediate 3D Features for Vision-Centric Autonomous Driving

AAAI 2024technical

Multi-camera perception tasks have gained significant attention in the field of autonomous driving. However, existing frameworks based on Lift-Splat-Shoot (LSS) in the multi-camera setting cannot produce suitable dense 3D features due to the projection nature and uncontrollable densification process…

2024

Robustness of AI-Image Detectors: Fundamental Limits and Practical Attacks

ICLR 2024poster

In light of recent advancements in generative AI models, it has become essential to distinguish genuine content from AI-generated one to prevent the malicious usage of fake materials as authentic ones and vice versa. Various techniques have been introduced for identifying AI-generated images, with w…

2023

CLIP Is Also an Efficient Segmenter: A Text-Driven Approach for Weakly Supervised Semantic Segmentation

CVPR 2023poster

Weakly supervised semantic segmentation (WSSS) with image-level labels is a challenging task. Mainstream approaches follow a multi-stage framework and suffer from high training costs. In this paper, we explore the potential of Contrastive Language-Image Pre-training models (CLIP) to localize differe…

2023

MonoNeRD: NeRF-like Representations for Monocular 3D Object Detection

ICCV 2023poster

In the field of monocular 3D detection, it is common practice to utilize scene geometric clues to enhance the detector's performance. However, many existing works adopt these clues explicitly such as estimating a depth map and back-projecting it into 3D space. This explicit methodology induces spars…

Cited by 35PDFcodeScholar
2023

One-shot Implicit Animatable Avatars with Model-based Priors

ICCV 2023poster

Existing neural rendering methods for creating human avatars typically either require dense input signals such as video or multi-view images, or leverage a learned prior from large-scale specific 3D human datasets such that reconstruction can be performed with sparse-view inputs. Most of these metho…

Cited by 41PDFcodeScholar
2023

PVT-SSD: Single-Stage 3D Object Detector With Point-Voxel Transformer

CVPR 2023poster

Recent Transformer-based 3D object detectors learn point cloud features either from point- or voxel-based representations. However, the former requires time-consuming sampling while the latter introduces quantization errors. In this paper, we present a novel Point-Voxel Transformer for single-stage…

2023

Spuriosity Rankings: Sorting Data to Measure and Mitigate Biases

NeurIPS 2023spotlight

We present a simple but effective method to measure and mitigate model biases caused by reliance on spurious cues. Instead of requiring costly changes to one's data or model training, our method better utilizes the data one already has by sorting them. Specifically, we rank images within their class…

Cited by 15SourcePDFScholar
2023

Towards In-Distribution Compatible Out-of-Distribution Detection

AAAI 2023technical

Deep neural network, despite its remarkable capability of discriminating targeted in-distribution samples, shows poor performance on detecting anomalous out-of-distribution data. To address this defect, state-of-the-art solutions choose to train deep networks on an auxiliary dataset of outliers. Var…

Cited by 2SourcePDFScholar
2022

CrossFormer: A Versatile Vision Transformer Hinging on Cross-scale Attention

ICLR 2022poster

Transformers have made great progress in dealing with computer vision tasks. However, existing vision transformers have not yet possessed the ability of building the interactions among features of different scales, which is perceptually important to visual inputs. The reasons are two-fold: (1) Input…

2022

Graph R-CNN: Towards Accurate 3D Object Detection with Semantic-Decorated Local Graph

ECCV 2022poster

"Two-stage detectors have gained much popularity in 3D object detection. Most two-stage 3D detectors utilize grid points, voxel grids, or sampled keypoints for RoI feature extraction in the second stage. Such methods, however, are inefficient in handling unevenly distributed and sparse outdoor point…

2022

Improved Certified Defenses against Data Poisoning with (Deterministic) Finite Aggregation

ICML 2022spotlight

Data poisoning attacks aim at manipulating model behaviors through distorting training data. Previously, an aggregation-based certified defense, Deep Partition Aggregation (DPA), was proposed to mitigate this threat. DPA predicts through an aggregation of base classifiers trained on disjoint subsets…

2022

Masked Autoencoders for Point Cloud Self-Supervised Learning

ECCV 2022poster

"As a promising scheme of self-supervised learning, masked autoencoding has significantly advanced natural language processing and computer vision. Inspired by this, we propose a neat scheme of masked autoencoders for point cloud self-supervised learning, addressing the challenges posed by point clo…

2021

Accelerate CNNs from Three Dimensions: A Comprehensive Pruning Framework

ICML 2021spotlight

Most neural network pruning methods, such as filter-level and layer-level prunings, prune the network model along one dimension (depth, width, or resolution) solely to meet a computational budget. However, such a pruning policy often leads to excessive reduction of that dimension, thus inducing a hu…

Cited by 77SourcePDFScholar
2021

On Feature Decorrelation in Self-Supervised Learning

ICCV 2021poster

In self-supervised representation learning, a common idea behind most of the state-of-the-art approaches is to enforce the robustness of the representations to predefined augmentations. A potential issue of this idea is the existence of completely collapsed solutions (i.e., constant features), which…

Cited by 236PDFcodeScholar
2020

A Differential Approach for Rain Field Tomographic Reconstruction Using Microwave Signals from Leo Satellites

ICASSP 2020accepted

A differential approach is proposed for tomographic rain field reconstruction using the estimated signal-to-noise ratio of microwave signals from low earth orbit satellites at the ground receivers, with the unknown baseline values eliminated before using least squares to reconstruct the attenuation…

Cited by 0SourceScholar
2020

The Secret Revealer: Generative Model-Inversion Attacks Against Deep Neural Networks

CVPR 2020oral

This paper studies model-inversion attacks, in which the access to a model is abused to infer information about the training data. Since its first introduction by [??], such attacks have raised serious concerns given that training data usually contain privacy sensitive information. Thus far, success…

Cited by 614PDFcodeScholar