← Search

Chaojie Wang

22 accepted papers

2026

Human-AI Curation Synergy: Scaling Preference Data Curation via Human-Guided AI Feedback

ICLR 2026poster

Despite the critical role of reward models (RMs) in reinforcement learning from human feedback (RLHF), current state-of-the-art open RMs perform poorly on most existing evaluation benchmarks, failing to capture the spectrum of nuanced and sophisticated human preferences. Even approaches incorporatin…

Cited by 0SourcecodeScholar
2025

Cradle: Empowering Foundation Agents towards General Computer Control

ICML 2025poster

Despite their success in specific scenarios, existing foundation agents still struggle to generalize across various virtual scenarios, mainly due to the dramatically different encapsulations of environments with manually designed observation and action spaces. To handle this issue, we propose the Ge…

2025

DAPO : Improving Multi-Step Reasoning Abilities of Large Language Models with Direct Advantage-Based Policy Optimization

NeurIPS 2025spotlight

The role of reinforcement learning (RL) in enhancing the reasoning of large language models (LLMs) is becoming increasingly significant. Despite the success of RL in many scenarios, there are still many challenges in improving the reasoning of LLMs. One key challenge is the sparse reward, which intr…

Cited by 0SourceScholar
2025

Incentivizing LLMs to Self-Verify Their Answers

NeurIPS 2025poster

Large Language Models (LLMs) have demonstrated remarkable progress in complex reasoning tasks through both post-training and test-time scaling laws. While prevalent test-time scaling approaches are often realized by using external reward models to guide the model generation process, we find that onl…

Cited by 0SourcecodeScholar
2025

Removing Prompt-template Bias in Reinforcement Learning from Human Feedback

ACL 2025finding

Reinforcement Learning from Human Feedback (RLHF) has become an essential technique for enhancing pre-trained large language models (LLMs) to generate responses that align with human preferences and societal values. Although RLHF has shown promise, the training of reward models (RMs) still faces the…

Cited by 0SourcePDFScholar
2024

Candidate Label Set Pruning: A Data-centric Perspective for Deep Partial-label Learning

ICLR 2024oral

Partial-label learning (PLL) allows each training example to be equipped with a set of candidate labels. Existing deep PLL research focuses on a \emph{learning-centric} perspective to design various training strategies for label disambiguation i.e., identifying the concealed true label from the cand…

Cited by 6SourcePDFScholar
2024

Improving Unsupervised Hierarchical Representation with Reinforcement Learning

CVPR 2024poster

Learning representations to capture the very fundamental understanding of the world is a key challenge in machine learning. The hierarchical structure of explanatory factors hidden in data is such a general representation and could be potentially achieved with a hierarchical VAE. However training a…

2024

Latent Logic Tree Extraction for Event Sequence Explanation from LLMs

ICML 2024poster

Modern high-stakes systems, such as healthcare or robotics, often generate vast streaming event sequences. Our goal is to design an efficient, plug-and-play tool to elicit logic tree-based explanations from Large Language Models (LLMs) to provide customized insights into each observed event sequence…

Cited by 5SourcePDFScholar
2023

Few-shot Generation via Recalling Brain-Inspired Episodic-Semantic Memory

NeurIPS 2023poster

Aimed at adapting a generative model to a novel generation task with only a few given data samples, the capability of few-shot generation is crucial for many real-world applications with limited data, \emph{e.g.}, artistic domains. Instead of training from scratch, recent works tend to leverage the…

Cited by 6SourcePDFScholar
2022

A Variational Edge Partition Model for Supervised Graph Representation Learning

NeurIPS 2022accept

Graph neural networks (GNNs), which propagate the node features through the edges and learn how to transform the aggregated features under label supervision, have achieved great success in supervised feature extraction for both node-level and graph-level classification tasks. However, GNNs typicall…

2022

Alleviating "Posterior Collapse'' in Deep Topic Models via Policy Gradient

NeurIPS 2022accept

Deep topic models have been proven as a promising way to extract hierarchical latent representations from documents represented as high-dimensional bag-of-words vectors. However, the representation capability of existing deep topic models is still limited by the phenomenon of "posterior collapse", w…

Cited by 10SourcePDFScholar
2022

Bayesian Deep Embedding Topic Meta-Learner

ICML 2022spotlight

Existing deep topic models are effective in capturing the latent semantic structures in textual data but usually rely on a plethora of documents. This is less than satisfactory in practical applications when only a limited amount of data is available. In this paper, we propose a novel framework that…

Cited by 6SourcePDFScholar
2022

Knowledge-Aware Bayesian Deep Topic Model

NeurIPS 2022accept

We propose a Bayesian generative model for incorporating prior domain knowledge into hierarchical topic modeling. Although embedded topic models (ETMs) and its variants have gained promising performance in text analysis, they mainly focus on mining word co-occurrence patterns, ignoring potentially e…

2022

Out-of-Distribution Detection with An Adaptive Likelihood Ratio on Informative Hierarchical VAE

NeurIPS 2022accept

Unsupervised out-of-distribution (OOD) detection is essential for the reliability of machine learning. In the literature, existing work has shown that higher-level semantics captured by hierarchical VAEs can be used to detect OOD instances. However, we empirically show that, the inherent issue of hi…

Cited by 24SourcePDFScholar
2021

EnsLM: Ensemble Language Model for Data Diversity by Semantic Clustering

ACL 2021long

Natural language processing (NLP) often faces the problem of data diversity such as different domains, themes, styles, and so on. Therefore, a single language model (LM) is insufficient to learn all knowledge from diverse samples. To solve this problem, we firstly propose an autoencoding topic model…

2021

Leveraging Table Content for Zero-shot Text-to-SQL with Meta-Learning

AAAI 2021technical

Single-table text-to-SQL aims to transform a natural language question into a SQL query according to one single table. Recent work has made promising progress on this task by pre-trained language models and a multi-submodule framework. However, zero-shot table, that is, the invisible table in the t…

2021

Sawtooth Factorial Topic Embeddings Guided Gamma Belief Network

ICML 2021spotlight

Hierarchical topic models such as the gamma belief network (GBN) have delivered promising results in mining multi-layer document representations and discovering interpretable topic taxonomies. However, they often assume in the prior that the topics at each layer are independently drawn from the Diri…

2021

TopicNet: Semantic Graph-Guided Topic Discovery

NeurIPS 2021poster

Existing deep hierarchical topic models are able to extract semantically meaningful topics from a text corpus in an unsupervised manner and automatically organize them into a topic hierarchy. However, it is unclear how to incorporate prior belief such as knowledge graph to guide the learning of th…

2020

Bidirectional Convolutional Poisson Gamma Dynamical Systems

NeurIPS 2020poster

Incorporating the natural document-sentence-word structure into hierarchical Bayesian modeling, we propose convolutional Poisson gamma dynamical systems (PGDS) that introduce not only word-level probabilistic convolutions, but also sentence-level stochastic temporal transitions. With word-level conv…

2020

Deep Relational Topic Modeling via Graph Poisson Gamma Belief Network

NeurIPS 2020poster

To analyze a collection of interconnected documents, relational topic models (RTMs) have been developed to describe both the link structure and document content, exploring their underlying relationships via a single-layer latent representation with limited expressive capability. To better utilize th…

2020

Learning Dynamic Hierarchical Topic Graph with Graph Convolutional Network for Document Classification

AISTATS 2020poster

Constructing a graph with graph convolutional network (GCN) to explore the relational structure of the data has attracted lots of interests in various tasks. However, for document classification, existing graph based methods often focus on the straightforward word-word and word-document relations,…