← Search

Yong Yu

50 accepted papers

2026

ATGen: Adversarial Reinforcement Learning for Test Case Generation

ICLR 2026poster

Large Language Models (LLMs) show remarkable code generation capabilities but often produce imperfect code with subtle bugs. A critical bottleneck for improving code quality is the scarcity of high-quality test cases. Existing approaches, primarily based on Supervised Fine-Tuning (SFT) over static d…

Cited by 0SourceScholar
2026

AdverMCTS: Combating Pseudo-Correctness in Code Generation via Adversarial Monte Carlo Tree Search

ICML 2026poster

Recent advancements in Large Language Models (LLMs) have successfully employed search-based strategies to enhance code generation. However, existing methods typically rely on static, sparse public test cases for verification, leading to pseudo-correctness—where solutions overfit the visible public t…

Cited by 0SourceScholar
2026

CATArena: Evaluating Evolutionary Capabilities of Code Agents via Iterative Tournaments

ICML 2026poster

Current evaluation for Large Language Model (LLM) code agents predominantly focus on generating functional code in single-turn scenarios, which fails to evaluate the agent's capability for continuous code optimization and multi-turn iterative development. To bridge this gap, we introduce CATArena, a…

Cited by 0SourceScholar
2026

H-Zero: Cross-Humanoid Locomotion Pretraining Enables Few-Shot Novel Embodiment Transfer

ICRA 2026poster

The rapid advancement of humanoid robotics has intensified the need for robust and adaptable controllers to enable stable and efficient locomotion across diverse platforms. However, developing such controllers remains a significant challenge because existing solutions are tailored to specific robot …

2026

Manipulation as in Simulation: Enabling Accurate Geometry Perception in Robots

ICLR 2026poster

Modern robotic manipulation primarily relies on visual observations in a 2D color space for skill learning but suffers from poor generalization. In contrast, humans, living in a 3D world, depend more on physical properties-such as distance, size, and shape-than on texture when interacting with objec…

Cited by 0SourcecodeScholar
2026

Position: Academic Conferences are Potentially Facing Denominator Gaming Caused by Fully Automated Scientific Agents

ICML 2026poster

The implicit policy of maintaining relatively stable acceptance rates at top AI conferences, despite exponentially growing submissions, introduces a critical structural vulnerability. This position paper characterizes a new systemic threat we term ***Agentic Denominator Gaming***, in which a malicio…

Cited by 0SourceScholar
2025

Beyond Graph Convolution: Multimodal Recommendation with Topology-aware MLPs

AAAI 2025technical

Given the large volume of side information from different modalities, multimodal recommender systems have become increasingly vital, as they exploit richer semantic information beyond user-item interactions. Recent works highlight that leveraging Graph Convolutional Networks (GCNs) to explicitly mod…

2025

Boost, Disentangle, and Customize: A Robust System2-to-System1 Pipeline for Code Generation

ACL 2025finding

To address these limitations, we propose BDC, a novel framework that Boosts reasoning exploration via multi-agent collaboration, Disentangles heterogeneous data into specialized experts, and Customizes solutions through dynamic model composition. BDC integrates a Monte Carlo Tree-of-Agents algorithm…

Cited by 0SourcePDFScholar
2025

CodePRM: Execution Feedback-enhanced Process Reward Model for Code Generation

ACL 2025finding

Code generation is a critical reasoning task for large language models (LLMs). Recent advancements have focused on optimizing the thought process of code generation, achieving significant improvements. However, such thought process lacks effective process supervision, making it hard to optimize the…

2025

DebateCoder: Towards Collective Intelligence of LLMs via Test Case Driven LLM Debate for Code Generation

ACL 2025long

With the impressive reasoning and text generation capabilities of large language models (LLMs), methods leveraging multiple LLMs to debate each other have garnered increasing attention. However, existing debate-based approaches remain limited in effectiveness in structured and detailed domains repre…

2025

Large Language Models are Demonstration Pre-Selectors for Themselves

ICML 2025poster

In-context learning with large language models (LLMs) delivers strong few-shot performance by choosing few-shot demonstrations from the entire training dataset. However, previous few-shot in-context learning methods, which calculate similarity scores for choosing demonstrations, incur high computati…

Cited by 0SourcePDFScholar
2025

NL-Debugging: Exploiting Natural Language as an Intermediate Representation for Code Debugging

EMNLP 2025

Debugging is a critical aspect of LLM’s coding ability. Early debugging efforts primarily focused on code-level analysis, which often falls short when addressing complex programming errors that require a deeper understanding of algorithmic logic. Recent advancements in large language models (LLMs) h

2025

RethinkMCTS: Refining Erroneous Thoughts in Monte Carlo Tree Search for Code Generation

EMNLP 2025

Tree search methods have demonstrated impressive performance in code generation. Previous methods combine tree search with reflection that summarizes past mistakes to achieve iterative improvement. However, these methods face significant challenges. First, they search directly within the code langua

2025

Retrieval-Augmented Process Reward Model for Generalizable Mathematical Reasoning

ACL 2025finding

While large language models (LLMs) have significantly advanced mathematical reasoning, Process Reward Models (PRMs) have been developed to evaluate the logical validity of reasoning steps. However, PRMs still struggle with out-of-distribution (OOD) challenges. This paper identifies the OOD issues in…

2025

Stop DDoS Attacking the Research Community with AI-Generated Survey Papers

NeurIPS 2025oral

Survey papers are foundational to the scholarly progress of research communities, offering structured overviews that guide both novices and experts across disciplines. However, the recent surge of AI-generated surveys, especially enabled by large language models (LLMs), has transformed this traditio…

Cited by 0SourceScholar
2025

World Model-Based Perception for Visual Legged Locomotion

ICRA 2025

Legged locomotion over various terrains is challenging and requires precise perception of the robot and its surroundings from both proprioception and vision. However, learning directly from high-dimensional visual input is often data-inefficient and intricate. To address this issue, traditional meth

Cited by 23SourcecodeScholar
2024

MADiff: Offline Multi-agent Learning with Diffusion Models

NeurIPS 2024poster

Offline reinforcement learning (RL) aims to learn policies from pre-existing datasets without further interactions, making it a challenging task. Q-learning algorithms struggle with extrapolation errors in offline settings, while supervised learning methods are constrained by model expressiveness. R…

2023

Learning Decomposed Spatial Relations for Multi-Variate Time-Series Modeling

AAAI 2023technical

Modeling multi-variate time-series (MVTS) data is a long-standing research subject and has found wide applications. Recently, there is a surge of interest in modeling spatial relations between variables as graphs, i.e., first learning one static graph for each dataset and then exploiting the graph s…

Cited by 20SourcePDFScholar
2023

Lending Interaction Wings to Recommender Systems with Conversational Agents

NeurIPS 2023poster

An intelligent conversational agent (a.k.a., chat-bot) could embrace conversational technologies to obtain user preferences online, to overcome inherent limitations of recommender systems trained over the offline historical user behaviors. In this paper, we propose CORE, a new offline-training and o…

Cited by 15SourcePDFScholar
2023

Set-to-Sequence Ranking-Based Concept-Aware Learning Path Recommendation

AAAI 2023technical

With the development of the online education system, personalized education recommendation has played an essential role. In this paper, we focus on developing path recommendation systems that aim to generating and recommending an entire learning path to the given user in each session. Noticing that…

Cited by 11SourcePDFScholar
2023

Sim-to-Real Transfer for Quadrupedal Locomotion via Terrain Transformer

ICRA 2023poster

Deep reinforcement learning has recently emerged as an appealing alternative for legged locomotion over multiple terrains by training a policy in physical simulation and then transferring it to the real world (i.e., sim-to-real transfer). Despite considerable progress, the capacity and scalability o…

Cited by 22SourceScholar
2022

Graph-Enhanced Exploration for Goal-oriented Reinforcement Learning

ICLR 2022poster

Goal-oriented Reinforcement Learning (GoRL) is a promising approach for scaling up RL techniques on sparse reward environments requiring long horizon planning. Recent works attempt to build suitable abstraction graph of the environment and enhance GoRL with classical graphical methods such as shorte…

Cited by 4SourcePDFScholar
2022

Honor of Kings Arena: an Environment for Generalization in Competitive Reinforcement Learning

NeurIPS 2022accept

This paper introduces Honor of Kings Arena, a reinforcement learning (RL) environment based on the Honor of Kings, one of the world’s most popular games at present. Compared to other environments studied in most previous work, ours presents new generalization challenges for competitive reinforcement…

2022

Inductive Relation Prediction Using Analogy Subgraph Embeddings

ICLR 2022poster

Prevailing methods for relation prediction in heterogeneous graphs aim at learning latent representations (i.e., embeddings) of observed nodes and relations, and thus are limited to the transductive setting where the relation types must be known during training. Here, we propose ANalogy SubGraphE…

Cited by 7SourcePDFScholar
2022

Multi-View Graph Representation for Programming Language Processing: An Investigation into Algorithm Detection

AAAI 2022technical

Program representation, which aims at converting program source code into vectors with automatically extracted features, is a fundamental problem in programming language processing (PLP). Recent work tries to represent programs with neural networks based on source code structures. However, such meth…

2022

PAEG: Phrase-level Adversarial Example Generation for Neural Machine Translation

COLING 2022main

While end-to-end neural machine translation (NMT) has achieved impressive progress, noisy input usually leads models to become fragile and unstable. Generating adversarial examples as the augmented data has been proved to be useful to alleviate this problem. Existing methods for adversarial example…

Cited by 10SourcePDFScholar
2022

Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization

ICML 2022spotlight

Recent progress in state-only imitation learning extends the scope of applicability of imitation learning to real-world settings by relieving the need for observing expert actions. However, existing solutions only learn to extract a state-to-action mapping policy from the data, without considering h…

2022

Promoting Quality and Diversity in Population-based Reinforcement Learning via Hierarchical Trajectory Space Exploration

ICRA 2022poster

Quality Diversity (QD) algorithms in population-based reinforcement learning aim to optimize agents' returns and diversity among the population simultaneously. It is conducive to solving exploration problems in reinforcement learning and potentially getting multiple good and diverse strategies. Howe…

Cited by 5SourceScholar
2022

Why Propagate Alone? Parallel Use of Labels and Features on Graphs

ICLR 2022poster

One of the challenges of graph-based semi-supervised learning over ordinary supervised learning for classification tasks lies in label utilization. The direct use of ground-truth labels in graphs for training purposes can result in a parametric model learning trivial degenerate solutions (e.g., an…

Cited by 12SourcePDFScholar
2021

Glancing Transformer for Non-Autoregressive Neural Machine Translation

ACL 2021long

Recent work on non-autoregressive neural machine translation (NAT) aims at improving the efficiency by parallel decoding without sacrificing the quality. However, existing NAT methods are either inferior to Transformer or require multiple decoding passes, leading to reduced speedup. We propose the G…

2021

Learning Logic Rules for Document-Level Relation Extraction

EMNLP 2021main

Document-level relation extraction aims to identify relations between entities in a whole document. Prior efforts to capture long-range dependencies have relied heavily on implicitly powerful representations learned through (graph) neural networks, which makes the model less transparent. To tackle t…

2021

MARS: Markov Molecular Sampling for Multi-objective Drug Discovery

ICLR 2021spotlight

Searching for novel molecules with desired chemical properties is crucial in drug discovery. Existing work focuses on developing neural models to generate either molecular sequences or chemical graphs. However, it remains a big challenge to find novel and diverse compounds satisfying several propert…

2021

MapGo: Model-Assisted Policy Optimization for Goal-Oriented Tasks

IJCAI 2021poster

In Goal-oriented Reinforcement learning, relabeling the raw goals in past experience to provide agents with hindsight ability is a major solution to the reward sparsity problem. In this paper, to enhance the diversity of relabeled goals, we develop FGI (Foresight Goal Inference), a new relabeling st…

2021

On Effective Scheduling of Model-based Reinforcement Learning

NeurIPS 2021poster

Model-based reinforcement learning has attracted wide attention due to its superior sample efficiency. Despite its impressive success so far, it is still unclear how to appropriately schedule the important hyperparameters to achieve adequate performance, such as the real data ratio for policy optimi…

2021

Universal Trading for Order Execution with Oracle Policy Distillation

AAAI 2021technical

As a fundamental problem in algorithmic trading, order execution aims at fulfilling a specific trading order, either liquidation or acquirement, for a given instrument. Towards effective execution strategy, recent years have witnessed the shift from the analytical view with model-based market assump…

Cited by 60SourcePDFScholar
2020

Aggregating Crowd Wisdom with Side Information via a Clustering-based Label-aware Autoencoder

IJCAI 2020poster

Aggregating crowd wisdom infers true labels for objects, from multiple noisy labels provided by various sources. Besides labels from sources, side information such as object features is also introduced to achieve higher inference accuracy. Usually, the learning-from-crowds framework is adopted. Howe…

2020

DropNAS: Grouped Operation Dropout for Differentiable Architecture Search

IJCAI 2020poster

Neural architecture search (NAS) has shown encouraging results in automating the architecture design. Recently, DARTS relaxes the search process with a differentiable formulation that leverages weight-sharing and SGD for cost reduction of NAS. In DARTS, all candidate operations are trained simulta…

2020

Efficient and Robust High-Dimensional Linear Contextual Bandits

IJCAI 2020poster

The linear contextual bandits is a sequential decision-making problem where an agent decides among sequential actions given their corresponding contexts. Since large-scale data sets become more and more common, we study the linear contextual bandits in high-dimensional situations. Recent works focus…

Cited by 0SourcePDFScholar
2020

Improving Knowledge Tracing via Pre-training Question Embeddings

IJCAI 2020poster

Knowledge tracing (KT) defines the task of predicting whether students can correctly answer questions based on their historical response. Although much research has been devoted to exploiting the question information, plentiful advanced information among questions and skills hasn't been well extract…

2020

Model-based Policy Optimization with Unsupervised Model Adaptation

NeurIPS 2020spotlight

Model-based reinforcement learning methods learn a dynamics model with real data sampled from the environment and leverage it to generate simulated data to derive an agent. However, due to the potential distribution mismatch between simulated data and real data, this could lead to degraded performan…

2020

Multi-Agent Interactions Modeling with Correlated Policies

ICLR 2020poster

In multi-agent systems, complex interacting behaviors arise due to the high correlations among agents. However, previous work on modeling multi-agent interactions from demonstrations is primarily constrained by assuming the independence among policies and their reward structures. In this paper, we…

Cited by 27SourcecodeScholar
2019

AdaShift: Decorrelation and Convergence of Adaptive Learning Rate Methods

ICLR 2019poster

Adam is shown not being able to converge to the optimal solution in certain cases. Researchers recently propose several algorithms to avoid the issue of non-convergence of Adam, but their efficiency turns out to be unsatisfactory in practice. In this paper, we provide a new insight into the non-conv…

Cited by 80SourcePDFScholar
2019

Lipschitz Generative Adversarial Nets

ICML 2019oral

In this paper we show that generative adversarial networks (GANs) without restriction on the discriminative function space commonly suffer from the problem that the gradient produced by the discriminator is uninformative to guide the generator. By contrast, Wasserstein GAN (WGAN), where the discrimi…

Cited by 107SourcePDFScholar
2018

Activation Maximization Generative Adversarial Nets

ICLR 2018poster

Class labels have been empirically shown useful in improving the sample quality of generative adversarial nets (GANs). In this paper, we mathematically study the properties of the current variants of GANs that make use of class label information. With class aware gradient and cross-entropy decomposi…

2018

Path-Level Network Transformation for Efficient Architecture Search

ICML 2018oral

We introduce a new function-preserving transformation for efficient neural architecture search. This network transformation allows reusing previously trained networks and existing successful architectures that improves sample efficiency. We aim to address the limitation of current network transforma…