← Search

Zhiwei Jia

11 accepted papers

2025

Reward Fine-Tuning Two-Step Diffusion Models via Learning Differentiable Latent-Space Surrogate Reward

CVPR 2025poster

Recent research has shown that fine-tuning diffusion models (DMs) with arbitrary rewards, including non-differentiable ones, is feasible with reinforcement learning (RL) techniques, enabling flexible model alignment. However, applying existing RL methods to step-distilled DMs is challenging for ultr…

Cited by 0SourcePDFScholar
2024

Chain-of-Thought Predictive Control

ICML 2024poster

We study generalizable policy learning from demonstrations for complex low-level control (e.g., contact-rich object manipulations). We propose a novel hierarchical imitation learning method that utilizes sub-optimal demos. Firstly, we propose an observation space-agnostic approach that efficiently d…

2023

KAFA: Rethinking Image Ad Understanding with Knowledge-Augmented Feature Adaptation of Vision-Language Models

ACL 2023industry

Image ad understanding is a crucial task with wide real-world applications. Although highly challenging with the involvement of diverse atypical scenes, real-world entities, and reasoning over scene-texts, how to interpret image ads is relatively under-explored, especially in the era of foundational…

Cited by 5SourcePDFScholar
2023

MetaCLUE: Towards Comprehensive Visual Metaphors Research

CVPR 2023poster

Creativity is an indispensable part of human cognition and also an inherent part of how we make sense of the world. Metaphorical abstraction is fundamental in communicating creative ideas through nuanced relationships between abstract concepts such as feelings. While computer vision benchmarks and a…

2022

Improving Policy Optimization with Generalist-Specialist Learning

ICML 2022spotlight

Generalization in deep reinforcement learning over unseen environment variations usually requires policy learning over a large set of diverse training variations. We empirically observe that an agent trained on many variations (a generalist) tends to learn faster at the beginning, yet its performanc…

2022

Learning to Act with Affordance-Aware Multimodal Neural SLAM

IROS 2022poster

Recent years have witnessed an emerging paradigm shift toward embodied artificial intelligence, in which an agent must learn to solve challenging tasks by interacting with its environment. There are several challenges in solving embodied multimodal tasks, including long-horizon planning, vision-and-…

Cited by 18SourcecodeScholar
2021

ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrations

NeurIPS 2021poster

Object manipulation from 3D visual inputs poses many challenges on building generalizable perception and policy models. However, 3D assets in existing benchmarks mostly lack the diversity of 3D shapes that align with real-world intra-class complexity in topology and geometry. Here we propose SAPIEN…

Cited by 145SourcecodeScholar
2021

Semantically Robust Unpaired Image Translation for Data With Unmatched Semantics Statistics

ICCV 2021poster

Many applications of unpaired image-to-image translation require the input contents to be preserved semantically during translations. Unaware of the inherently unmatched semantics distributions between source and target domains, existing distribution matching methods (i.e., GAN-based) can give undes…

Cited by 28PDFcodeScholar
2020

One-Pixel Signature: Characterizing CNN Models for Backdoor Detection

ECCV 2020poster

We tackle the convolution neural networks (CNNs) backdoor detection problem by proposing a new representation called one-pixel signature. Our task is to detect/classify if a CNN model has been maliciously inserted with an unknown Trojan trigger or not. Here, each CNN model is associated with a signa…

Cited by 77SourcePDFScholar
2020

Refactoring Policy for Compositional Generalizability using Self-Supervised Object Proposals

NeurIPS 2020poster

We study how to learn a policy with compositional generalizability. We propose a two-stage framework, which refactorizes a high-reward teacher policy into a generalizable student policy with strong inductive bias. Particularly, we implement an object-centric GNN-based student policy, whose input obj…