← Search

Jianye HAO

173 accepted papers

2026

$\textbf{Re}^{2}$: Unlocking LLM Reasoning via Reinforcement Learning with Re-solving

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has shown promise in enhancing the reasoning performance of large language models (LLMs) by increasing test-time compute. However, even after extensive RLVR training, such models still tend to generate unnecessary and low-quality steps in their…

Cited by 0SourceScholar
2026

A Hierarchical Circuit Symbolic Discovery Framework for Efficient Logic Optimization

ICLR 2026poster

The efficiency of Logic Optimization (LO) has become one of the key bottlenecks in chip design. To prompt efficient LO, many graph-based machine learning (ML) methods, such as graph neural networks (GNNs), have been proposed to predict and prune a large number of ineffective subgraphs of the LO heur…

Cited by 0SourceScholar
2026

Adversarial Latent Embedding Repair for LLM Continual Learning

ICML 2026poster

Research on continual learning for LLMs seeks to acquire new skills without catastrophic forgetting of established prior knowledge. However, domain-specific fine-tuning still triggers severe, long-tailed forgetting issues even under narrow updates, particularly when the pre-training data is inaccess…

Cited by 0SourceScholar
2026

AgentSwift: Efficient LLM Agent Design via Value-Guided Hierarchical Search

AAAI 2026technical

Large language model (LLM) agents have demonstrated strong capabilities across diverse domains, yet automated agent design remains a significant challenge. Current automated agent design approaches are often constrained by limited search spaces that primarily optimize workflows but fail to integrate

Cited by 0SourcePDFScholar
2026

Boosting Multi-Domain Reasoning of LLMs via Curvature-Guided Policy Optimization

ICLR 2026poster

Multi-domain reinforcement learning (RL) for large language models (LLMs) involves highly intricate reward surfaces, posing significant challenges in finding parameters that excel across all domains. Recent empirical studies have further highlighted conflicts among domains, where gains in one capabi…

Cited by 0SourcecodeScholar
2026

Bottleneck-Guided Spectral Subgoals For Offline Goal-Conditioned RL

ICML 2026poster

Offline goal-conditioned RL (OGCRL) learns to reach arbitrary goals from offline dataset, but long-horizon performance hinges on crossing a handful of hard-to-cross bottlenecks. These bottlenecks not only dictate the feasible paths toward the goal but also act as critical keypoints, marking the tran…

Cited by 0SourceScholar
2026

CellAgent: LLM-Driven Multi-Agent Framework for Natural Language-Based Single-Cell Analysis

ICLR 2026poster

Single-cell RNA sequencing (scRNA-seq) and spatial transcriptomics (ST) data analysis are pivotal for advancing biological research, enabling precise characterization of cellular heterogeneity. However, existing analysis approaches require extensive manual programming and complex tool integration, p…

Cited by 0SourcecodeScholar
2026

D-ARL: A Distribution-Matched Asynchronous Reinforcement Learning Framework for Language Reasoning

ICML 2026poster

Asynchronous reinforcement learning (RL) has shown notable success in accelerating the post-training of large language models (LLMs). However, its decoupled data generation and training paradigm introduces a fundamental distributional mismatch between data generated by stale behavior policies and cu…

Cited by 0SourceScholar
2026

EVLP: Learning Unified Embodied Vision-Language Planner with Reinforced Supervised Fine-Tuning

ICLR 2026poster

In complex embodied long-horizon manipulation tasks, effective task decomposition and execution require synergistic integration of textual logical reasoning and visual-spatial imagination to ensure efficient and accurate operation. Current methods fail to adopt a unified generation framework for mul…

Cited by 0SourceScholar
2026

Embodied-R1: Reinforced Embodied Reasoning for General Robotic Manipulation

ICLR 2026poster

Generalization in embodied AI is hindered by the "seeing-to-doing gap", stemming from data scarcity and embodiment heterogeneity. To address this, we pioneer "pointing" as a unified, embodiment-agnostic intermediate representation, defining four core embodied pointing abilities that bridge high-leve…

Cited by 0SourcecodeScholar
2026

Evolving Graph Structured Programs for Circuit Generation with Large Language Models

ICLR 2026poster

Logic synthesis (LS), which aims to generate a *compact* logic circuit graph with minimized size while *accurately* satisfying a given functionality, plays an important role in chip design. However, existing LS methods struggle to balance circuit structure compactness and functional accuracy, often…

Cited by 0SourceScholar
2026

Following the Navigation: Enhancing Small Language Models Contextual Reasoning with LLM Guidance

ICLR 2026poster

Large language models (LLMs), such as OpenAI-o1 and DeepSeek-R1, excel in contextual reasoning by leveraging extensive world knowledge and deep contextual understanding. However, their high computational costs limit deployment in resource-constrained settings. Conversely, small language models (SLMs…

Cited by 0SourceScholar
2026

From Seeing to Doing: Bridging Reasoning and Decision for Robotic Manipulation

ICLR 2026poster

Achieving generalization in robotic manipulation remains a critical challenge, particularly for unseen scenarios and novel tasks. Current Vision-Language-Action (VLA) models, while building on top of general Vision-Language Models (VLMs), still fall short of achieving robust zero-shot performance du…

Cited by 0SourcecodeScholar
2026

Global Prior Meets Local Consistency: Dual-Memory Augmented Vision-Language-Action Model for Efficient Robotic Manipulation

CVPR 2026

Hierarchical Vision-Language-Action (VLA) models have rapidly become a dominant paradigm for robotic manipulation. It typically comprising a Vision-Language backbone for perception and understanding, together with a generative policy for action generation. However, its performance is increasingly bo

Cited by 0SourcecodeScholar
2026

Human-like Abstract Visual Reasoning via Understanding and Solving Reasoning Loop

CVPR 2026

Abstract visual reasoning benchmarks such as ARC-AGI evaluate the ability to infer generalizable transformation rules from few graphical demonstrations, a capability where current deep learning models severely underperform. Mainstream LLMs achieve only 15.8% (DeepSeek-R1) and 34.5% (o3-mini-high) ac

Cited by 0SourceScholar
2026

K²-Agent: Co-Evolving Know-What and Know-How for Hierarchical Mobile Device Control

ICLR 2026poster

Existing mobile device control agents often perform poorly when solving complex tasks requiring long-horizon planning and precise operations, typically due to a lack of relevant task experience or unfamiliarity with skill execution. We propose $\textbf{K²-Agent}$, a hierarchical framework that model…

Cited by 0SourcecodeScholar
2026

Latent-Guided Reasoning: Empowering Small LLMs with Large-Model Thinking

ICLR 2026poster

Large Language Models (LLMs) have demonstrated remarkable capabilities in complex reasoning tasks, but their high computational costs limit their widespread practical application. We argue that this inefficiency arises from the tight coupling of high-level cognitive planning (devising the solution s…

Cited by 0SourceScholar
2026

Neuro-evolutionary Continual Reinforcement Learning

ICML 2026spotlight

Deploying robots in open‑ended real‑world environments demands continual learning capabilities to adapt to an ever-expanding range of tasks. This requires retaining previously acquired skills without forgetting while effectively leveraging prior knowledge to learn new ones. Inspired by neuroscience,…

Cited by 0SourceScholar
2026

OmniEVA: Embodied Versatile Planner via Task-Adaptive 3D-Grounded and Embodiment-aware Reasoning

ICLR 2026poster

Recent advances in multimodal large language models (MLLMs) have opened new opportunities for embodied intelligence, enabling multimodal understanding, reasoning, and interaction, as well as continuous spatial decision-making. Nevertheless, current MLLM-based embodied systems face two critical limit…

Cited by 0SourceScholar
2026

One Demo Is All It Takes: Planning Domain Derivation with LLMs from A Single Demonstration

ICLR 2026poster

Pre-trained large language models (LLMs) show promise for robotic task planning but often struggle to guarantee correctness in long-horizon problems. Task and motion planning (TAMP) addresses this by grounding symbolic plans in low-level execution, yet it relies heavily on manually engineered planni…

Cited by 0SourceScholar
2026

OpenPyRo-A1: An Open Python-Based Low-Cost Bimanual Robot for Embodied AI

RA-L 2026

Many real-world tasks, such as assembly, cooking, and object handovers, require bi-manual coordination. Learning such skills via imitation remains challenging due to dataset scarcity, mainly caused by the high cost of bi-manual robotic platforms and barriers to entry in robotics software. To address

Cited by 1SourceScholar
2026

OpenPyRo-A1: An Open Python-Based Low-Cost Bimanual Robot for Embodied AI

ICRA 2026poster

Many real-world tasks, such as assembly, cooking, and object handovers, require bi-manual coordination. Learning such skills via imitation remains challenging due to dataset scarcity, mainly caused by the high cost of bi-manual robotic platforms and barriers to entry in robotics software. To address…

Cited by 0SourceScholar
2026

Opt-Miner: Empowering Information-Seeking Agent with Tree-Guided Data Synthesis for Optimization Modeling

ICML 2026poster

Large Language Model (LLM) agents have shown significant potential in automated optimization modeling for mathematical problems. However, real-world problems are still challenging due to their knowledge-intensive nature. Existing methods, constrained by static parametric knowledge, often lack the do…

Cited by 0SourceScholar
2026

Opt-Verifier: Unleashing the Power of LLMs for Optimization Modeling via Dual-Side Verification

ICML 2026poster

Building mathematical optimization models is critical in operations research (OR), while it requires substantial human expertise. Recent advancements have utilized large language models (LLMs) to automate this modeling process. However, existing works often struggle to verify the correctness of the …

Cited by 0SourceScholar
2026

SemanticVLA: Towards Semantic Reasoning over Action Memorization via Synergistic Explicit Trace and Latent Action Planning

CVPR 2026

Vision-Language-Action (VLA) models have emerged as a promising paradigm where pretrained Vision-Language Models (VLMs) serve as System 2 for high-level reasoning, connected to action experts as System 1 for low-level motor control.However, current works fail to genuinely leverage VLM capabilities:

Cited by 0SourceScholar
2026

Short Chains, Deep Thoughts: Balancing Reasoning Efficiency and Intra-Segment Capability via Split-Merge Optimization

ICML 2026poster

While Large Reasoning Models (LRMs) have demonstrated impressive capabilities in solving complex tasks through the generation of long reasoning chains, this reliance on verbose generation results in significant latency and computational overhead. To address these challenges, we propose \textbf{CoSMo…

Cited by 0SourceScholar
2026

Squeeze the Soaked Sponge: Efficient Off-policy RFT for Large Language Model

ICLR 2026poster

Reinforcement Learning (RL) has demonstrated its potential to improve the reasoning ability of Large Language Models (LLMs), yet most existing Reinforcement Finetuning (RFT) methods are inherently \textit{on-policy} RL, failing to reuse historical data and thus preventing efficient scaling. In this…

Cited by 0SourceScholar
2026

TACTIC: Task-Aware Sparse Coordination Graphs for Multi-Task Multi-agent Reinforcement Learning

ICML 2026poster

Value factorization eases non-stationarity in MARL, but its static coordination assumptions hinder generalization on long-horizon tasks with shifting dependencies. Prior VQ-VAE methods abstract trajectories yet miss time-varying inter-agent dependencies. We present TACTIC, a CTDE framework with thre…

Cited by 0SourceScholar
2026

The Rank and Gradient Lost in Non-stationarity: Sample Weight Decay for Mitigating Plasticity Loss in Reinforcement Learning

ICLR 2026poster

Deep reinforcement learning (RL) suffers from plasticity loss severely due to the nature of non-stationarity, which impairs the ability to adapt to new data and learn continually. Unfortunately, our understanding of how plasticity loss arises, dissipates, and can be dissolved remains limited to empi…

Cited by 0SourceScholar
2026

UniUGG: Unified 3D Understanding and Generation via Geometric-Semantic Encoding

ICLR 2026poster

Despite the impressive progress on understanding and generating images shown by the recent unified architectures, the integration of 3D tasks remains challenging and largely unexplored. In this paper, we introduce UniUGG, the first unified understanding and generation framework for 3D modalities. Ou…

Cited by 0SourcecodeScholar
2026

ViMo: A Generative Visual GUI World Model for App Agents

ICLR 2026poster

App agents, which autonomously operate mobile Apps through GUIs, have gained significant interest in real-world applications. Yet, they often struggle with long-horizon planning, failing to find the optimal actions for complex tasks with longer steps. To address this, world models are used to predic…

Cited by 0SourceScholar
2026

Why Attention Patterns Exist: A Unifying Temporal Perspective Analysis

ICLR 2026poster

Attention patterns play a crucial role in both training and inference of large language models (LLMs). Prior works have identified individual patterns—such as retrieval heads, sink heads, and diagonal traces—but these observations remain fragmented and lack a unifying explanation. To bridge this gap…

Cited by 0SourcecodeScholar
2025

3D-AffordanceLLM: Harnessing Large Language Models for Open-Vocabulary Affordance Detection in 3D Worlds

ICLR 2025poster

3D Affordance detection is a challenging problem with broad applications on various robotic tasks. Existing methods typically formulate the detection paradigm as a label-based semantic segmentation task. This paradigm relies on predefined labels and lacks the ability to comprehend complex natural l…

Cited by 1SourcePDFScholar
2025

A Graph Enhanced Symbolic Discovery Framework For Efficient Logic Optimization

ICLR 2025poster

The efficiency of Logic Optimization (LO) has become one of the key bottlenecks in chip design. To prompt efficient LO, previous studies propose using a key scoring function to predict and prune a large number of ineffective nodes of the LO heuristics. However, the existing scoring functions struggl…

Cited by 0SourcePDFScholar
2025

Accelerating Large Language Model Reasoning via Speculative Search

ICML 2025poster

Tree-search-based reasoning methods have significantly enhanced the reasoning capability of large language models (LLMs) by facilitating the exploration of multiple intermediate reasoning steps, i.e., thoughts. However, these methods suffer from substantial inference latency, as they have to generat…

Cited by 0SourcePDFScholar
2025

Accurate KV Cache Eviction via Anchor Direction Projection for Efficient LLM Inference

NeurIPS 2025poster

Key-Value (KV) cache eviction---which retains the KV pairs of the most important tokens while discarding less important ones---is a critical technique for optimizing both memory usage and inference latency in large language models (LLMs). However, existing approaches often rely on simple heuristics-…

Cited by 0SourceScholar
2025

Apollo-MILP: An Alternating Prediction-Correction Neural Solving Framework for Mixed-Integer Linear Programming

ICLR 2025poster

Leveraging machine learning (ML) to predict an initial solution for mixed-integer linear programming (MILP) has gained considerable popularity in recent years. These methods predict a solution and fix a subset of variables to reduce the problem dimension. Then, they solve the reduced problem to obta…

Cited by 7SourcePDFScholar
2025

AttentionPredictor: Temporal Patterns Matter for KV Cache Compression

NeurIPS 2025poster

With the development of large language models (LLMs), efficient inference through Key-Value (KV) cache compression has attracted considerable attention, especially for long-context generation. To compress the KV cache, recent methods identify critical KV tokens through static modeling of attention s…

Cited by 0SourcecodeScholar
2025

Benchmarking End-To-End Performance of AI-Based Chip Placement Algorithms

NeurIPS 2025poster

Chip placement is a critical step in the Electronic Design Automation (EDA) workflow, which aims to arrange chip modules on the canvas to optimize the performance, power, and area (PPA) metrics of final designs. Recent advances show great potential of AI-based algorithms in chip placement. However,…

Cited by 0SourceScholar
2025

Boosting Multi-Domain Fine-Tuning of Large Language Models through Evolving Interactions between Samples

ICML 2025poster

The multi-domain fine-tuning of large language models (LLMs) confronts a notorious trade-off among abilities across domains. Existing studies attribute this trade-off to the conflicts between samples rooted in inherent semantics. Recent approaches attempt to mitigate these conflicts through the empi…

Cited by 0SourcePDFScholar
2025

COLA: Towards Efficient Multi-Objective Reinforcement Learning with Conflict Objective Regularization in Latent Space

NeurIPS 2025poster

Many real-world control problems require continual policy adjustments to balance multiple objectives, which requires the acquisition of high-quality policies to cover diverse preferences. Multi-Objective Reinforcement Learning (MORL) provides a general framework to solve such problems. However, curr…

Cited by 0SourcecodeScholar
2025

CORE: Collaborative Optimization with Reinforcement Learning and Evolutionary Algorithm for Floorplanning

NeurIPS 2025poster

Floorplanning is the initial step in the physical design process of Electronic Design Automation (EDA), directly influencing subsequent placement, routing, and final power of the chip. However, the solution space in floorplanning is vast, and current algorithms often struggle to explore it sufficien…

Cited by 0SourcecodeScholar
2025

Computing Circuits Optimization via Model-Based Circuit Genetic Evolution

ICLR 2025poster

Optimizing computing circuits such as multipliers and adders is a fundamental challenge in modern integrated circuit design. Recent efforts propose formulating this optimization problem as a reinforcement learning (RL) proxy task, offering a promising approach to search high-speed and area-efficient…

Cited by 4SourcePDFScholar
2025

Conditioning Matters: Training Diffusion Policies is Faster Than You Think

NeurIPS 2025poster

Diffusion policies have emerged as a mainstream paradigm for building vision-language-action (VLA) models. Although they demonstrate strong robot control capabilities, their training efficiency remains suboptimal. In this work, we identify a fundamental challenge in conditional diffusion policy trai…

Cited by 0SourceScholar
2025

DistRL: An Asynchronous Distributed Reinforcement Learning Framework for On-Device Control Agent

ICLR 2025poster

On-device control agents, especially on mobile devices, are responsible for operating mobile devices to fulfill users' requests, enabling seamless and intuitive interactions. Integrating Multimodal Large Language Models (MLLMs) into these agents enhances their ability to understand and execute compl…

Cited by 13SourcePDFScholar
2025

DualRAG: A Dual-Process Approach to Integrate Reasoning and Retrieval for Multi-Hop Question Answering

ACL 2025long

Multi-Hop Question Answering (MHQA) tasks permeate real-world applications, posing challenges in orchestrating multi-step reasoning across diverse knowledge domains. While existing approaches have been improved with iterative retrieval, they still struggle to identify and organize dynamic knowledge.…

Cited by 0SourcePDFScholar
2025

Dynamic Configuration for Cutting Plane Separators via Reinforcement Learning on Incremental Graph

NeurIPS 2025poster

Cutting planes (cuts) are essential for solving mixed-integer linear programming (MILP) problems, as they tighten the feasible solution space and accelerate the solving process. Modern MILP solvers offer diverse cutting plane separators to generate cuts, enabling users to leverage their potential co…

Cited by 0SourceScholar
2025

ET-Plan-Bench: Embodied Task-level Planning Benchmark Towards Spatial-Temporal Cognition with Foundation Models

IROS 2025

Recent advancements in Large Language Models (LLMs) have catalyzed numerous efforts to apply these technologies to embodied tasks, with a particular focus on high-level task planning and task decomposition. LLMs face challenges in understanding the physical world, especially regarding spatial, tempo

Cited by 11SourcecodeScholar
2025

EchoTraffic: Enhancing Traffic Anomaly Understanding with Audio-Visual Insights

CVPR 2025poster

Traffic Anomaly Understanding (TAU) is essential for improving public safety and transportation efficiency by enabling timely detection and response to incidents. Beyond existing methods, which rely largely on visual data, we propose to consider audio cues, a valuable source that offers strong hints…

2025

High-Performance Arithmetic Circuit Optimization via Differentiable Architecture Search

NeurIPS 2025spotlight

Arithmetic circuit optimization remains a fundamental challenge in modern integrated circuit design. Recent advances have cast this problem within the Learning to Optimize (L2O) paradigm, where intelligent agents autonomously explore high-performance design spaces with encouraging results. However,…

Cited by 0SourceScholar
2025

HyperTree Planning: Enhancing LLM Reasoning via Hierarchical Thinking

ICML 2025poster

Recent advancements have significantly enhanced the performance of large language models (LLMs) in tackling complex reasoning tasks, achieving notable success in domains like mathematical and logical reasoning. However, these methods encounter challenges with complex planning tasks, primarily due to…

Cited by 0SourcePDFScholar
2025

Improving Generalization in Offline Reinforcement Learning via Latent Distribution Representation Learning

AAAI 2025technical

Dealing with the distribution shift is a significant challenge when building offline reinforcement learning (RL) models that can generalize from a static dataset to out-of-distribution (OOD) scenarios. Previous approaches have employed pessimism or conservatism strategies. More recently, data-driven…

Cited by 0SourcePDFScholar
2025

LaMPlace: Learning to Optimize Cross-Stage Metrics in Macro Placement

ICLR 2025oral

Machine learning techniques have shown great potential in enhancing macro placement, a critical stage in modern chip design. However, existing methods primarily focus on *online* optimization of *intermediate surrogate metrics* that are available at the current placement stage, rather than directly…

Cited by 2SourcePDFScholar
2025

LaRes: Evolutionary Reinforcement Learning with LLM-based Adaptive Reward Search

NeurIPS 2025poster

The integration of evolutionary algorithms (EAs) with reinforcement learning (RL) has shown superior performance compared to standalone methods. However, previous research focuses on exploration in policy parameter space, while overlooking the reward function search. To bridge this gap, we propose *…

Cited by 0SourcecodeScholar
2025

Lightweight Neural App Control

ICLR 2025spotlight

This paper introduces a novel mobile phone control architecture, Lightweight Multi-modal App Control (LiMAC), for efficient interactions and control across various Android apps. LiMAC takes as input a textual goal and a sequence of past mobile observations, such as screenshots and corresponding UI…

Cited by 3SourcePDFScholar
2025

LogicTree: Improving Complex Reasoning of LLMs via Instantiated Multi-step Synthetic Logical Data

NeurIPS 2025spotlight

Despite their remarkable performance on various tasks, Large Language Models (LLMs) still struggle with logical reasoning, particularly in complex and multi-step reasoning processes. Among various efforts to enhance LLMs' reasoning capabilities, synthesizing large-scale, high-quality logical reason…

Cited by 0SourceScholar
2025

MODULI: Unlocking Preference Generalization via Diffusion Models for Offline Multi-Objective Reinforcement Learning

ICML 2025poster

Multi-objective Reinforcement Learning (MORL) seeks to develop policies that simultaneously optimize multiple conflicting objectives, but it requires extensive online interactions. Offline MORL provides a promising solution by training on pre-collected datasets to generalize to any preference upon d…

Cited by 0SourcePDFScholar
2025

OptiTree: Hierarchical Thoughts Generation with Tree Search for LLM Optimization Modeling

NeurIPS 2025poster

Optimization modeling is one of the most crucial but technical parts of operations research (OR). To automate the modeling process, existing works have leveraged large language models (LLMs), prompting them to break down tasks into steps for generating variables, constraints, and objectives. How…

Cited by 0SourceScholar
2025

R*: Efficient Reward Design via Reward Structure Evolution and Parameter Alignment Optimization with Large Language Models

ICML 2025poster

Reward functions are crucial for policy learning. Large Language Models (LLMs), with strong coding capabilities and valuable domain knowledge, provide an automated solution for high-quality reward design. However, code-based reward functions require precise guiding logic and parameter configuratio…

Cited by 0SourcePDFScholar
2025

Reinforced In-Context Black-Box Optimization

IJCAI 2025

Black-Box Optimization (BBO) has found successful applications in many fields of science and engineering. Recently, there has been a growing interest in meta-learning particular components of BBO algorithms to speed up optimization and get rid of tedious hand-crafted heuristics. As an extension, lea

2025

RoboAnnotatorX: A Comprehensive and Universal Annotation Framework for Accurate Understanding of Long-horizon Robot Demonstration

ICCV 2025poster

Recent advances in robotics have produced numerous valuable large-scale demonstration datasets, yet their potential remains underutilized due to annotation limitations. Current datasets often suffer from sparse temporal annotations, and inconsistent labeling granularity, particularly for complex lon…

Cited by 0SourcePDFScholar
2025

SPA-BENCH: A COMPREHENSIVE BENCHMARK FOR SMARTPHONE AGENT EVALUATION

ICLR 2025spotlight

Smartphone agents are increasingly important for helping users control devices efficiently, with (Multimodal) Large Language Model (MLLM)-based approaches emerging as key contenders. Fairly comparing these agents is essential but challenging, requiring a varied task scope, the integration of agents…

2025

STAR: Learning Diverse Robot Skill Abstractions through Rotation-Augmented Vector Quantization

ICML 2025spotlight

Transforming complex actions into discrete skill abstractions has demonstrated strong potential for robotic manipulation.Existing approaches mainly leverage latent variable models, e.g., VQ-VAE, to learn skill abstractions through learned vectors (codebooks), while they suffer from codebook collapse…

2025

SWAMamba: A Sliding Window Attention Mamba Framework for Predicting Translation Elongation Rates

AAAI 2025technical

Translation elongation is essential for cellular proteostasis and is implicated in cancer and neurodegeneration. Accurately predicting the rate of ribosome elongation in each codon (also called ribosomal A site) on mRNA is important for understanding and modulating protein synthesis. However, predic…

2025

Spatial-Temporal Graph Diffusion Policy with Kinematic Modeling for Bimanual Robotic Manipulation

CVPR 2025poster

Despite the significant success of imitation learning in robotic manipulation, its application to bimanual tasks remains highly challenging. Existing approaches mainly learn a policy to predict a distant next-best end-effector pose (NBP) and then compute the corresponding joint rotation angles for m…

Cited by 2SourcePDFScholar
2025

Succeed or Learn Slowly: Sample Efficient Off-Policy Reinforcement Learning for Mobile App Control

NeurIPS 2025poster

Reinforcement learning (RL) using foundation models for policy approximations in multi-turn tasks remains challenging. We identify two main limitations related to sparse reward settings and policy gradient updates, based on which we formulate a key insight: updates from positive samples with high re…

Cited by 0SourceScholar
2025

The Graph’s Apprentice: Teaching an LLM Low-Level Knowledge for Circuit Quality Estimation

IJCAI 2025

Logic synthesis is a crucial phase in the circuit design process, responsible for transforming hardware description language (HDL) designs into optimized netlists. However, traditional logic synthesis methods are computationally intensive, restricting their iterative use in refining chip designs. Re

2025

Trajectory World Models for Heterogeneous Environments

ICML 2025poster

Heterogeneity in sensors and actuators across environments poses a significant challenge to building large-scale pre-trained world models on top of this low-dimensional sensor information. In this work, we explore pre-training world models for heterogeneous environments by addressing key transfer ba…

2025

Uncertainty-quantified Rollout Policy Adaptation for Unlabelled Cross-domain Video Temporal Grounding

NeurIPS 2025poster

Video Temporal Grounding (TG) aims to temporally locate video segments matching a natural language description (a query) in a long video. While Vision-Language Models (VLMs) are effective at holistic semantic matching, they often struggle with fine-grained temporal localisation. Recently, Group Rela…

Cited by 0SourceScholar
2025

War of Thoughts: Competition Stimulates Stronger Reasoning in Large Language Models

ACL 2025finding

Recent advances in Large Language Models (LLMs) have reshaped the landscape of reasoning tasks, particularly through test-time scaling (TTS) to enhance LLM reasoning. Prior research has used structures such as trees or graphs to guide LLMs in searching for optimal solutions. These methods are time-c…

2024

A Circuit Domain Generalization Framework for Efficient Logic Synthesis in Chip Design

ICML 2024spotlight

Logic Synthesis (LS) plays a vital role in chip design. A key task in LS is to simplify circuits---modeled by directed acyclic graphs (DAGs)---with functionality-equivalent transformations. To tackle this task, many LS heuristics apply transformations to subgraphs---rooted at each node on an input D…

2024

A Hierarchical Adaptive Multi-Task Reinforcement Learning Framework for Multiplier Circuit Design

ICML 2024poster

Multiplier design---which aims to explore a large combinatorial design space to simultaneously optimize multiple conflicting objectives---is a fundamental problem in the integrated circuits industry. Although traditional approaches tackle the multi-objective multiplier optimization problem by manual…

Cited by 17SourcePDFScholar
2024

A Transfer Approach Using Graph Neural Networks in Deep Reinforcement Learning

AAAI 2024technical

Transfer learning (TL) has shown great potential to improve Reinforcement Learning (RL) efficiency by leveraging prior knowledge in new tasks. However, much of the existing TL research focuses on transferring knowledge between tasks that share the same state-action spaces. Further, transfer from mu…

2024

AlignDiff: Aligning Diverse Human Preferences via Behavior-Customisable Diffusion Model

ICLR 2024poster

Aligning agent behaviors with diverse human preferences remains a challenging problem in reinforcement learning (RL), owing to the inherent abstractness and mutability of human preferences. To address these issues, we propose AlignDiff, a novel framework that leverages RLHF to quantify human prefere…

2024

CleanDiffuser: An Easy-to-use Modularized Library for Diffusion Models in Decision Making

NeurIPS 2024poster

Leveraging the powerful generative capability of diffusion models (DMs) to build decision-making agents has achieved extensive success. However, there is still a demand for an easy-to-use and modularized open-source library that offers customized and efficient development for DM-based decision-makin…

2024

DiffuserLite: Towards Real-time Diffusion Planning

NeurIPS 2024poster

Diffusion planning has been recognized as an effective decision-making paradigm in various domains. The capability of generating high-quality long-horizon trajectories makes it a promising research direction. However, existing diffusion planning methods suffer from low decision-making frequencies du…

2024

ENOTO: Improving Offline-to-Online Reinforcement Learning with Q-Ensembles

IJCAI 2024poster

Offline reinforcement learning (RL) is a learning paradigm where an agent learns from a fixed dataset of experience. However, learning solely from a static dataset can limit the performance due to the lack of exploration. To overcome it, offline-to-online RL combines offline pre-training with online…

Cited by 6SourcePDFScholar
2024

EWEK-QA : Enhanced Web and Efficient Knowledge Graph Retrieval for Citation-based Question Answering Systems

ACL 2024long

The emerging citation-based QA systems are gaining more attention especially in generative AI search applications. The importance of extracted knowledge provided to these systems is vital from both accuracy (completeness of information) and efficiency (extracting the information in a timely manner).…

2024

EvoRainbow: Combining Improvements in Evolutionary Reinforcement Learning for Policy Search

ICML 2024poster

Both Evolutionary Algorithms (EAs) and Reinforcement Learning (RL) have demonstrated powerful capabilities in policy search with different principles. A promising direction is to combine the respective strengths of both for efficient policy optimization. To this end, many works have proposed various…

2024

FlexPlanner: Flexible 3D Floorplanning via Deep Reinforcement Learning in Hybrid Action Space with Multi-Modality Representation

NeurIPS 2024poster

In the Integrated Circuit (IC) design flow, floorplanning (FP) determines the position and shape of each block. Serving as a prototype for downstream tasks, it is critical and establishes the upper bound of the final PPA (Power, Performance, Area). However, with the emergence of 3D IC with stacked l…

Cited by 0SourcePDFScholar
2024

Generate Subgoal Images before Act: Unlocking the Chain-of-Thought Reasoning in Diffusion Model for Robot Manipulation with Multimodal Prompts

CVPR 2024poster

Robotics agents often struggle to understand and follow the multi-modal prompts in complex manipulation scenes which are challenging to be sufficiently and accurately described by text alone. Moreover for long-horizon manipulation tasks the deviation from general instruction tends to accumulate if l…

Cited by 11SourcePDFScholar
2024

HarmonyDream: Task Harmonization Inside World Models

ICML 2024poster

Model-based reinforcement learning (MBRL) holds the promise of sample-efficient learning by utilizing a world model, which models how the environment works and typically encompasses components for two tasks: observation modeling and reward modeling. In this paper, through a dedicated empirical inves…

2024

Hybrid CtrlFormer: Learning Adaptive Search Space Partition for Hybrid Action Control via Transformer-based Monte Carlo Tree Search

UAI 2024poster

Hybrid action control tasks are common in the real world, which require controlling some discrete and continuous actions simultaneously. To solve these tasks, existing Deep Reinforcement learning (DRL) methods either directly build a separate policy for each type of action or simplify the hybrid act…

2024

Improving Generalization in Offline Reinforcement Learning via Adversarial Data Splitting

ICML 2024poster

Offline Reinforcement Learning (RL) commonly suffers from the out-of-distribution (OOD) overestimation issue due to the distribution shift. Prior work gradually shifts their focus from suppressing OOD overestimation to avoiding overly conservative learning from suboptimal behavior policies to improv…

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

Iteratively Refined Behavior Regularization for Offline Reinforcement Learning

NeurIPS 2024poster

One of the fundamental challenges for offline reinforcement learning (RL) is ensuring robustness to data distribution. Whether the data originates from a near-optimal policy or not, we anticipate that an algorithm should demonstrate its ability to learn an effective control policy that seamlessly al…

Cited by 7SourcePDFScholar
2024

KISA: A Unified Keyframe Identifier and Skill Annotator for Long-Horizon Robotics Demonstrations

ICML 2024poster

Robotic manipulation tasks often span over long horizons and encapsulate multiple subtasks with different skills. Learning policies directly from long-horizon demonstrations is challenging without intermediate keyframes guidance and corresponding skill annotations. Existing approaches for keyframe i…

Cited by 3SourcePDFScholar
2024

Multiagent Gumbel MuZero: Efficient Planning in Combinatorial Action Spaces

AAAI 2024technical

AlphaZero and MuZero have achieved state-of-the-art (SOTA) performance in a wide range of domains, including board games and robotics, with discrete and continuous action spaces. However, to obtain an improved policy, they often require an excessively large number of simulations, especially for doma…

2024

OVD-Explorer: Optimism Should Not Be the Sole Pursuit of Exploration in Noisy Environments

AAAI 2024technical

In reinforcement learning, the optimism in the face of uncertainty (OFU) is a mainstream principle for directing exploration towards less explored areas, characterized by higher uncertainty. However, in the presence of environmental stochasticity (noise), purely optimistic exploration may lead to ex…

2024

PERIA: Perceive, Reason, Imagine, Act via Holistic Language and Vision Planning for Manipulation

NeurIPS 2024poster

Long-horizon manipulation tasks with general instructions often implicitly encapsulate multiple sub-tasks, posing significant challenges in instruction following. While language planning is a common approach to decompose general instructions into stepwise sub-instructions, text-only guidance may lac…

Cited by 2SourcePDFScholar
2024

PORTAL: Automatic Curricula Generation for Multiagent Reinforcement Learning

AAAI 2024technical

Despite many breakthroughs in recent years, it is still hard for MultiAgent Reinforcement Learning (MARL) algorithms to directly solve complex tasks in MultiAgent Systems (MASs) from scratch. In this work, we study how to use Automatic Curriculum Learning (ACL) to reduce the number of environmental…

Cited by 8SourcePDFScholar
2024

PreRoutGNN for Timing Prediction with Order Preserving Partition: Global Circuit Pre-training, Local Delay Learning and Attentional Cell Modeling

AAAI 2024technical

Pre-routing timing prediction has been recently studied for evaluating the quality of a candidate cell placement in chip design. It involves directly estimating the timing metrics for both pin-level (slack, slew) and edge-level (net delay, cell delay), without time-consuming routing. However, it oft…

2024

Reinforcement Learning within Tree Search for Fast Macro Placement

ICML 2024poster

Macro placement is a crucial step in modern chip design, and reinforcement learning (RL) has recently emerged as a promising technique for improving the placement quality. However, existing RL-based techniques are hindered by their low sample efficiency, requiring numerous online rollouts or substan…

Cited by 17SourcePDFScholar
2024

Rethinking Branching on Exact Combinatorial Optimization Solver: The First Deep Symbolic Discovery Framework

ICLR 2024poster

Machine learning (ML) has been shown to successfully accelerate solving NP-hard combinatorial optimization (CO) problems under the branch and bound framework. However, the high training and inference cost and limited interpretability of ML approaches severely limit their wide application to modern…

Cited by 11SourcePDFScholar
2024

Rethinking Decision Transformer via Hierarchical Reinforcement Learning

ICML 2024poster

Decision Transformer (DT) is an innovative algorithm leveraging recent advances of the transformer architecture in reinforcement learning (RL). However, a notable limitation of DT is its reliance on recalling trajectories from datasets, losing the capability to seamlessly stitch sub-optimal trajecto…

Cited by 16SourcePDFScholar
2024

Safe Table Tennis Swing Stroke with Low-Cost Hardware

ICRA 2024poster

Playing table tennis with a human player is a challenging robotic task due to its dynamic nature. Despite a number of researches being devoted to developing robotic table tennis systems, most of the works have demanding hardware requirements and ignore safety measures when generating the swing stoke…

Cited by 0SourceScholar
2024

Sample-Efficient Multiagent Reinforcement Learning with Reset Replay

ICML 2024poster

The popularity of multiagent reinforcement learning (MARL) is growing rapidly with the demand for real-world tasks that require swarm intelligence. However, a noticeable drawback of MARL is its low sample efficiency, which leads to a huge amount of interactions with the environment. Surprisingly, fe…

Cited by 0SourcePDFScholar
2024

Sample-Efficient Quality-Diversity by Cooperative Coevolution

ICLR 2024spotlight

Quality-Diversity (QD) algorithms, as a subset of evolutionary algorithms, have emerged as a powerful optimization paradigm with the aim of generating a set of high-quality and diverse solutions. Although QD has demonstrated competitive performance in reinforcement learning, its low sample efficienc…

Cited by 11SourcePDFScholar
2024

The Ladder in Chaos: Improving Policy Learning by Harnessing the Parameter Evolving Path in A Low-dimensional Space

NeurIPS 2024poster

Knowing the learning dynamics of policy is significant to unveiling the mysteries of Reinforcement Learning (RL). It is especially crucial yet challenging to Deep RL, from which the remedies to notorious issues like sample inefficiency and learning instability could be obtained. In this paper, we st…

Cited by 1SourcePDFScholar
2024

Towards General Algorithm Discovery for Combinatorial Optimization: Learning Symbolic Branching Policy from Bipartite Graph

ICML 2024poster

Machine learning (ML) approaches have been successfully applied to accelerating exact combinatorial optimization (CO) solvers. However, many of them fail to explain what patterns they have learned that accelerate the CO algorithms due to the black-box nature of ML models like neural networks, and th…

Cited by 6SourcePDFScholar
2024

Towards Next-Generation Logic Synthesis: A Scalable Neural Circuit Generation Framework

NeurIPS 2024poster

Logic Synthesis (LS) aims to generate an optimized logic circuit satisfying a given functionality, which generally consists of circuit translation and optimization. It is a challenging and fundamental combinatorial optimization problem in integrated circuit design. Traditional LS approaches rely on…

Cited by 5SourcePDFScholar
2024

Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedback

ICLR 2024poster

Reinforcement Learning with Human Feedback (RLHF) has received significant attention for performing tasks without the need for costly manual reward design by aligning human preferences. It is crucial to consider diverse human feedback types and various learning methods in different environments. How…

2024

Unlock the Cognitive Generalization of Deep Reinforcement Learning via Granular Ball Representation

ICML 2024poster

The policies learned by humans in simple scenarios can be deployed in complex scenarios with the same task logic through limited feature alignment training, a process referred to as cognitive generalization or systematic generalization. Thus, a plausible conjecture is that unlocking cognitive genera…

Cited by 6SourcePDFScholar
2024

Unlock the Intermittent Control Ability of Model Free Reinforcement Learning

NeurIPS 2024poster

Intermittent control problems are common in real world. The interactions between the decision maker and the executor can be discontinuous (intermittent) due to various types of interruptions, e.g. unstable communication channel. Due to intermittent interaction, agents are unable to acquire the state…

Cited by 0SourcePDFScholar
2024

Value-Evolutionary-Based Reinforcement Learning

ICML 2024poster

Combining Evolutionary Algorithms (EAs) and Reinforcement Learning (RL) for policy search has been proven to improve RL performance. However, previous works largely overlook value-based RL in favor of merging EAs with policy-based RL. This paper introduces Value-Evolutionary-Based Reinforcement Lear…

2024

iVideoGPT: Interactive VideoGPTs are Scalable World Models

NeurIPS 2024poster

World models empower model-based agents to interactively explore, reason, and plan within imagined environments for real-world decision-making. However, the high demand for interactivity poses challenges in harnessing recent advancements in video generative models for developing world models at scal…

2024

vMFER: Von Mises-Fisher Experience Resampling Based on Uncertainty of Gradient Directions for Policy Improvement

IJCAI 2024poster

Reinforcement Learning (RL) is a widely employed technique in decision-making problems, encompassing two fundamental operations -- policy evaluation and policy improvement. Enhancing learning efficiency remains a key challenge in RL, with many efforts focused on using ensemble critics to boost polic…

Cited by 1SourcePDFScholar
2023

Boosting Multiagent Reinforcement Learning via Permutation Invariant and Permutation Equivariant Networks

ICLR 2023poster

The state space in Multiagent Reinforcement Learning (MARL) grows exponentially with the agent number. Such a curse of dimensionality results in poor scalability and low sample efficiency, inhibiting MARL for decades. To break this curse, we propose a unified agent permutation framework that exploit…

Cited by 30SourcePDFScholar
2023

ChiPFormer: Transferable Chip Placement via Offline Decision Transformer

ICML 2023poster

Placement is a critical step in modern chip design, aiming to determine the positions of circuit modules on the chip canvas. Recent works have shown that reinforcement learning (RL) can improve human performance in chip placement. However, such an RL-based approach suffers from long training time an…

Cited by 49SourcePDFScholar
2023

Co-Speech Gesture Synthesis by Reinforcement Learning With Contrastive Pre-Trained Rewards

CVPR 2023poster

There is a growing demand of automatically synthesizing co-speech gestures for virtual characters. However, it remains a challenge due to the complex relationship between input speeches and target gestures. Most existing works focus on predicting the next gesture that fits the data best, however, su…

2023

DAG Matters! GFlowNets Enhanced Explainer for Graph Neural Networks

ICLR 2023poster

Uncovering rationales behind predictions of graph neural networks (GNNs) has received increasing attention over the years. Existing literature mainly focus on selecting a subgraph, through combinatorial optimization, to provide faithful explanations. However, the exponential size of candidate subgra…

2023

ERL-Re$^2$: Efficient Evolutionary Reinforcement Learning with Shared State Representation and Individual Policy Representation

ICLR 2023poster

Deep Reinforcement Learning (Deep RL) and Evolutionary Algorithm (EA) are two major paradigms of policy optimization with distinct learning principles, i.e., gradient-based v.s. gradient-free. An appealing research direction is integrating Deep RL and EA to devise new methods by fusing their complem…

2023

EUCLID: Towards Efficient Unsupervised Reinforcement Learning with Multi-choice Dynamics Model

ICLR 2023poster

Unsupervised reinforcement learning (URL) poses a promising paradigm to learn useful behaviors in a task-agnostic environment without the guidance of extrinsic rewards to facilitate the fast adaptation of various downstream tasks. Previous works focused on the pre-training in a model-free manner whi…

Cited by 14SourcePDFScholar
2023

Generative Flow Networks for Precise Reward-Oriented Active Learning on Graphs

IJCAI 2023poster

Many score-based active learning methods have been successfully applied to graph-structured data, aiming to reduce the number of labels and achieve better performance of graph neural networks based on predefined score functions. However, these algorithms struggle to learn policy distributions that a…

Cited by 3SourcePDFScholar
2023

Learnable Behavior Control: Breaking Atari Human World Records via Sample-Efficient Behavior Selection

ICLR 2023top-5%

The exploration problem is one of the main challenges in deep reinforcement learning (RL). Recent promising works tried to handle the problem with population-based methods, which collect samples with diverse behaviors derived from a population of different exploratory policies. Adaptive policy selec…

Cited by 17SourcePDFScholar
2023

MetaDiffuser: Diffusion Model as Conditional Planner for Offline Meta-RL

ICML 2023poster

Recently, diffusion model shines as a promising backbone for the sequence modeling paradigm in offline reinforcement learning(RL). However, these works mostly lack the generalization ability across tasks with reward or dynamics change. To tackle this challenge, in this paper we propose a task-orient…

Cited by 58SourcePDFScholar
2023

Models as Agents: Optimizing Multi-Step Predictions of Interactive Local Models in Model-Based Multi-Agent Reinforcement Learning

AAAI 2023technical

Research in model-based reinforcement learning has made significant progress in recent years. Compared to single-agent settings, the exponential dimension growth of the joint state-action space in multi-agent systems dramatically increases the complexity of the environment dynamics, which makes it i…

2023

Neighbor Auto-Grouping Graph Neural Networks for Handover Parameter Configuration in Cellular Network

AAAI 2023technical

The mobile communication enabled by cellular networks is the one of the main foundations of our modern society. Optimizing the performance of cellular networks and providing massive connectivity with improved coverage and user experience has a considerable social and economic impact on our daily lif…

Cited by 0SourcePDFScholar
2023

Out-of-distribution Detection with Implicit Outlier Transformation

ICLR 2023poster

Outlier exposure (OE) is powerful in out-of-distribution (OOD) detection, enhancing detection capability via model fine-tuning with surrogate OOD data. However, surrogate data typically deviate from test OOD data. Thus, the performance of OE when facing unseen OOD data, can be weaken. To address thi…

2023

RACE: Improve Multi-Agent Reinforcement Learning with Representation Asymmetry and Collaborative Evolution

ICML 2023poster

Multi-Agent Reinforcement Learning (MARL) has demonstrated its effectiveness in learning collaboration, but it often struggles with low-quality reward signals and high non-stationarity. In contrast, Evolutionary Algorithm (EA) has shown better convergence, robustness, and signal quality insensitivit…

2023

Spectral Augmentations for Graph Contrastive Learning

AISTATS 2023poster

Contrastive learning has emerged as a premier method for learning representations with or without supervision. Recent studies have shown its utility in graph representation learning for pre-training. Despite successes, the understanding of how to design effective graph augmentations that can capture…

2023

SplitNet: A Reinforcement Learning Based Sequence Splitting Method for the MinMax Multiple Travelling Salesman Problem

AAAI 2023technical

MinMax Multiple Travelling Salesman Problem (mTSP) is an important class of combinatorial optimization problems with many practical applications, of which the goal is to minimize the longest tour of all vehicles. Due to its high computational complexity, existing methods for solving this problem can…

Cited by 11SourcePDFScholar
2023

Structure Aware Incremental Learning with Personalized Imitation Weights for Recommender Systems

AAAI 2023technical

Recommender systems now consume large-scale data and play a significant role in improving user experience. Graph Neural Networks (GNNs) have emerged as one of the most effective recommender system models because they model the rich relational information. The ever-growing volume of data can make tra…

Cited by 10SourcePDFScholar
2023

Traj-MAE: Masked Autoencoders for Trajectory Prediction

ICCV 2023poster

Trajectory prediction has been a crucial task in building a reliable autonomous driving system by anticipating possible dangers. One key issue is to generate consistent trajectory predictions without colliding. To overcome the challenge, we propose an efficient masked autoencoder for trajectory pred…

Cited by 57PDFScholar
2022

Cross-domain adaptive transfer reinforcement learning based on state-action correspondence

UAI 2022poster

Despite the impressive success achieved in various domains, deep reinforcement learning (DRL) is still faced with the sample inefficiency problem. Transfer learning (TL), which leverages prior knowledge from different but related tasks to accelerate the target task learning, has emerged as a promis…

2022

DOMINO: Decomposed Mutual Information Optimization for Generalized Context in Meta-Reinforcement Learning

NeurIPS 2022accept

Adapting to the changes in transition dynamics is essential in robotic applications. By learning a conditional policy with a compact context, context-aware meta-reinforcement learning provides a flexible way to adjust behavior according to dynamics changes. However, in real-world applications, the a…

Cited by 13SourcePDFScholar
2022

GALOIS: Boosting Deep Reinforcement Learning via Generalizable Logic Synthesis

NeurIPS 2022accept

Despite achieving superior performance in human-level control problems, unlike humans, deep reinforcement learning (DRL) lacks high-order intelligence (e.g., logic deduction and reuse), thus it behaves ineffectively than humans regarding learning and generalization in complex problems. Previous work…

Cited by 30SourcePDFScholar
2022

HyAR: Addressing Discrete-Continuous Action Reinforcement Learning via Hybrid Action Representation

ICLR 2022poster

Discrete-continuous hybrid action space is a natural setting in many practical problems, such as robot control and game AI. However, most previous Reinforcement Learning (RL) works only demonstrate the success in controlling with either discrete or continuous action space, while seldom take into acc…

Cited by 69SourcePDFScholar
2022

Individual Reward Assisted Multi-Agent Reinforcement Learning

ICML 2022spotlight

In many real-world multi-agent systems, the sparsity of team rewards often makes it difficult for an algorithm to successfully learn a cooperative team policy. At present, the common way for solving this problem is to design some dense individual rewards for the agents to guide the cooperation. Howe…

2022

Learning Pseudometric-based Action Representations for Offline Reinforcement Learning

ICML 2022spotlight

Offline reinforcement learning is a promising approach for practical applications since it does not require interactions with real-world environments. However, existing offline RL methods only work well in environments with continuous or small discrete action spaces. In environments with large and d…

Cited by 20SourcePDFScholar
2022

Learning State Representations via Retracing in Reinforcement Learning

ICLR 2022poster

We propose learning via retracing, a novel self-supervised approach for learning the state representation (and the associated dynamics model) for reinforcement learning tasks. In addition to the predictive (reconstruction) supervision in the forward direction, we propose to include "retraced" transi…

2022

Multiagent Q-learning with Sub-Team Coordination

NeurIPS 2022accept

In many real-world cooperative multiagent reinforcement learning (MARL) tasks, teams of agents can rehearse together before deployment, but then communication constraints may force individual agents to execute independently when deployed. Centralized training and decentralized execution (CTDE) is in…

Cited by 10SourcePDFScholar
2022

Neuro-Symbolic Hierarchical Rule Induction

ICML 2022spotlight

We propose Neuro-Symbolic Hierarchical Rule Induction, an efficient interpretable neuro-symbolic model, to solve Inductive Logic Programming (ILP) problems. In this model, which is built from a pre-defined set of meta-rules organized in a hierarchical structure, first-order rules are invented by lea…

2022

PAnDR: Fast Adaptation to New Environments from Offline Experiences via Decoupling Policy and Environment Representations

IJCAI 2022poster

Deep Reinforcement Learning (DRL) has been a promising solution to many complex decision-making problems. Nevertheless, the notorious weakness in generalization among environments prevent widespread application of DRL agents in real-world scenarios. Although advances have been made recently, most pr…

Cited by 8SourcePDFScholar
2022

PMIC: Improving Multi-Agent Reinforcement Learning with Progressive Mutual Information Collaboration

ICML 2022spotlight

Learning to collaborate is critical in Multi-Agent Reinforcement Learning (MARL). Previous works promote collaboration by maximizing the correlation of agents’ behaviors, which is typically characterized by Mutual Information (MI) in different forms. However, we reveal sub-optimal collaborative beha…

2022

Plan To Predict: Learning an Uncertainty-Foreseeing Model For Model-Based Reinforcement Learning

NeurIPS 2022accept

In Model-based Reinforcement Learning (MBRL), model learning is critical since an inaccurate model can bias policy learning via generating misleading samples. However, learning an accurate model can be difficult since the policy is continually updated and the induced distribution over visited states…

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

Socially-Attentive Policy Optimization in Multi-Agent Self-Driving System

CoRL 2022poster

As increasing numbers of autonomous vehicles (AVs) are being deployed, it is important to construct a multi-agent self-driving (MASD) system for navigating traffic flows of AVs. In an MASD system, AVs not only navigate themselves to pursue their own goals, but also interact with each other to preven…

Cited by 12SourceScholar
2022

The Policy-gradient Placement and Generative Routing Neural Networks for Chip Design

NeurIPS 2022accept

Placement and routing are two critical yet time-consuming steps of chip design in modern VLSI systems. Distinct from traditional heuristic solvers, this paper on one hand proposes an RL-based model for mixed-size macro placement, which differs from existing learning-based placers that often consider…

Cited by 42SourcePDFScholar
2022

Transformer-based Working Memory for Multiagent Reinforcement Learning with Action Parsing

NeurIPS 2022accept

Learning in real-world multiagent tasks is challenging due to the usual partial observability of each agent. Previous efforts alleviate the partial observability by historical hidden states with Recurrent Neural Networks, however, they do not consider the multiagent characters that either the multia…

Cited by 20SourcePDFScholar
2022

Versatile Multi-stage Graph Neural Network for Circuit Representation

NeurIPS 2022accept

Due to the rapid growth in the scale of circuits and the desire for knowledge transfer from old designs to new ones, deep learning technologies have been widely exploited in Electronic Design Automation (EDA) to assist circuit design. In chip design cycles, we might encounter heterogeneous and diver…

Cited by 39SourcePDFScholar
2022

What about Inputting Policy in Value Function: Policy Representation and Policy-Extended Value Function Approximator

AAAI 2022technical

We study Policy-extended Value Function Approximator (PeVFA) in Reinforcement Learning (RL), which extends conventional value function approximator (VFA) to take as input not only the state (and action) but also an explicit policy representation. Such an extension enables PeVFA to preserve values of…

Cited by 26SourcePDFScholar
2021

A Hierarchical Reinforcement Learning Based Optimization Framework for Large-scale Dynamic Pickup and Delivery Problems

NeurIPS 2021poster

The Dynamic Pickup and Delivery Problem (DPDP) is an essential problem in the logistics domain, which is NP-hard. The objective is to dynamically schedule vehicles among multiple sites to serve the online generated orders such that the overall transportation cost could be minimized. The critical cha…

Cited by 87SourcePDFScholar
2021

Adaptive Online Packing-guided Search for POMDPs

NeurIPS 2021poster

The partially observable Markov decision process (POMDP) provides a general framework for modeling an agent's decision process with state uncertainty, and online planning plays a pivotal role in solving it. A belief is a distribution of states representing state uncertainty. Methods for large-scale…

2021

Addressing Action Oscillations through Learning Policy Inertia

AAAI 2021technical

Deep reinforcement learning (DRL) algorithms have been demonstrated to be effective on a wide range of challenging decision making and control tasks. However, these methods typically suffer from severe action oscillations in particular in discrete action setting, which means that agents select diffe…

2021

An Efficient Transfer Learning Framework for Multiagent Reinforcement Learning

NeurIPS 2021poster

Transfer Learning has shown great potential to enhance single-agent Reinforcement Learning (RL) efficiency. Similarly, Multiagent RL (MARL) can also be accelerated if agents can share knowledge with each other. However, it remains a problem of how an agent should learn from other agents. In this pap…

2021

CausalVAE: Disentangled Representation Learning via Neural Structural Causal Models

CVPR 2021poster

Learning disentanglement aims at finding a low dimensional representation which consists of multiple explanatory and generative factors of the observational data. The framework of variational autoencoder (VAE) is commonly used to disentangle independent factors from observations. However, in real sc…

Cited by 337PDFScholar
2021

Dynamic Bottleneck for Robust Self-Supervised Exploration

NeurIPS 2021poster

Exploration methods based on pseudo-count of transitions or curiosity of dynamics have achieved promising results in solving reinforcement learning with sparse rewards. However, such methods are usually sensitive to environmental dynamics-irrelevant information, e.g., white-noise. To handle such dyn…

2021

Flattening Sharpness for Dynamic Gradient Projection Memory Benefits Continual Learning

NeurIPS 2021poster

The backpropagation networks are notably susceptible to catastrophic forgetting, where networks tend to forget previously learned skills upon learning new ones. To address such the 'sensitivity-stability' dilemma, most previous efforts have been contributed to minimizing the empirical risk with diff…

2021

Foresee then Evaluate: Decomposing Value Estimation with Latent Future Prediction

AAAI 2021technical

Value function is the central notion of Reinforcement Learning (RL). Value estimation, especially with function approximation, can be challenging since it involves the stochasticity of environmental dynamics and reward signals that can be sparse and delayed in some cases. A typical model-free RL alg…

2021

Model-Based Reinforcement Learning via Imagination with Derived Memory

NeurIPS 2021poster

Model-based reinforcement learning aims to improve the sample efficiency of policy learning by modeling the dynamics of the environment. Recently, the latent dynamics model is further developed to enable fast planning in a compact space. It summarizes the high-dimensional experiences of an agent, wh…

Cited by 9SourcePDFScholar
2021

Ordering-Based Causal Discovery with Reinforcement Learning

IJCAI 2021poster

It is a long-standing question to discover causal relations among a set of variables in many empirical sciences. Recently, Reinforcement Learning (RL) has achieved promising results in causal discovery from observational data. However, searching the space of directed graphs and enforcing acyclic…

2021

Principled Exploration via Optimistic Bootstrapping and Backward Induction

ICML 2021spotlight

One principled approach for provably efficient exploration is incorporating the upper confidence bound (UCB) into the value function as a bonus. However, UCB is specified to deal with linear and tabular settings and is incompatible with Deep Reinforcement Learning (DRL). In this paper, we propose a…

2021

Relational Navigation Learning in Continuous Action Space among Crowds

ICRA 2021poster

In this paper, a novel navigation learning method in continuous action space among crowds based on relational graph is proposed which can be directly deployed on differential-drive mobile robots without any change. More specifically, in order to increase generalization ability in crowd sizes, Graph…

Cited by 9SourceScholar
2021

Towards Effective Context for Meta-Reinforcement Learning: an Approach based on Contrastive Learning

AAAI 2021technical

Context, the embedding of previous collected trajectories, is a powerful construct for Meta-Reinforcement Learning (Meta-RL) algorithms. By conditioning on an effective context, Meta-RL policies can easily generalize to new tasks within a few adaptation steps. We argue that improving the quality of…

2020

Action Semantics Network: Considering the Effects of Actions in Multiagent Systems

ICLR 2020poster

In multiagent systems (MASs), each agent makes individual decisions but all of them contribute globally to the system evolution. Learning in MASs is difficult since each agent's selection of actions must take place in the presence of other co-learning agents. Moreover, the environmental stochasticit…

Cited by 48SourcecodeScholar
2020

Dynamic Knapsack Optimization Towards Efficient Multi-Channel Sequential Advertising

ICML 2020poster

In E-commerce, advertising is essential for merchants to reach their target users. The typical objective is to maximize the advertiser’s cumulative revenue over a period of time under a budget constraint. In real applications, an advertisement (ad) usually needs to be exposed to the same user multip…

Cited by 29SourcePDFScholar
2020

Efficient Deep Reinforcement Learning via Adaptive Policy Transfer

IJCAI 2020poster

Transfer learning has shown great potential to accelerate Reinforcement Learning (RL) by leveraging prior knowledge from past learned policies of relevant tasks. Existing approaches either transfer previous knowledge by explicitly computing similarities between tasks or select appropriate source pol…

2020

Generating Behavior-Diverse Game AIs with Evolutionary Multi-Objective Deep Reinforcement Learning

IJCAI 2020poster

Generating diverse behaviors for game artificial intelligence (Game AI) has been long recognized as a challenging task in the game industry. Designing a Game AI with a satisfying behavioral characteristic (style) heavily depends on the domain knowledge and is hard to achieve manually. Deep reinforce…

Cited by 0SourcePDFScholar
2020

KoGuN: Accelerating Deep Reinforcement Learning via Integrating Human Suboptimal Knowledge

IJCAI 2020poster

Reinforcement learning agents usually learn from scratch, which requires a large number of interactions with the environment. This is quite different from the learning process of human. When faced with a new task, human naturally have the common sense and use the prior knowledge to derive an initial…

Cited by 0SourcePDFScholar
2020

Learning to Accelerate Heuristic Searching for Large-Scale Maximum Weighted b-Matching Problems in Online Advertising

IJCAI 2020poster

Bipartite b-matching is fundamental in algorithm design, and has been widely applied into diverse applications, such as economic markets, labor markets, etc. These practical problems usually exhibit two distinct features: large-scale and dynamic, which requires the matching algorithm to be repeatedl…

Cited by 0SourcePDFScholar
2020

Learning to Utilize Shaping Rewards: A New Approach of Reward Shaping

NeurIPS 2020poster

Reward shaping is an effective technique for incorporating domain knowledge into reinforcement learning (RL). Existing approaches such as potential-based reward shaping normally make full use of a given shaping reward function. However, since the transformation of human knowledge into numeric reward…

Cited by 246SourcePDFScholar
2020

Q-value Path Decomposition for Deep Multiagent Reinforcement Learning

ICML 2020poster

Recently, deep multiagent reinforcement learning (MARL) has become a highly active research area as many real-world problems can be inherently viewed as multiagent systems. A particularly interesting and widely applicable class of problems is the partially observable cooperative multiagent setting,…

Cited by 73SourcePDFScholar
2020

SMARTS: An Open-Source Scalable Multi-Agent RL Training School for Autonomous Driving

CoRL 2020

Interaction is fundamental in autonomous driving (AD). Despite more than a decade of intensive R&D in AD, how to dynamically interact with diverse road users in various contexts still remains unsolved. Multi-agent learning has recently seen big breakthroughs and has much to offer towards solving rea

2020

Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets

IJCAI 2020poster

Generative adversarial imitation learning (GAIL) has shown promising results by taking advantage of generative adversarial nets, especially in the field of robot learning. However, the requirement of isolated single modal demonstrations limits the scalability of the approach to real world scenarios…

Cited by 0SourcePDFScholar
2018

A Deep Bayesian Policy Reuse Approach Against Non-Stationary Agents

NeurIPS 2018poster

In multiagent domains, coping with non-stationary agents that change behaviors from time to time is a challenging problem, where an agent is usually required to be able to quickly detect the other agent's policy during online interaction, and then adapt its own policy accordingly. This paper studies…

Cited by 103SourcePDFScholar