← Search

Jiawei Xu

29 accepted papers

2026

AirBender: Adaptive Transportation of Bendable Objects Using Dual UAVs

ICRA 2026poster

The interaction of robots with bendable objects in midair presents significant challenges in control, often resulting in performance degradation and potential crashes, especially for aerial robots due to their limited actuation capabilities and constant need to remain airborne. This paper presents a…

2026

Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generations

ICML 2026poster

High-quality kernel is critical for scalable AI systems, and enabling LLMs to generate such code would advance AI development. However, training LLMs for this task requires sufficient data, a robust environment, and the process is often vulnerable to _reward hacking_ and _lazy optimization_. In thes…

Cited by 0SourceScholar
2026

LogART: Pushing the Limit of Efficient Logarithmic Post-Training Quantization

ICLR 2026poster

Efficient deployment of deep neural networks increasingly relies on Post-Training Quantization (PTQ). Logarithmic PTQ, in particular, promises multiplier-free hardware efficiency, but its performance is often limited by the nonlinear and symmetric quantization grid and standard rounding-to-nearest (…

Cited by 0SourcecodeScholar
2026

STLA: Spatiotemporal Lookahead Alignment for Post-Training Quantization

ICML 2026poster

Adaptive rounding techniques in Post-Training Quantization (PTQ) enable the efficient deployment of Large Language Models (LLMs) with low resource and data dependencies. While learning-based rounding methods are accurate yet costly, compensation-based approaches offer a highly efficient alternative.…

Cited by 0SourceScholar
2026

Scheduling Thoughts: Learning the Order of Thought in Diffusion Language Models

ICML 2026poster

Masked diffusion language models decode by iteratively unmasking tokens, where the unmasking order defines an ``order of thought'' that strongly influences generation quality yet is typically chosen heuristically. We derive a tractable upper bound on the sequential decoding mismatch, measured by the…

Cited by 0SourceScholar
2026

The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL

ICML 2026poster

Reinforcement Learning for Large Language Models (LLMs) often suffers from training collapse in long-horizon tasks due to exploding gradient variance. To mitigate this, baseline is commonly introduced for advantage computation; however, traditional value models remain difficult to optimize, and stan…

Cited by 0SourceScholar
2026

Trust Region Masking for Long-Horizon LLM Reinforcement Learning

ICML 2026poster

Policy gradient methods for Large Language Models (LLMs) optimize a policy $\pi_\theta$ via a surrogate objective computed from samples of a rollout policy $\pi_{\text{roll}}$. However, modern LLM-RL pipelines suffer from unavoidable implementation divergences—such as backend discrepancies, Mixture-…

Cited by 0SourceScholar
2025

AD-GS: Object-Aware B-Spline Gaussian Splatting for Self-Supervised Autonomous Driving

ICCV 2025poster

Modeling and rendering dynamic urban driving scenes is crucial for self-driving simulation. Current high-quality methods typically rely on costly manual object tracklet annotations, while self-supervised approaches fail to capture dynamic object motions accurately and decompose scenes properly, resu…

Cited by 0SourcePDFScholar
2025

ADG: Ambient Diffusion-Guided Dataset Recovery for Corruption-Robust Offline Reinforcement Learning

NeurIPS 2025poster

Real-world datasets collected from sensors or human inputs are prone to noise and errors, posing significant challenges for applying offline reinforcement learning (RL). While existing methods have made progress in addressing corrupted actions and rewards, they remain insufficient for handling corru…

Cited by 0SourceScholar
2025

Mapping from Meaning: Addressing the Miscalibration of Prompt-Sensitive Language Models

AAAI 2025technical

An interesting behavior in large language models (LLMs) is prompt sensitivity. When provided with different but semantically equivalent versions of the same prompt, models may produce very different distributions of answers. This suggests that the uncertainty reflected in a model's output distributi…

2025

MedHallu: A Comprehensive Benchmark for Detecting Medical Hallucinations in Large Language Models

EMNLP 2025

Advancements in Large Language Models (LLMs) and their increasing use in medical question-answering necessitate rigorous evaluation of their reliability. A critical challenge lies in hallucination, where models generate plausible yet factually incorrect outputs. In the medical domain, this poses ser

Cited by 0SourcePDFScholar
2025

Rapidly Tunable Stiffness Soft Gripper for Multifunctional Grasping

RA-L 2025

The flexible gripper exhibits limited load capacity while possessing dexterity and adaptability to complex environments. Conversely, the thermally-driven variable stiffness gripper boasts high load capacity but suffers from slow response times. This study presents a multifunctional flexible gripper

Cited by 3SourceScholar
2025

Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modeling

ICLR 2025poster

Learning policy from offline datasets through offline reinforcement learning (RL) holds promise for scaling data-driven decision-making while avoiding unsafe and costly online interactions. However, real-world data collected from sensors or humans often contains noise and errors, posing a significan…

2024

CrackNex: a Few-shot Low-light Crack Segmentation Model Based on Retinex Theory for UAV Inspections

ICRA 2024poster

Routine visual inspections of concrete structures are imperative for upholding the safety and integrity of critical infrastructure. Such visual inspections sometimes happen under low-light conditions, e.g., checking for bridge health. Crack segmentation under such conditions is challenging due to th…

Cited by 13SourcecodeScholar
2024

Grid4D: 4D Decomposed Hash Encoding for High-Fidelity Dynamic Gaussian Splatting

NeurIPS 2024poster

Recently, Gaussian splatting has received more and more attention in the field of static scene rendering. Due to the low computational overhead and inherent flexibility of explicit representations, plane-based explicit methods are popular ways to predict deformations for Gaussian-based dynamic scene…

Cited by 2SourcePDFScholar
2024

Q-Star Meets Scalable Posterior Sampling: Bridging Theory and Practice via HyperAgent

ICML 2024poster

We propose HyperAgent, a reinforcement learning (RL) algorithm based on the hypermodel framework for exploration in RL. HyperAgent allows for the efficient incremental approximation of posteriors associated with an optimal action-value function ($Q^\star$) without the need for conjugacy and follows…

2024

Relative Policy-Transition Optimization for Fast Policy Transfer

AAAI 2024technical

We consider the problem of policy transfer between two Markov Decision Processes (MDPs). We introduce a lemma based on existing theoretical results in reinforcement learning to measure the relativity gap between two arbitrary MDPs, that is the difference between any two cumulative expected returns d…

Cited by 0SourcePDFScholar
2024

SELECT: A Large-Scale Benchmark of Data Curation Strategies for Image Classification

NeurIPS 2024poster

Data curation is the problem of how to collect and organize samples into a dataset that supports efficient learning. Despite the centrality of the task, little work has been devoted towards a large-scale, systematic comparison of various curation methods. In this work, we take steps towards a formal…

2024

Towards Robust Offline Reinforcement Learning under Diverse Data Corruption

ICLR 2024spotlight

Offline reinforcement learning (RL) presents a promising approach for learning reinforced policies from offline datasets without the need for costly or unsafe interactions with the environment. However, datasets collected by humans in real-world environments are often noisy and may even be malicious…

2023

MeGraph: Capturing Long-Range Interactions by Alternating Local and Hierarchical Aggregation on Multi-Scaled Graph Hierarchy

NeurIPS 2023poster

Graph neural networks, which typically exchange information between local neighbors, often struggle to capture long-range interactions (LRIs) within the graph. Building a graph hierarchy via graph pooling methods is a promising approach to address this challenge; however, hierarchical information pr…

2023

SBlimp: Design, Model, and Translational Motion Control for a Swing-Blimp

IROS 2023poster

We present an aerial vehicle composed of a custom quadrotor with tilted rotors and a helium balloon, called SBlimp. We propose a novel control strategy that takes advantage of the natural stable attitude of the blimp to control translational motion. Different from cascade controllers in the literatu…

Cited by 4SourceScholar
2023

Toward Fine Contact Interactions: Learning to Control Normal Contact Force with Limited Information

ICRA 2023poster

Dexterous manipulation of objects through fine control of physical contacts is essential for many important tasks of daily living. A fundamental ability underlying fine contact control is compliant control, i.e., controlling the contact forces while moving. For robots, the most widely explored appro…

Cited by 2SourceScholar