← Search

YANG YU

145 accepted papers

2026

ADM-v2: Pursuing Full-Horizon Roll-out in Dynamics Models for Offline Policy Learning and Evaluation

ICLR 2026poster

Model-based methods for offline Reinforcement Learning transfer extensive policy exploration and evaluation to data-driven dynamics models, effectively saving real-world samples in the offline setting. We expect the dynamics model to allow the policy to roll out full-horizon episodes, which is cruci…

Cited by 0SourcecodeScholar
2026

An LLM-based Quantitative Framework for Evaluating High-Stealthy Backdoor Risks in OSS Supply Chains

AAAI 2026technical

In modern software development workflows, the open-source software supply chain significantly contributes to efficient and convenient engineering practices. With increasing system complexity, it has become a common practice to use open-source software as third-party dependencies. However, due to the

Cited by 0SourcePDFScholar
2026

Clipping Low-Probability Tokens in SFT Yields a Generalizable Initialization for RL

ICML 2026poster

Supervised Fine-Tuning (SFT) is a critical step for adapting Large Language Models (LLMs) to specialized domains, often serving as an initialization for subsequent reinforcement learning (RL). However, SFT can overfit a small set of expert data, harming generalization and eroding prior knowledge. Th…

Cited by 0SourceScholar
2026

Decentralized and Disentangled Task–Role Representation Learning for Generalizable Offline Multi-Agent Meta Reinforcement Learning

ICML 2026poster

Offline meta reinforcement learning (RL) enables agents to learn a unified policy from multi-task offline data to support generalization in out-of-distribution (OOD) tasks. Recent approaches in single-agent RL tackle this by learning an efficient task representation to distinguish between tasks, sho…

Cited by 0SourceScholar
2026

Escaping the Subspace Trap: The Role of Optimizer Geometry in Model Width Expansion

ICML 2026poster

Pre-training large language models from scratch is prohibitively expensive as model scales increase. A practical alternative is Model Width Expansion (MWE), which grows a larger model from a well-pretrained ''seed'' model to inherit existing capabilities at initialization. However, we identify a phe…

Cited by 0SourceScholar
2026

Exploring Reliable Spatiotemporal Dependencies for Efficient Visual Tracking

AAAI 2026technical

Recent advances in transformer-based lightweight object tracking have established new standards across benchmarks, leveraging the global receptive field and powerful feature extraction capabilities of attention mechanisms. Despite these achievements, existing methods universally employ sparse sampli

Cited by 0SourcePDFScholar
2026

HEV Generative Sandbox: A Framework for Assessing Domain-Specific Social Risks Through Human-LLM Simulation

AAAI 2026technical

Deploying Large Language Models (LLMs) in specialized domains introduces significant societal and compliance risks, including bias amplification, misinformation propagation, and privacy violations. These risks predominantly emerge from the dynamic interactions between LLMs and humans in specific con

Cited by 0SourcePDFScholar
2026

Hierarchical Value-Decomposed Offline Reinforcement Learning for Whole-Body Control

ICLR 2026poster

Scaling imitation learning to high-DoF whole-body robots is fundamentally limited by the \textbf{curse of dimensionality} and the prohibitive cost of collecting expert demonstrations. We argue that the core bottleneck is paradigmatic: real-world supervision for whole-body control is inherently imper…

Cited by 0SourceScholar
2026

Learning Disentangled Multi-Agent World Model for Decentralized Control

ICML 2026poster

World models enable learning policies via latent imagination, offering benefits such as history compression and sample efficiency. The primary challenge in applying world models to multi-agent tasks is that modeling multi-agent dynamics in latent space requires integrating information from different…

Cited by 0SourceScholar
2026

Multi-agent In-context Coordination via Decentralized Memory Retrieval

AAAI 2026technical

Large transformer models, trained on diverse datasets, have demonstrated impressive few-shot performance on previously unseen tasks without requiring parameter updates. This capability has also been explored in Reinforcement Learning (RL), where agents interact with the environment to retrieve conte

Cited by 0SourcePDFScholar
2026

Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Role of Bellman Constraints

ICML 2026poster

Adversarial imitation learning (AIL) achieves high-quality imitation by mitigating compounding errors in behavioral cloning (BC), but often exhibits training instability due to adversarial optimization. To avoid this issue, a class of non-adversarial Q-based imitation learning (IL) methods, represen…

Cited by 0SourceScholar
2026

Offline Multi-agent Continual Cooperation via Skill Partition and Reuse

ICML 2026poster

Extracting skills from multi-agent offline dataset improves learning efficiency via sharing task-invariant coordination skills among tasks. In settings where tasks occur sequentially and the space of skills grows exponentially, existing approaches that rely on heuristically designed and fixed-sized …

Cited by 0SourceScholar
2026

PDD-RRG: Posterior Diagnostic Decision for Study-level Radiology Report Generation

IJCAI 2026

Automatic radiology report generation (RRG) aims to simulate the workflow of radiologists, assisting them in clinical diagnosis. However, existing methods often fall short in utilizing all information relevant to the examination, as is typically done in clinical practice. Although some works attempt

Cited by 0Scholar
2026

Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learning

ICML 2026poster

Adversarial imitation learning (AIL) demonstrates superior expert sample efficiency compared to behavioral cloning (BC), yet requires substantial online environment interaction. While recent empirical work has explored initializing AIL algorithms with BC-pretrained policies to address this limitatio…

Cited by 0SourceScholar
2026

ReLAM: Learning Anticipation Model for Rewarding Visual Robotic Manipulation

ICML 2026poster

Reward design remains a critical bottleneck in visual reinforcement learning (RL) for robotic manipulation. In simulated environments, rewards are conventionally designed based on the distance to a target position. However, such precise positional information is often unavailable in real-world visua…

Cited by 0SourceScholar
2026

Speedup Patch: Learning a Plug-and-Play Policy to Accelerate Embodied Manipulation

ICML 2026poster

While current embodied policies exhibit remarkable manipulation skills, their execution remains unsatisfactorily slow as they inherit the tardy pacing of human demonstrations. Existing acceleration methods typically require policy retraining or costly online interactions, limiting their scalability …

Cited by 0SourceScholar
2026

SurgPub-Video: A Comprehensive Surgical Video Framework for Enhanced Surgical Intelligence in Vision-Language Model

AAAI 2026technical

Vision-Language Models (VLMs) have shown significant potential in surgical scene analysis, yet existing models are limited by frame-level datasets and lack high-quality video data with procedural surgical knowledge. To address these challenges, we make the following contributions: (i) SurgPub-Video,

Cited by 0SourcePDFScholar
2026

Towards Automated Kernel Generation in the Era of LLMs: A Survey

IJCAI 2026

The performance of modern AI systems is fundamentally constrained by the quality of their underlying kernels, which translate high-level algorithmic semantics into low-level hardware operations. Achieving near-optimal kernels requires expert-level understanding of hardware architectures and programm

Cited by 0Scholar
2026

Towards Complete Multi-Agent Coordination Policy Learning via Denoising Maximum Entropy Optimization

ICML 2026poster

Parameter sharing is a widely used technique in Multi-Agent Reinforcement Learning (MARL) that enhances sample efficiency by equipping agents with a unified policy. While effective in homogeneous settings, it often struggles in heterogeneous environments where agents possess diverse capabilities. Co…

Cited by 0SourceScholar
2026

Towards Practical World Model-based Reinforcement Learning for Vision-Language-Action Models

ICML 2026poster

Vision-Language-Action (VLA) models show strong generalization for robotic control, but finetuning them with reinforcement learning (RL) is constrained by the high cost and safety risks of real-world interaction. Training VLA models in interactive world models avoids these issues but introduces seve…

Cited by 0SourceScholar
2025

Adaptable Safe Policy Learning from Multi-task Data with Constraint Prioritized Decision Transformer

NeurIPS 2025poster

Learning safe reinforcement learning (RL) policies from offline multi-task datasets without direct environmental interaction is crucial for efficient and reliable deployment of RL agents. Benefiting from their scalability and strong in-context learning capabilities, recent approaches attempt to util…

Cited by 0SourceScholar
2025

Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning

ICLR 2025poster

Model-based methods in reinforcement learning offer a promising approach to enhance data efficiency by facilitating policy exploration within a dynamics model. However, accurately predicting sequential steps in the dynamics model remains a challenge due to the bootstrapping prediction, which attribu…

2025

Behavior-Regularized Diffusion Policy Optimization for Offline Reinforcement Learning

ICML 2025poster

Behavior regularization, which constrains the policy to stay close to some behavior policy, is widely used in offline reinforcement learning (RL) to manage the risk of hazardous exploitation of unseen actions. Nevertheless, existing literature on behavior-regularized RL primarily focuses on explicit…

Cited by 1SourcePDFScholar
2025

Controlling Large Language Model with Latent Action

ICML 2025poster

Adapting Large Language Models (LLMs) to downstream tasks using Reinforcement Learning (RL) has proven to be an effective approach. However, LLMs do not inherently define the structure of an agent for RL training, particularly in terms of specifying the action space. This paper studies learning a co…

Cited by 0SourcePDFScholar
2025

Dual-AEB: Synergizing Rule-Based and Multimodal Large Language Models for Effective Emergency Braking

ICRA 2025

Automatic Emergency Braking (AEB) systems are a crucial component in ensuring the safety of passengers in autonomous vehicles. Conventional AEB systems primarily rely on closed-set perception modules to recognize traffic conditions and assess collision risks. To enhance the adaptability of AEB syste

Cited by 3SourcecodeScholar
2025

Efficient Multi-agent Offline Coordination via Diffusion-based Trajectory Stitching

ICLR 2025poster

Learning from offline data without interacting with the environment is a promising way to fully leverage the intelligent decision-making capabilities of multi-agent reinforcement learning (MARL). Previous approaches have primarily focused on developing learning techniques, such as conservative metho…

Cited by 1SourcePDFScholar
2025

Focus-Then-Reuse: Fast Adaptation in Visual Perturbation Environments

NeurIPS 2025poster

Visual reinforcement learning has shown promise in various real-world applications. However, deploying policies in complex real-world environments with visual perturbations remains a significant challenge. We notice that humans tend to filter information at the object level prior to decision-making,…

Cited by 0SourcecodeScholar
2025

GRAIN: Multi-Granular and Implicit Information Aggregation Graph Neural Network for Heterophilous Graphs

AAAI 2025technical

Graph neural networks (GNNs) have shown significant success in learning graph representations. However, recent studies reveal that GNNs often fail to outperform simple MLPs on heterophilous graph tasks, where connected nodes may differ in features or labels, challenging the homophily assumption. Exi…

Cited by 0SourcePDFScholar
2025

Geometric Mixture Models for Electrolyte Conductivity Prediction

NeurIPS 2025poster

Accurate prediction of ionic conductivity in electrolyte systems is crucial for advancing numerous scientific and technological applications. While significant progress has been made, current research faces two fundamental challenges: (1) the lack of high-quality standardized benchmarks, and (2) ina…

Cited by 0SourceScholar
2025

GuideNER: Annotation Guidelines Are Better than Examples for In-Context Named Entity Recognition

AAAI 2025technical

Large language models (LLMs) demonstrate impressive performance on downstream tasks through in-context learning(ICL). However, there is a significant gap between their performance in Named Entity Recognition (NER) and in fine-tuning methods. We believe this discrepancy is due to inconsistencies in l…

2025

Improving Reward Model Generalization from Adversarial Process Enhanced Preferences

ICML 2025poster

In sequential decision-making, the reward function serves as the primary supervision signal, guiding agents to acquire the desired behaviors. Traditional reward modeling methods rely heavily on human expertise, limiting their scalability. Automated preference generation from suboptimal demonstration…

Cited by 0SourcePDFScholar
2025

LLM Data Selection and Utilization via Dynamic Bi-level Optimization

ICML 2025poster

While large-scale training data is fundamental for developing capable large language models (LLMs), strategically selecting high-quality data has emerged as a critical approach to enhance training efficiency and reduce computational costs. Current data selection methodologies predominantly rely on s…

Cited by 0SourcePDFScholar
2025

LLM-Assisted Semantically Diverse Teammate Generation for Efficient Multi-agent Coordination

ICML 2025poster

Training with diverse teammates is the key for learning generalizable agents. Typical approaches aim to generate diverse teammates by utilizing techniques like randomization, designing regularization terms, or reducing policy compatibility, etc. However, such teammates lack semantic information, res…

2025

LLMOPT: Learning to Define and Solve General Optimization Problems from Scratch

ICLR 2025poster

Optimization problems are prevalent across various scenarios. Formulating and then solving optimization problems described by natural language often requires highly specialized human expertise, which could block the widespread application of optimization-based decision making. To automate problem fo…

2025

Learning View-invariant World Models for Visual Robotic Manipulation

ICLR 2025poster

Robotic manipulation tasks often rely on visual inputs from cameras to perceive the environment. However, previous approaches still suffer from performance degradation when the camera’s viewpoint changes during manipulation. In this paper, we propose ReViWo (Representation learning for View-invarian…

Cited by 0SourcePDFScholar
2025

Learning to Reuse Policies in State Evolvable Environments

ICML 2025poster

The policy trained via reinforcement learning (RL) makes decisions based on sensor-derived state features. It is common for state features to evolve for reasons such as periodic sensor maintenance or the addition of new sensors for performance improvement. The deployed policy fails in new state spac…

Cited by 0SourcePDFScholar
2025

MRE-MI: A Multi-image Dataset for Multimodal Relation Extraction in Social Media Posts

NAACL 2025findings

Despite recent advances in Multimodal Relation Extraction (MRE), existing datasets and approaches primarily focus on single-image scenarios, overlooking the prevalent real-world cases where relationships are expressed through multiple images alongside text. To address this limitation, we present MRE…

2025

MedUnifier: Unifying Vision-and-Language Pre-training on Medical Data with Vision Generation Task using Discrete Visual Representations

CVPR 2025poster

Despite significant progress in Vision-Language Pre-training (VLP), current approaches predominantly emphasize feature extraction and cross-modal comprehension, with limited attention to generating or transforming visual content. This gap hinders the model's ability to synthesize coherent and novel…

2025

Multi-Agent Imitation by Learning and Sampling from Factorized Soft Q-Function

NeurIPS 2025poster

Learning from multi-agent expert demonstrations, known as Multi-Agent Imitation Learning (MAIL), provides a promising approach to sequential decision-making. However, existing MAIL methods including Behavior Cloning (BC) and Adversarial Imitation Learning (AIL) face significant challenges: BC suffer…

Cited by 0SourcecodeScholar
2025

On the Optimization Landscape of Low Rank Adaptation Methods for Large Language Models

ICLR 2025poster

Training Large Language Models (LLMs) poses significant memory challenges, making low-rank adaptation methods an attractive solution. Previously, Low-Rank Adaptation (LoRA) addressed this by adding a trainable low-rank matrix to the frozen pre-trained weights in each layer, reducing the number of tr…

Cited by 0SourcePDFScholar
2025

POINTS-Reader: Distillation-Free Adaptation of Vision-Language Models for Document Conversion

EMNLP 2025

High-quality labeled data is essential for training accurate document conversion models, particularly in domains with complex formats such as tables, formulas, and multi-column text. However, manual annotation is both costly and time-consuming, while automatic labeling using existing models often la

2025

Q-Adapter: Customizing Pre-trained LLMs to New Preferences with Forgetting Mitigation

ICLR 2025poster

Large Language Models (LLMs), trained on a large amount of corpus, have demonstrated remarkable abilities. However, it may not be sufficient to directly apply open-source LLMs like Llama to certain real-world scenarios, since most of them are trained for \emph{general} purposes. Thus, the demands fo…

2025

SOO-Bench: Benchmarks for Evaluating the Stability of Offline Black-Box Optimization

ICLR 2025poster

Black-box optimization aims to find the optima through building a model close to the black-box objective function based on function value evaluation. However, in many real-world tasks, such as the design of molecular formulas and mechanical structures, it is perilous, costly, or even infeasible to e…

2025

Semantic Temporal Abstraction via Vision-Language Model Guidance for Efficient Reinforcement Learning

ICLR 2025poster

Extracting temporally extended skills can significantly improve the efficiency of reinforcement learning (RL) by breaking down complex decision-making problems with sparse rewards into simpler subtasks and enabling more effective credit assignment. However, existing abstraction methods either discov…

Cited by 0SourcePDFScholar
2025

TCM-Ladder: A Benchmark for Multimodal Question Answering on Traditional Chinese Medicine

NeurIPS 2025poster

Traditional Chinese Medicine (TCM), as an effective alternative medicine, has been receiving increasing attention. In recent years, the rapid development of large language models (LLMs) tailored for TCM has highlighted the urgent need for an objective and comprehensive evaluation framework to assess…

Cited by 0SourcecodeScholar
2025

VA-AR: Learning Velocity-Aware Action Representations with Mixture of Window Attention

AAAI 2025technical

Action recognition is a crucial task in artificial intelligence, with significant implications across various domains. We initially perform a comprehensive analysis of seven prominent action recognition methods across five widely-used datasets. This analysis reveals a critical, yet previously overlo…

2024

ACT: Empowering Decision Transformer with Dynamic Programming via Advantage Conditioning

AAAI 2024technical

Decision Transformer (DT), which employs expressive sequence modeling techniques to perform action generation, has emerged as a promising approach to offline policy optimization. However, DT generates actions conditioned on a desired future return, which is known to bear some weaknesses such as the…

2024

ADMN: Agent-Driven Modular Network for Dynamic Parameter Sharing in Cooperative Multi-Agent Reinforcement Learning

IJCAI 2024poster

Parameter sharing is a common strategy in multi-agent reinforcement learning (MARL) to make the training more efficient and scalable. However, applying parameter sharing among agents indiscriminately hinders the emergence of agents diversity and degrades the final cooperative performance. To better…

Cited by 1SourcePDFScholar
2024

ANEDL: Adaptive Negative Evidential Deep Learning for Open-Set Semi-supervised Learning

AAAI 2024technical

Semi-supervised learning (SSL) methods assume that labeled data, unlabeled data and test data are from the same distribution. Open-set semi-supervised learning (Open-set SSL) con- siders a more practical scenario, where unlabeled data and test data contain new categories (outliers) not observed in l…

Cited by 5SourcePDFScholar
2024

Bias and Volatility: A Statistical Framework for Evaluating Large Language Model's Stereotypes and the Associated Generation Inconsistency

NeurIPS 2024poster

We present a novel statistical framework for analyzing stereotypes in large language models (LLMs) by systematically estimating the bias and variation in their generation. Current evaluation metrics in the alignment literature often overlook the randomness of stereotypes caused by the inconsistent g…

Cited by 2SourceScholar
2024

Car-Studio: Learning Car Radiance Fields From Single-View and Unlimited In-the-Wild Images

RA-L 2024

Compositional neural scene graph studies have shown that radiance fields can be an efficient tool in an editable autonomous driving simulator. However, previous studies learned within a sequence of autonomous driving datasets, resulting in unsatisfactory blurring when rotating the car in the simulat

Cited by 6SourceScholar
2024

Causality Based Front-door Defense Against Backdoor Attack on Language Models

ICML 2024poster

We have developed a new framework based on the theory of causal inference to protect language models against backdoor attacks. Backdoor attackers can poison language models with different types of triggers, such as words, sentences, grammar, and style, enabling them to selectively modify the decisio…

2024

Continual Multi-Objective Reinforcement Learning via Reward Model Rehearsal

IJCAI 2024poster

Multi-objective reinforcement learning (MORL) approaches address real-world problems with multiple objectives by learning policies maximizing returns weighted by different user preferences. Typical methods assume the objectives remain unchanged throughout the agent's lifetime. However, in some real-…

Cited by 0SourcePDFScholar
2024

Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics

ICML 2024poster

Developing policies that can adapt to non-stationary environments is essential for real-world reinforcement learning applications. Nevertheless, learning such adaptable policies in offline settings, with only a limited set of pre-collected trajectories, presents significant challenges. A key difficu…

2024

Deep Demonstration Tracing: Learning Generalizable Imitator Policy for Runtime Imitation from a Single Demonstration

ICML 2024poster

One-shot imitation learning (OSIL) is to learn an imitator agent that can execute multiple tasks with only a single demonstration. In real-world scenario, the environment is dynamic, e.g., unexpected changes can occur after demonstration. Thus, achieving generalization of the imitator agent is cruci…

2024

Efficient Recurrent Off-Policy RL Requires a Context-Encoder-Specific Learning Rate

NeurIPS 2024poster

Real-world decision-making tasks are usually partially observable Markov decision processes (POMDPs), where the state is not fully observable. Recent progress has demonstrated that recurrent reinforcement learning (RL), which consists of a context encoder based on recurrent neural networks (RNNs) fo…

2024

Energy-Guided Diffusion Sampling for Offline-to-Online Reinforcement Learning

ICML 2024poster

Combining offline and online reinforcement learning (RL) techniques is indeed crucial for achieving efficient and safe learning where data acquisition is expensive. Existing methods replay offline data directly in the online phase, resulting in a significant challenge of data distribution shift and…

2024

Episodic Return Decomposition by Difference of Implicitly Assigned Sub-trajectory Reward

AAAI 2024technical

Real-world decision-making problems are usually accompanied by delayed rewards, which affects the sample efficiency of Reinforcement Learning, especially in the extremely delayed case where the only feedback is the episodic reward obtained at the end of an episode. Episodic return decomposition is a…

2024

Flow to Better: Offline Preference-based Reinforcement Learning via Preferred Trajectory Generation

ICLR 2024poster

Offline preference-based reinforcement learning (PbRL) offers an effective solution to overcome the challenges associated with designing rewards and the high costs of online interactions. In offline PbRL, agents are provided with a fixed dataset containing human preferences between pairs of trajecto…

Cited by 20SourcePDFScholar
2024

Focus-Then-Decide: Segmentation-Assisted Reinforcement Learning

AAAI 2024technical

Visual Reinforcement Learning (RL) is a promising approach to achieve human-like intelligence. However, it currently faces challenges in learning efficiently within noisy environments. In contrast, humans can quickly identify task-relevant objects in distraction-filled surroundings by applying previ…

2024

Generalizable Task Representation Learning for Offline Meta-Reinforcement Learning with Data Limitations

AAAI 2024technical

Generalization and sample efficiency have been long-standing issues concerning reinforcement learning, and thus the field of Offline Meta-Reinforcement Learning (OMRL) has gained increasing attention due to its potential of solving a wide range of problems with static and limited offline data. Exist…

2024

KALM: Knowledgeable Agents by Offline Reinforcement Learning from Large Language Model Rollouts

NeurIPS 2024poster

Reinforcement learning (RL) traditionally trains agents using interaction data, which limits their capabilities to the scope of the training data. To create more knowledgeable agents, leveraging knowledge from large language models (LLMs) has shown a promising way. Despite various attempts to combin…

2024

Language Model Self-improvement by Reinforcement Learning Contemplation

ICLR 2024poster

Language model self-improvement (LMSI) techniques have recently gained significant attention as they improve language models without requiring external supervision. A common approach is reinforcement learning from AI feedback (RLAIF), which trains a reward model based on AI preference data and emplo…

Cited by 41SourcePDFScholar
2024

Limited Preference Aided Imitation Learning from Imperfect Demonstrations

ICML 2024poster

Imitation learning mimics high-quality policies from expert data for sequential decision-making tasks. However, its efficacy is hindered in scenarios where optimal demonstrations are unavailable, and only imperfect demonstrations are present. To address this issue, introducing additional limited hum…

Cited by 1SourcePDFScholar
2024

MGCL: Multi-Granularity Clue Learning for Emotion-Cause Pair Extraction via Cross-Grained Knowledge Distillation

EMNLP 2024finding

Emotion-cause pair extraction (ECPE) aims to identify emotion clauses and their corresponding cause clauses within a document. Traditional methods often rely on coarse-grained clause-level annotations, which can overlook valuable fine-grained clues. To address this issue, we propose Multi-Granularit…

Cited by 0SourcePDFScholar
2024

Multi-Agent Domain Calibration with a Handful of Offline Data

NeurIPS 2024poster

The shift in dynamics results in significant performance degradation of policies trained in the source domain when deployed in a different target domain, posing a challenge for the practical application of reinforcement learning (RL) in real-world scenarios. Domain transfer methods aim to bridge thi…

Cited by 0SourcePDFScholar
2024

Offline Transition Modeling via Contrastive Energy Learning

ICML 2024poster

Learning a high-quality transition model is of great importance for sequential decision-making tasks, especially in offline settings. Nevertheless, the complex behaviors of transition dynamics in real-world environments pose challenges for the standard forward models because of their inductive bias…

Cited by 3SourcePDFScholar
2024

Policy Learning from Tutorial Books via Understanding, Rehearsing and Introspecting

NeurIPS 2024oral

When humans need to learn a new skill, we can acquire knowledge through written books, including textbooks, tutorials, etc. However, current research for decision-making, like reinforcement learning (RL), has primarily required numerous real interactions with the target environment to learn a skill,…

2024

Policy Rehearsing: Training Generalizable Policies for Reinforcement Learning

ICLR 2024poster

Human beings can make adaptive decisions in a preparatory manner, i.e., by making preparations in advance, which offers significant advantages in scenarios where both online and offline experiences are expensive and limited. Meanwhile, current reinforcement learning methods commonly rely on numerous…

Cited by 4SourcePDFScholar
2024

Policy-conditioned Environment Models are More Generalizable

ICML 2024poster

In reinforcement learning, it is crucial to have an accurate environment dynamics model to evaluate different policies' value in downstream tasks like offline policy optimization and policy evaluation. However, the learned model is known to be inaccurate in predictions when evaluating target policie…

Cited by 2SourcePDFScholar
2024

Pre-training General User Representation with Multi-type APP Behaviors

IJCAI 2024poster

In numerous user-centric services on mobile applications (apps), accurately mining user interests and generating effective user representations are paramount. Traditional approaches, which often involve training task-specific user representations, are becoming increasingly impractical due to their h…

2024

Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximation

NeurIPS 2024poster

As a prominent category of imitation learning methods, adversarial imitation learning (AIL) has garnered significant practical success powered by neural network approximation. However, existing theoretical studies on AIL are primarily limited to simplified scenarios such as tabular and linear functi…

Cited by 1SourcePDFScholar
2024

ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Models

ICML 2024poster

Reinforcement Learning from Human Feedback (RLHF) is key to aligning Large Language Models (LLMs), typically paired with the Proximal Policy Optimization (PPO) algorithm. While PPO is a powerful method designed for general reinforcement learning tasks, it is overly sophisticated for LLMs, leading to…

2024

Rethinking the Development of Large Language Models from the Causal Perspective: A Legal Text Prediction Case Study

AAAI 2024technical

While large language models (LLMs) exhibit impressive performance on a wide range of NLP tasks, most of them fail to learn the causality from correlation, which disables them from learning rationales for predicting. Rethinking the whole developing process of LLMs is of great urgency as they are adop…

2024

Reward-Consistent Dynamics Models are Strongly Generalizable for Offline Reinforcement Learning

ICLR 2024spotlight

Learning a precise dynamics model can be crucial for offline reinforcement learning, which, unfortunately, has been found to be quite challenging. Dynamics models that are learned by fitting historical transitions often struggle to generalize to unseen transitions. In this study, we identify a hidde…

2024

Towards an Information Theoretic Framework of Context-Based Offline Meta-Reinforcement Learning

NeurIPS 2024spotlight

As a marriage between offline RL and meta-RL, the advent of offline meta-reinforcement learning (OMRL) has shown great promise in enabling RL agents to multi-task and quickly adapt while acquiring knowledge safely. Among which, context-based OMRL (COMRL) as a popular paradigm, aims to learn a univer…

2024

Unmixing Before Fusion: A Generalized Paradigm for Multi-Source-based Hyperspectral Image Synthesis

CVPR 2024poster

In the realm of AI data serves as a pivotal resource. Real-world hyperspectral images (HSIs) bearing wide spectral characteristics are particularly valuable. However the acquisition of HSIs is always costly and time-intensive resulting in a severe data-thirsty issue in HSI research and applications.…

2023

A Unified View of Deep Learning for Reaction and Retrosynthesis Prediction: Current Status and Future Challenges

IJCAI 2023poster

Reaction and retrosynthesis prediction are two fundamental tasks in computational chemistry. In recent years, these two tasks have attracted great attentions from both machine learning and drug discovery communities. Various deep learning approaches have been proposed to tackle these two problems an…

Cited by 14SourcePDFScholar
2023

AdaptSSR: Pre-training User Model with Augmentation-Adaptive Self-Supervised Ranking

NeurIPS 2023poster

User modeling, which aims to capture users' characteristics or interests, heavily relies on task-specific labeled data and suffers from the data sparsity issue. Several recent studies tackled this problem by pre-training the user model on massive user behavior sequences with a contrastive learning t…

2023

Adversarial Counterfactual Environment Model Learning

NeurIPS 2023spotlight

An accurate environment dynamics model is crucial for various downstream tasks in sequential decision-making, such as counterfactual prediction, off-policy evaluation, and offline reinforcement learning. Currently, these models were learned through empirical risk minimization (ERM) by step-wise fit…

2023

Anti-drifting Feature Selection via Deep Reinforcement Learning (Student Abstract)

AAAI 2023technical

Feature selection (FS) is a crucial procedure in machine learning pipelines for its significant benefits in removing data redundancy and mitigating model overfitting. Since concept drift is a widespread phenomenon in streaming data and could severely affect model performance, effective FS on concept…

Cited by 0SourcePDFScholar
2023

CMMA: Benchmarking Multi-Affection Detection in Chinese Multi-Modal Conversations

NeurIPS 2023poster

Human communication has a multi-modal and multi-affection nature. The inter-relatedness of different emotions and sentiments poses a challenge to jointly detect multiple human affections with multi-modal clues. Recent advances in this field employed multi-task learning paradigms to render the inter-…

2023

DL-NET: Dilation Location Network for Temporal Action Detection

ICASSP 2023accepted

Temporal Action Detection(TAD) is a challenge task in video understanding. The current methods mainly use global features for boundary matching or predefine all possible proposals, while ignoring long context information and local action boundary features, resulting in the decline of detection accur…

Cited by 0SourceScholar
2023

Deep Anomaly Detection and Search via Reinforcement Learning (Student Abstract)

AAAI 2023technical

Semi-supervised anomaly detection is a data mining task which aims at learning features from partially-labeled datasets. We propose Deep Anomaly Detection and Search (DADS) with reinforcement learning. During the training process, the agent searches for possible anomalies in unlabeled dataset to enh…

Cited by 1SourcePDFScholar
2023

Discovering Generalizable Multi-agent Coordination Skills from Multi-task Offline Data

ICLR 2023poster

Cooperative multi-agent reinforcement learning (MARL) faces the challenge of adapting to multiple tasks with varying agents and targets. Previous multi-task MARL approaches require costly interactions to simultaneously learn or fine-tune policies in different tasks. However, the situation that an ag…

Cited by 31SourcePDFScholar
2023

Doubly Stochastic Graph-based Non-autoregressive Reaction Prediction

IJCAI 2023poster

Organic reaction prediction is a critical task in drug discovery. Recently, researchers have achieved non-autoregressive reaction prediction by modeling the redistribution of electrons, resulting in state-of-the-art top-1 accuracy, and enabling parallel sampling. However, the current non-autoregress…

Cited by 9SourcePDFScholar
2023

Fast Teammate Adaptation in the Presence of Sudden Policy Change

UAI 2023poster

Cooperative multi-agent reinforcement learning (MARL), where agents coordinates with teammate(s) for a shared goal, may sustain non-stationary caused by the policy change of teammates. Prior works mainly concentrate on the policy change cross episodes, ignoring the fact that teammates may suffer f…

Cited by 13SourcePDFScholar
2023

Imitation Learning from Imperfection: Theoretical Justifications and Algorithms

NeurIPS 2023spotlight

Imitation learning (IL) algorithms excel in acquiring high-quality policies from expert data for sequential decision-making tasks. But, their effectiveness is hampered when faced with limited expert data. To tackle this challenge, a novel framework called (offline) IL with supplementary data has bee…

2023

Learning Generalizable Batch Active Learning Strategies via Deep Q-networks (Student Abstract)

AAAI 2023technical

To handle a large amount of unlabeled data, batch active learning (BAL) queries humans for the labels of a batch of the most valuable data points at every round. Most current BAL strategies are based on human-designed heuristics, such as uncertainty sampling or mutual information maximization. Howev…

Cited by 0SourcePDFScholar
2023

Learning World Models with Identifiable Factorization

NeurIPS 2023poster

Extracting a stable and compact representation of the environment is crucial for efficient reinforcement learning in high-dimensional, noisy, and non-stationary environments. Different categories of information coexist in such environments -- how to effectively extract and disentangle the informati…

Cited by 16SourcePDFScholar
2023

Model-Bellman Inconsistency for Model-based Offline Reinforcement Learning

ICML 2023poster

For offline reinforcement learning (RL), model-based methods are expected to be data-efficient as they incorporate dynamics models to generate more data. However, due to inevitable model errors, straightforwardly learning a policy in the model typically fails in the offline setting. Previous studies…

2023

Natural Language Instruction-following with Task-related Language Development and Translation

NeurIPS 2023poster

Natural language-conditioned reinforcement learning (RL) enables agents to follow human instructions. Previous approaches generally implemented language-conditioned RL by providing the policy with human instructions in natural language (NL) and training the policy to follow instructions. In this is…

Cited by 11SourcePDFScholar
2023

Object-Oriented Option Framework for Robotics Manipulation in Clutter

IROS 2023poster

Domestic service robots are becoming increasingly popular due to their ability to help people with household tasks. These robots often encounter the challenge of manipulating objects in cluttered environments (MoC), which is difficult due to the complexity of effective planning and control. Previous…

Cited by 11SourceScholar
2023

Policy Regularization with Dataset Constraint for Offline Reinforcement Learning

ICML 2023poster

We consider the problem of learning the best possible policy from a fixed dataset, known as offline Reinforcement Learning (RL). A common taxonomy of existing offline RL works is policy regularization, which typically constrains the learned policy by distribution or support of the behavior policy. H…

2023

Policy-Independent Behavioral Metric-Based Representation for Deep Reinforcement Learning

AAAI 2023technical

Behavioral metrics can calculate the distance between states or state-action pairs from the rewards and transitions difference. By virtue of their capability to filter out task-irrelevant information in theory, using them to shape a state embedding space becomes a new trend of representation learnin…

Cited by 4SourcePDFScholar
2023

Provably Efficient Adversarial Imitation Learning with Unknown Transitions

UAI 2023poster

Imitation learning (IL) has proven to be an effective method for learning good policies from expert demonstrations. Adversarial imitation learning (AIL), a subset of IL methods, is particularly promising, but its theoretical foundation in the presence of unknown transitions has yet to be fully devel…

2023

Robust Multi-Agent Coordination via Evolutionary Generation of Auxiliary Adversarial Attackers

AAAI 2023technical

Cooperative Multi-agent Reinforcement Learning (CMARL) has shown to be promising for many real-world applications. Previous works mainly focus on improving coordination ability via solving MARL-specific challenges (e.g., non-stationarity, credit assignment, scalability), but ignore the policy pertur…

2023

Uncertainty Estimation by Fisher Information-based Evidential Deep Learning

ICML 2023poster

Uncertainty estimation is a key factor that makes deep learning reliable in practical applications. Recently proposed evidential neural networks explicitly account for different uncertainties by treating the network's outputs as evidence to parameterize the Dirichlet distribution, and achieve impres…

2023

Untargeted Attack against Federated Recommendation Systems via Poisonous Item Embeddings and the Defense

AAAI 2023technical

Federated recommendation (FedRec) can train personalized recommenders without collecting user data, but the decentralized nature makes it susceptible to poisoning attacks. Most previous studies focus on the targeted attack to promote certain items, while the untargeted attack that aims to degrade th…

2022

Active Hierarchical Exploration with Stable Subgoal Representation Learning

ICLR 2022poster

Goal-conditioned hierarchical reinforcement learning (GCHRL) provides a promising approach to solving long-horizon tasks. Recently, its success has been extended to more general settings by concurrently learning hierarchical policies and subgoal representations. Although GCHRL possesses superior exp…

2022

Adapt to Environment Sudden Changes by Learning a Context Sensitive Policy

AAAI 2022technical

Dealing with real-world reinforcement learning (RL) tasks, we shall be aware that the environment may have sudden changes. We expect that a robust policy is able to handle such changes and adapt to the new environment rapidly. Context-based meta reinforcement learning aims at learning environment…

2022

Bayesian Optimistic Optimization: Optimistic Exploration for Model-based Reinforcement Learning

NeurIPS 2022accept

Reinforcement learning (RL) is a general framework for modeling sequential decision making problems, at the core of which lies the dilemma of exploitation and exploration. An agent failing to explore systematically will inevitably fail to learn efficiently. Optimism in the face of uncertainty (OFU)…

Cited by 11SourcePDFScholar
2022

Context-Aware Sparse Deep Coordination Graphs

ICLR 2022spotlight

Learning sparse coordination graphs adaptive to the coordination dynamics among agents is a long-standing problem in cooperative multi-agent learning. This paper studies this problem and proposes a novel method using the variance of payoff functions to construct context-aware sparse coordination top…

2022

Efficient Multi-Agent Communication via Shapley Message Value

IJCAI 2022poster

Utilizing messages from teammates is crucial in cooperative multi-agent tasks due to the partially observable nature of the environment. Naively asking messages from all teammates without pruning may confuse individual agents, hindering the learning process and impairing the whole system's performan…

2022

Efficient Multi-agent Communication via Self-supervised Information Aggregation

NeurIPS 2022accept

Utilizing messages from teammates can improve coordination in cooperative Multi-agent Reinforcement Learning (MARL). To obtain meaningful information for decision-making, previous works typically combine raw messages generated by teammates with local information as inputs for policy. However, neglec…

Cited by 35SourcePDFScholar
2022

Exploring Complementarity of Global and Local Spatiotemporal Information for Fake Face Video Detection

ICASSP 2022accepted

The spread of fake face videos leads to severe social concerns, which promotes the development of detection methods for these videos. Existing patch-based methods focus on local regions to find forgery common clues, while ignoring the important role of the global information. In this paper, a novel…

Cited by 0SourceScholar
2022

Invariant Action Effect Model for Reinforcement Learning

AAAI 2022technical

Good representations can help RL agents perform concise modeling of their surroundings, and thus support effective decision-making in complex environments. Previous methods learn good representations by imposing extra constraints on dynamics. However, in the causal perspective, the causation betw…

Cited by 13SourcePDFScholar
2022

Multi-Agent Concentrative Coordination with Decentralized Task Representation

IJCAI 2022poster

Value-based multi-agent reinforcement learning (MARL) methods hold the promise of promoting coordination in cooperative settings. Popular MARL methods mainly focus on the scalability or the representational capacity of value functions. Such a learning paradigm can reduce agents' uncertainties and pr…

2022

Multi-Agent Incentive Communication via Decentralized Teammate Modeling

AAAI 2022technical

Effective communication can improve coordination in cooperative multi-agent reinforcement learning (MARL). One popular communication scheme is exchanging agents' local observations or latent embeddings and using them to augment individual local policy input. Such a communication paradigm can reduce…

2022

Multi-agent Dynamic Algorithm Configuration

NeurIPS 2022accept

Automated algorithm configuration relieves users from tedious, trial-and-error tuning tasks. A popular algorithm configuration tuning paradigm is dynamic algorithm configuration (DAC), in which an agent learns dynamic configuration policies across instances by reinforcement learning (RL). However, i…

2022

NeoRL: A Near Real-World Benchmark for Offline Reinforcement Learning

NeurIPS 2022accept

Offline reinforcement learning (RL) aims at learning effective policies from historical data without extra environment interactions. During our experience of applying offline RL, we noticed that previous offline RL benchmarks commonly involve significant reality gaps, which we have identified includ…

2022

Pseudo-label Guided Cross-video Pixel Contrast for Robotic Surgical Scene Segmentation with Limited Annotations

IROS 2022poster

Surgical scene segmentation is fundamentally crucial for prompting cognitive assistance in robotic surgery. However, pixel-wise annotating surgical video in a frame-by-frame manner is expensive and time consuming. To greatly reduce the labeling burden, in this work, we study semi-supervised scene se…

Cited by 6SourcecodeScholar
2022

Tiny-NewsRec: Effective and Efficient PLM-based News Recommendation

EMNLP 2022main

News recommendation is a widely adopted technique to provide personalized news feeds for the user. Recently, pre-trained language models (PLMs) have demonstrated the great capability of natural language understanding and benefited news recommendation via improving news modeling. However, most existi…

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

Circles are like Ellipses, or Ellipses are like Circles? Measuring the Degree of Asymmetry of Static and Contextual Word Embeddings and the Implications to Representation Learning

AAAI 2021technical

Human judgments of word similarity have been a popular method of evaluating the quality of word embedding. But it fails to measure the geometry properties such as asymmetry. For example, it is more natural to say ``Ellipses are like Circles'' than ``Circles are like Ellipses''. Such asymmetry has be…

Cited by 0SourcePDFScholar
2021

Cross-modal Domain Adaptation for Cost-Efficient Visual Reinforcement Learning

NeurIPS 2021poster

In visual-input sim-to-real scenarios, to overcome the reality gap between images rendered in simulators and those from the real world, domain adaptation, i.e., learning an aligned representation space between simulators and the real world, then training and deploying policies in the aligned represe…

2021

Fast Pareto Optimization for Subset Selection with Dynamic Cost Constraints

IJCAI 2021poster

Subset selection with cost constraints is a fundamental problem with various applications such as influence maximization and sensor placement. The goal is to select a subset from a ground set to maximize a monotone objective function such that a monotone cost function is upper bounded by a budget. P…

Cited by 14SourcePDFScholar
2021

HieRec: Hierarchical User Interest Modeling for Personalized News Recommendation

ACL 2021long

User interest modeling is critical for personalized news recommendation. Existing news recommendation methods usually learn a single user embedding for each user from their previous behaviors to represent their overall interest. However, user interest is usually diverse and multi-grained, which is d…

2021

NewsBERT: Distilling Pre-trained Language Model for Intelligent News Application

EMNLP 2021finding

Pre-trained language models (PLMs) like BERT have made great progress in NLP. News articles usually contain rich textual information, and PLMs have the potentials to enhance news text modeling for various intelligent news applications like news recommendation and retrieval. However, most existing PL…

Cited by 47SourcePDFScholar
2021

Offline Model-based Adaptable Policy Learning

NeurIPS 2021poster

In reinforcement learning, a promising direction to avoid online trial-and-error costs is learning from an offline dataset. Current offline reinforcement learning methods commonly learn in the policy space constrained to in-support regions by the offline dataset, in order to ensure the robustness of…

2021

QPLEX: Duplex Dueling Multi-Agent Q-Learning

ICLR 2021poster

We explore value-based multi-agent reinforcement learning (MARL) in the popular paradigm of centralized training with decentralized execution (CTDE). CTDE has an important concept, Individual-Global-Max (IGM) principle, which requires the consistency between joint and local action selections to supp…

2021

Regret Minimization Experience Replay in Off-Policy Reinforcement Learning

NeurIPS 2021poster

In reinforcement learning, experience replay stores past samples for further reuse. Prioritized sampling is a promising technique to better utilize these samples. Previous criteria of prioritization include TD error, recentness and corrective feedback, which are mostly heuristically designed. In thi…

2020

RetroXpert: Decompose Retrosynthesis Prediction Like A Chemist

NeurIPS 2020spotlight

Retrosynthesis is the process of recursively decomposing target molecules into available building blocks. It plays an important role in solving problems in organic synthesis planning. To automate or assist in the retrosynthesis analysis, various retrosynthesis prediction algorithms have been propose…

2019

A GPS-aided Omnidirectional Visual-Inertial State Estimator in Ubiquitous Environments

IROS 2019poster

The visual-inertial navigation system (VINS) has been a practical approach for state estimation in recent years. In this paper, we propose a general GPS-aided omnidirectional visual-inertial state estimator capable of operating in ubiquitous environments and platforms. Our system consists of two par…

Cited by 59SourceScholar
2019

Automatic Calibration of Multiple 3D LiDARs in Urban Environments

IROS 2019poster

Multiple LiDARs have progressively emerged on autonomous vehicles for rendering a rich view and dense measurements. However, the lack of precise calibration negatively affects their potential applications. In this paper, we propose a novel system that enables automatic multi-LiDAR calibration method…

Cited by 64SourceScholar
2019

Bridging Machine Learning and Logical Reasoning by Abductive Learning

NeurIPS 2019poster

Perception and reasoning are two representative abilities of intelligence that are integrated seamlessly during human problem-solving processes. In the area of artificial intelligence (AI), the two abilities are usually realised by machine learning and logic programming, respectively. However, the t…