← Search

Jiawei Huang

34 accepted papers

2026

Achieving Structurally Robust Gromov Wasserstein Distance via Adaptive Dual-Mask

ICML 2026poster

The Gromov-Wasserstein (GW) distance enables comparison across different spaces but remains fragile to structural noise due to its global quadratic coupling. Existing robust extensions primarily rely on node-centric mass relaxation. However, we argue that this strategy is far from sufficient: it onl…

Cited by 0SourceScholar
2026

LoBCD-GW: A Fast and Data-Dependent Algorithm for Computing Gromov-Wasserstein Distance via Localized Block Coordinate Descent

ICML 2026poster

The Gromov-Wasserstein (GW) distance provides a powerful framework for aligning structured data by comparing the intrinsic geometries of metric measure spaces, and has become a fundamental tool in machine learning. Most existing methods leverage entropy regularization to reduce the computational com…

Cited by 0SourceScholar
2026

Risk-Sensitive Reinforcement Learning for Alleviating Exploration Dilemmas in Large Language Models

ICLR 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has proven effective for enhancing Large Language Models (LLMs) on complex reasoning tasks. Yet current methods face an exploration dilemma: standard RL struggles to escape the local optima of pre-trained LLMs’ sharply peaked initial policies, bo…

Cited by 0SourceScholar
2025

Bootstrap Your Uncertainty: Adaptive Robust Classification Driven by Optimal-Transport

NeurIPS 2025poster

Deep learning models often struggle with distribution shifts between training and deployment environments. Distributionally Robust Optimization (DRO) offers a promising framework by optimizing worst-case performance over a set of candidate distributions, which is called as the \emph{uncertainty set}…

Cited by 0SourceScholar
2025

Can RLHF be More Efficient with Imperfect Reward Models? A Policy Coverage Perspective

ICML 2025poster

Sample efficiency is critical for online Reinforcement Learning from Human Feedback (RLHF). While existing works investigate sample-efficient online exploration strategies, the potential of utilizing misspecified yet relevant reward models to accelerate learning remains underexplored. This paper stu…

2025

Learning to Steer Markovian Agents under Model Uncertainty

ICLR 2025poster

Designing incentives for an adapting population is a ubiquitous problem in a wide array of economic applications and beyond. In this work, we study how to design additional rewards to steer multi-agent systems towards desired policies \emph{without} prior knowledge of the agents' underlying learning…

2025

Macaque-Motion-Monitor Dataset: A New Benchmark for Macaque Action Recognition

ICASSP 2025accepted

Recent advancements in computational techniques significantly impact bioengineering, particularly in drug safety assessments and neuroscience trials using primate models. Macaques are extensively used due to their genetic and physiological similarities to humans. However, there is a scarcity of maca…

Cited by 0SourceScholar
2025

Spatiotemporal Decoupling for Efficient Vision-Based Occupancy Forecasting

CVPR 2025poster

The task of occupancy forecasting (OCF) involves utilizing past and present perception data to predict future occupancy states of autonomous vehicle surrounding environments, which is critical for downstream tasks such as obstacle avoidance and path planning. Existing 3D OCF approaches struggle to p…

2025

T2A-Feedback: Improving Basic Capabilities of Text-to-Audio Generation via Fine-grained AI Feedback

ACL 2025long

Text-to-audio (T2A) generation has achieved remarkable progress in generating a variety of audio outputs from language prompts. However, current state-of-the-art T2A models still struggle to satisfy human preferences for prompt-following and acoustic quality when generating complex multi-event audio…

Cited by 0SourcePDFScholar
2024

AudioGPT: Understanding and Generating Speech, Music, Sound, and Talking Head

AAAI 2024technical

Large language models (LLMs) have exhibited remarkable capabilities across a variety of domains and tasks, challenging our understanding of learning and cognition. Despite the recent success, current LLMs are not capable of processing complex audio information or conducting spoken conversations (lik…

2024

Cam4DOcc: Benchmark for Camera-Only 4D Occupancy Forecasting in Autonomous Driving Applications

CVPR 2024poster

Understanding how the surrounding environment changes is crucial for performing downstream tasks safely and reliably in autonomous driving applications. Recent occupancy estimation techniques using only camera images as input can provide dense occupancy representations of large-scale scenes based on…

2024

Frieren: Efficient Video-to-Audio Generation Network with Rectified Flow Matching

NeurIPS 2024poster

Video-to-audio (V2A) generation aims to synthesize content-matching audio from silent video, and it remains challenging to build V2A models with high generation quality, efficiency, and visual-audio temporal synchrony. We propose Frieren, a V2A model based on rectified flow matching. Frieren regres…

2024

Melting Pot Contest: Charting the Future of Generalized Cooperative Intelligence

NeurIPS 2024poster

Multi-agent AI research promises a path to develop human-like and human-compatible intelligent technologies that complement the solipsistic view of other approaches, which mostly do not consider interactions between agents. Aiming to make progress in this direction, the Melting Pot contest 2023 focu…

Cited by 0SourcePDFScholar
2024

MimicTalk: Mimicking a personalized and expressive 3D talking face in minutes

NeurIPS 2024poster

Talking face generation (TFG) aims to animate a target identity's face to create realistic talking videos. Personalized TFG is a variant that emphasizes the perceptual identity similarity of the synthesized result (from the perspective of appearance and talking style). While previous works typically…

2024

Model-Based RL for Mean-Field Games is not Statistically Harder than Single-Agent RL

ICML 2024poster

We study the sample complexity of reinforcement learning (RL) in Mean-Field Games (MFGs) with model-based function approximation that requires strategic exploration to find a Nash Equilibrium policy. We introduce the Partial Model-Based Eluder Dimension (P-MBED), a more effective notion to character…

2024

On the Statistical Efficiency of Mean-Field Reinforcement Learning with General Function Approximation

AISTATS 2024poster

In this paper, we study the fundamental statistical efficiency of Reinforcement Learning in Mean-Field Control (MFC) and Mean-Field Game (MFG) with general model-based function approximation. We introduce a new concept called Mean-Field Model-Based Eluder Dimension (MF-MBED), which characterizes the…

Cited by 16SourcePDFScholar
2024

Real3D-Portrait: One-shot Realistic 3D Talking Portrait Synthesis

ICLR 2024spotlight

One-shot 3D talking portrait generation aims to reconstruct a 3D avatar from an unseen image, and then animate it with a reference video or audio to generate a talking portrait video. The existing methods fail to simultaneously achieve the goals of accurate 3D avatar reconstruction and stable talkin…

2023

Make-An-Audio: Text-To-Audio Generation with Prompt-Enhanced Diffusion Models

ICML 2023poster

Large-scale multimodal generative modeling has created milestones in text-to-image and text-to-video generation. Its application to audio still lags behind for two main reasons: the lack of large-scale datasets with high-quality text-audio pairs, and the complexity of modeling long continuous audio…

2022

A Minimax Learning Approach to Off-Policy Evaluation in Confounded Partially Observable Markov Decision Processes

ICML 2022oral

We consider off-policy evaluation (OPE) in Partially Observable Markov Decision Processes (POMDPs), where the evaluation policy depends only on observable variables and the behavior policy depends on unobservable latent variables. Existing works either assume no unmeasured confounders, or focus on s…

2022

Coresets for Wasserstein Distributionally Robust Optimization Problems

NeurIPS 2022accept

Wasserstein distributionally robust optimization (\textsf{WDRO}) is a popular model to enhance the robustness of machine learning with ambiguous data. However, the complexity of \textsf{WDRO} can be prohibitive in practice since solving its ``minimax'' formulation requires a great amount of computat…

2022

On the Convergence Rate of Off-Policy Policy Optimization Methods with Density-Ratio Correction

AISTATS 2022poster

In this paper, we study the convergence properties of off-policy policy optimization algorithms with state-action density ratio correction under function approximation setting, where the objective function is formulated as a max-max-min problem. We first clearly characterize the bias of the learning…

Cited by 8SourcePDFScholar
2022

Tiered Reinforcement Learning: Pessimism in the Face of Uncertainty and Constant Regret

NeurIPS 2022accept

We propose a new learning framework that captures the tiered structure of many real-world user-interaction applications, where the users can be divided into two groups based on their different tolerance on exploration risks and should be treated separately. In this setting, we simultaneously maintai…

2022

Towards Deployment-Efficient Reinforcement Learning: Lower Bound and Optimality

ICLR 2022spotlight

Deployment efficiency is an important criterion for many real-world applications of reinforcement learning (RL). Despite the community's increasing interest, there lacks a formal theoretical formulation for the problem. In this paper, we propose such a formulation for deployment-efficient RL (DE-RL)…

Cited by 27SourcePDFScholar
2021

A Novel Sequential Coreset Method for Gradient Descent Algorithms

ICML 2021spotlight

A wide range of optimization problems arising in machine learning can be solved by gradient descent algorithms, and a central question in this area is how to efficiently compress a large-scale dataset so as to reduce the computational complexity. Coreset is a popular data compression technique that…

Cited by 24SourcePDFScholar
2020

WeightNet: Revisiting the Design Space of Weight Networks

ECCV 2020poster

We present a conceptually simple, flexible and effective framework for weight generating networks. Our approach is general that unifies two current distinct and extremely effective SENet and CondConv into the same framework on weight space. The method, called WeightNet, generalizes the two methods b…