← Search

Bo An

124 accepted papers

2026

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

ICML 2026poster

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

Cited by 0SourceScholar
2026

ArchetypeTrader: Reinforcement Learning for Selecting and Refining Learnable Strategic Archetypes in Quantitative Trading

AAAI 2026technical

Quantitative trading using mathematical models and automated execution to generate trading decisions has been widely applied acorss financial markets. Recently, reinforcement learning (RL) has emerged as a promising approach for developing profitable trading strategies, especially in highly volatile

Cited by 0SourcePDFScholar
2026

Beyond Trajectory-Level Attribution: Graph-Based Credit Assignment for Agentic Reinforcement Learning

ICML 2026poster

Group-based reinforcement learning (RL) methods have achieved remarkable success in improving the performance of large language models (LLMs) and have been rapidly extended to agentic tasks. However, their credit assignment relies heavily on coarse-grained trajectory-level attribution according to f…

Cited by 0SourceScholar
2026

Covariance Volume Maximization for Embodied Latent Exploration in Deep Reinforcement Learning

ICML 2026poster

Efficient exploration remains a key challenge in deep reinforcement learning, especially for embodied agents operating in realistic environments with high-dimensional observations and complex dynamics. Recent latent exploration methods define bonuses in a learned latent space, but often struggle in …

Cited by 0SourceScholar
2026

DynamicsBoost: Dynamic Plausible Video Generation via Annotation-Free Continuation Preference Optimization

CVPR 2026

Despite significant progress in text-to-video generation, current models still suffer from unrealistic dynamics, temporal inconsistency, and unstable semantic alignment. Existing preference alignment approaches rely on costly and often ambiguous human or VLM-based video preference annotation, which

Cited by 0SourceScholar
2026

Failure-Aware RL: Reliable Offline-To-Online Reinforcement Learning with Self-Recovery for Real-World Manipulation

ICRA 2026poster

Post-training algorithms based on deep reinforcement learning can push the limits of robotic models for specific objectives, such as generalizability, accuracy, and robustness. However, Intervention-requiring Failures (IR Failures) (e.g., a robot spilling water or breaking fragile glass) during real…

2026

GDBA Revisited: Unleashing the Power of Guided Local Search for Distributed Constraint Optimization

AAAI 2026technical

Local search is an important class of incomplete algorithms for solving Distributed Constraint Optimization Problems (DCOPs) but it often converges to poor local optima. While Generalized Distributed Breakout Algorithm (GDBA) provides a comprehensive rule set to escape premature convergence, its emp

Cited by 0SourcePDFScholar
2026

Generative Online Reinforcement Learning

ICML 2026poster

Reinforcement learning (RL) faces a persistent tension: policies that are stable to optimize (e.g., Gaussians) are often too simple to represent the multimodal action distributions required for complex control. Conversely, expressive generative policies—such as diffusion and flow matching—are freque…

Cited by 0SourceScholar
2026

Harnessing Reasoning Trajectories for Hallucination Detection via Answer-agreement Representation Shaping

ICML 2026poster

Large reasoning models (LRMs) often generate long, seemingly coherent reasoning traces yet still produce incorrect answers, making hallucination detection challenging. Although trajectories contain useful signals, directly using trace text or vanilla hidden states for detection is brittle: traces va…

Cited by 0SourceScholar
2026

Hierarchy-of-Groups Policy Optimization for Long-Horizon Agentic Tasks

ICLR 2026poster

Group-based reinforcement learning (RL), such as GRPO, has advanced the capabilities of large language models on long-horizon agentic tasks. To enable more fine-grained policy updates, recent research has increasingly shifted toward stepwise group-based policy optimization, which treats each step in…

Cited by 0SourcecodeScholar
2026

Letting Trajectories Spread: Quality-Preserving Control for Diverse Flow Matching

ICML 2026poster

Flow-based text-to-image models follow deterministic trajectories, making it costly to explore diverse modes under limited sampling budgets. Existing approaches to improving diversity often rely on retraining or degrade image fidelity. To address this limitation, we present a training-free, inferenc…

Cited by 0SourceScholar
2026

MobileIPL: Enhancing Mobile Agents Thinking Process via Iterative Preference Learning

ICLR 2026poster

The Chain of Action-Planning Thoughts (CoaT) paradigm has been shown to improve the reasoning performance of VLM-based mobile agents in GUI tasks. However, the scarcity of diverse CoaT trajectories limits the expressiveness and generalization ability of such agents. While self-training is commonly e…

Cited by 0SourceScholar
2026

REAR: Test-time Preference Realignment through Reward Decomposition

ICML 2026poster

Aligning large language models (LLMs) with diverse user preferences is a critical yet challenging task. While post-training methods can adapt models to specific needs, they often require costly data curation and additional training. Test-time scaling (TTS) presents an efficient, training-free altern…

Cited by 0SourceScholar
2026

SMAN-Bench: A Cross-System Benchmark for Mobile Agents under Single- and Multi-path, Ambiguous, and Noisy Tasks

ICLR 2026poster

VLM-based mobile agents are increasingly popular due to their capabilities to interact with smartphone GUIs and XML-structured texts and to complete daily tasks. However, existing online benchmarks fail to obtain stable critical reward signals under dynamic environmental changes, and neglect the inf…

Cited by 0SourcecodeScholar
2026

SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoning

ICLR 2026poster

Large Language Models (LLMs) can enhance their reasoning by interacting with external tools, a paradigm known as Tool-Integrated Reasoning (TIR). However, extending TIR to multi-turn settings using Reinforcement Learning (RL) often exhibits training instability and degraded performance. We attribute…

Cited by 0SourcecodeScholar
2026

Vulnerable Agent Identification in Large-Scale Multi-Agent Reinforcement Learning

ICML 2026poster

Partial agent failure becomes inevitable when systems scale up, making it crucial to identify the subset of agents whose failure causes worst-case system performance degradations. We study this Vulnerable Agent Identification (VAI) problem in large-scale multi-agent reinforcement learning (MARL). We…

Cited by 0SourceScholar
2025

AgentStudio: A Toolkit for Building General Virtual Agents

ICLR 2025poster

General virtual agents need to handle multimodal observations, master complex action spaces, and self-improve in dynamic, open-domain environments. However, existing environments are often domain-specific and require complex setups, which limits agent development and evaluation in real-world setting…

Cited by 31SourcePDFScholar
2025

Cradle: Empowering Foundation Agents towards General Computer Control

ICML 2025poster

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

2025

Deciphering the Extremes: A Novel Approach for Pathological Long-tailed Recognition in Scientific Discovery

NeurIPS 2025spotlight

Scientific discovery across diverse fields increasingly grapples with datasets exhibiting pathological long-tailed distributions: a few common phenomena overshadow a multitude of rare yet scientifically critical instances. Unlike standard benchmarks, these scientific datasets often feature extreme i…

Cited by 0SourceScholar
2025

EconGym: A Scalable AI Testbed with Diverse Economic Tasks

NeurIPS 2025poster

Artificial intelligence (AI) has become a powerful tool for economic research, enabling large-scale simulation and policy optimization. However, applying AI effectively requires simulation platforms for scalable training and evaluation—yet existing environments remain limited to simplified, narrowly…

Cited by 0SourceScholar
2025

Efficient Last-Iterate Convergence in Solving Extensive-Form Games

NeurIPS 2025poster

To establish last-iterate convergence for Counterfactual Regret Minimization (CFR) algorithms in learning a Nash equilibrium (NE) of extensive-form games (EFGs), recent studies reformulate learning an NE of the original EFG as learning the NEs of a sequence of (perturbed) regularized EFGs. Hence, pr…

Cited by 0SourcecodeScholar
2025

Empirical Study on Robustness and Resilience in Cooperative Multi-Agent Reinforcement Learning

NeurIPS 2025poster

In cooperative Multi-Agent Reinforcement Learning (MARL), it is a common practice to tune hyperparameters in ideal simulated environments to maximize cooperative performance. However, policies tuned for cooperation often fail to maintain robustness and resilience under real-world uncertainties. Buil…

Cited by 0SourceScholar
2025

Establishing Linear Surrogate Regret Bounds for Convex Smooth Losses via Convolutional Fenchel–Young Losses

NeurIPS 2025spotlight

Surrogate regret bounds, also known as excess risk bounds, bridge the gap between the convergence rates of surrogate and target losses. The regret transfer is lossless if the surrogate regret bound is linear. While convex smooth surrogate losses are appealing in particular due to the efficient estim…

Cited by 0SourceScholar
2025

Improving Reward Models with Proximal Policy Exploration for Preference-Based Reinforcement Learning

NeurIPS 2025poster

Reinforcement learning (RL) heavily depends on well-designed reward functions, which are often biased and difficult to design for complex behaviors. Preference-based RL (PbRL) addresses this by learning reward models from human feedback, but its practicality is constrained by a critical dilemma: wh…

Cited by 0SourceScholar
2025

Incentivizing LLMs to Self-Verify Their Answers

NeurIPS 2025poster

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

Cited by 0SourcecodeScholar
2025

Influence-Based Fair Selection for Sample-Discriminative Backdoor Attack

AAAI 2025technical

Backdoor attacks have posed a serious threat in machine learning models, wherein adversaries can poison training samples with maliciously crafted triggers to compromise the victim model. Advanced backdoor attack methods have focused on selectively poisoning more vulnerable training samples, achievin…

Cited by 0SourcePDFScholar
2025

Let's Revise Step-by-Step: A Unified Local Search Framework for Code Generation with LLMs

NeurIPS 2025poster

Large Language Models (LLMs) with inference-time scaling techniques show promise for code generation, yet face notable efficiency and scalability challenges. Construction-based tree-search methods suffer from rapid growth in tree size, high token consumption, and lack of anytime property. In contras…

Cited by 0SourceScholar
2025

MF-LLM: Simulating Population Decision Dynamics via a Mean-Field Large Language Model Framework

NeurIPS 2025poster

Simulating collective decision-making involves more than aggregating individual behaviors; it emerges from dynamic interactions among individuals. While large language models (LLMs) offer strong potential for social simulation, achieving quantitative alignment with real-world data remains a key chal…

Cited by 0SourcecodeScholar
2025

OPHR: Mastering Volatility Trading with Multi-Agent Deep Reinforcement Learning

NeurIPS 2025poster

Options markets represent one of the most sophisticated segments of the financial ecosystem, with prices that directly reflect market uncertainty. In this paper, we introduce the first reinforcement learning (RL) framework specifically designed for volatility trading through options, focusing on pro…

Cited by 0SourcecodeScholar
2025

OpticalNet: An Optical Imaging Dataset and Benchmark Beyond the Diffraction Limit

CVPR 2025highlight

Optical imaging capable of resolving nanoscale features would revolutionize scientific research and engineering applications across biomedicine, smart manufacturing, and semiconductor quality control. However, due to the physical phenomenon of diffraction, the optical resolution is limited to approx…

Cited by 0SourcePDFScholar
2025

Policy Optimization under Imperfect Human Interactions with Agent-Gated Shared Autonomy

ICLR 2025poster

We introduce AGSA, an Agent-Gated Shared Autonomy framework that learns from high-level human feedback to tackle the challenges of reward-free training, safe exploration, and imperfect low-level human control. Recent human-in-the loop learning methods enable human participants to intervene a learnin…

Cited by 0SourcePDFScholar
2025

Policy Regularization on Globally Accessible States in Cross-Dynamics Reinforcement Learning

ICML 2025spotlight

To learn from data collected in diverse dynamics, Imitation from Observation (IfO) methods leverage expert state trajectories based on the premise that recovering expert state distributions in other dynamics facilitates policy learning in the current one. However, Imitation Learning inherently impos…

Cited by 0SourcePDFScholar
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

Removing Prompt-template Bias in Reinforcement Learning from Human Feedback

ACL 2025finding

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

Cited by 0SourcePDFScholar
2025

Representation Surgery in Model Merging with Probabilistic Modeling

ICML 2025poster

Model merging aims to achieve multitask performance by merging multiple expert models without the need to access the raw training data. Recent research identified the \textit{representation bias} of model merging, characterized by a discrepancy in the representation distribution between the merged a…

Cited by 0SourcePDFScholar
2025

Towards Efficient Online Tuning of VLM Agents via Counterfactual Soft Reinforcement Learning

ICML 2025poster

Online fine-tuning vision-language model (VLM) agents with reinforcement learning (RL) has shown promise for equipping agents with multi-step, goal-oriented capabilities in dynamic environments. However, their open-ended textual action space and non-end-to-end nature of action generation present sig…

2024

Configurable Mirror Descent: Towards a Unification of Decision Making

ICML 2024poster

Decision-making problems, categorized as single-agent, e.g., Atari, cooperative multi-agent, e.g., Hanabi, competitive multi-agent, e.g., Hold'em poker, and mixed cooperative and competitive, e.g., football, are ubiquitous in the real world. Although various methods have been proposed to address the…

2024

Consistent Multi-Class Classification from Multiple Unlabeled Datasets

ICLR 2024spotlight

Weakly supervised learning aims to construct effective predictive models from imperfectly labeled data. The recent trend of weakly supervised learning has focused on how to learn an accurate classifier from completely unlabeled data, given little supervised information such as class priors. In this…

Cited by 0SourcePDFScholar
2024

EarnHFT: Efficient Hierarchical Reinforcement Learning for High Frequency Trading

AAAI 2024technical

High-frequency trading (HFT) is using computer algorithms to make trading decisions in short time scales (e.g., second-level), which is widely used in the Cryptocurrency (Crypto) market, (e.g., Bitcoin). Reinforcement learning (RL) in financial research has shown stellar performance on many quantita…

2024

IMM: An Imitative Reinforcement Learning Approach with Predictive Representation Learning for Automatic Market Making

IJCAI 2024poster

Market making (MM) via Reinforcement Learning (RL) has attracted significant attention in financial trading. Most existing RL-based MM methods focus on optimizing single-price level strategies which fail at frequent order cancellations and loss of queue priority. By comparison, strategies involving…

Cited by 2SourcePDFScholar
2024

Improving Unsupervised Hierarchical Representation with Reinforcement Learning

CVPR 2024poster

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

2024

Latent Logic Tree Extraction for Event Sequence Explanation from LLMs

ICML 2024poster

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

Cited by 5SourcePDFScholar
2024

MaNo: Exploiting Matrix Norm for Unsupervised Accuracy Estimation Under Distribution Shifts

NeurIPS 2024poster

Leveraging the model’s outputs, specifically the logits, is a common approach to estimating the test accuracy of a pre-trained neural network on out-of-distribution (OOD) samples without requiring access to the corresponding ground-truth labels. Despite their ease of implementation and computational…

2024

Market-GAN: Adding Control to Financial Market Data Generation with Semantic Context

AAAI 2024technical

Financial simulators play an important role in enhancing forecasting accuracy, managing risks, and fostering strategic financial decision-making. Despite the development of financial market simulation methodologies, existing frameworks often struggle with adapting to specialized simulation context.…

Cited by 10SourcePDFScholar
2024

Mitigating Underfitting in Learning to Defer with Consistent Losses

AISTATS 2024poster

Learning to defer (L2D) allows the classifier to defer its prediction to an expert for safer predictions, by balancing the system’s accuracy and extra costs incurred by consulting the expert. Various loss functions have been proposed for L2D, but they were shown to cause the underfitting of trained…

Cited by 8SourcePDFScholar
2024

On the Vulnerability of Adversarially Trained Models Against Two-faced Attacks

ICLR 2024poster

Adversarial robustness is an important standard for measuring the quality of learned models, and adversarial training is an effective strategy for improving the adversarial robustness of models. In this paper, we disclose that adversarially trained models are vulnerable to two-faced attacks, where s…

Cited by 0SourcePDFScholar
2024

PoRank: A Practical Framework for Learning to Rank Policies

IJCAI 2024poster

In many real-world scenarios, we need to select from a set of candidate policies before online deployment. Although existing Off-policy evaluation (OPE) methods can be used to estimate the online performance, they suffer from high variance. Fortunately, we care only about the ranking of the candidat…

2024

Reinforcement Nash Equilibrium Solver

IJCAI 2024poster

Nash Equilibrium (NE) is the canonical solution concept of game theory, which provides an elegant tool to understand the rationalities. Though mixed strategy NE exists in any game with finite players and actions, computing NE in two- or multi-player general-sum games is PPAD-Complete. Various altern…

2024

Resisting Stochastic Risks in Diffusion Planners with the Trajectory Aggregation Tree

ICML 2024spotlight

Diffusion planners have shown promise in handling long-horizon and sparse-reward tasks due to the non-autoregressive plan generation. However, their inherent stochastic risk of generating infeasible trajectories presents significant challenges to their reliability and stability. We introduce a novel…

2024

S$2$AC: Energy-Based Reinforcement Learning with Stein Soft Actor Critic

ICLR 2024poster

Learning expressive stochastic policies instead of deterministic ones has been proposed to achieve better stability, sample complexity and robustness. Notably, in Maximum Entropy reinforcement learning (MaxEnt RL), the policy is modeled as an expressive energy-based model (EBM) over the Q-values. Ho…

2024

Safe and Robust Subgame Exploitation in Imperfect Information Games

ICML 2024poster

Opponent exploitation is an important task for players to exploit the weaknesses of others in games. Existing approaches mainly focus on balancing between exploitation and exploitability but are often vulnerable to modeling errors and deceptive adversaries. To address this problem, our paper offers…

Cited by 2SourcePDFScholar
2024

Self-adaptive PSRO: Towards an Automatic Population-based Game Solver

IJCAI 2024poster

Policy-Space Response Oracles (PSRO) as a general algorithmic framework has achieved state-of-the-art performance in learning equilibrium policies of two-player zero-sum games. However, the hand-crafted hyperparameter value selection in most of the existing works requires extensive domain knowledge,…

Cited by 3SourcePDFScholar
2024

Solving Homogeneous and Heterogeneous Cooperative Tasks with Greedy Sequential Execution

ICLR 2024spotlight

Cooperative multi-agent reinforcement learning (MARL) is extensively used for solving complex cooperative tasks, and value decomposition methods are a prevalent approach for this domain. However, these methods have not been successful in addressing both homogeneous and heterogeneous tasks simultaneo…

Cited by 2SourcePDFScholar
2024

Synapse: Trajectory-as-Exemplar Prompting with Memory for Computer Control

ICLR 2024poster

Building agents with large language models (LLMs) for computer control is a burgeoning research area, where the agent receives computer states and performs actions to complete complex tasks. Previous computer agents have demonstrated the benefits of in-context learning (ICL); however, their performa…

2024

Transition-Informed Reinforcement Learning for Large-Scale Stackelberg Mean-Field Games

AAAI 2024technical

Many real-world scenarios including fleet management and Ad auctions can be modeled as Stackelberg mean-field games (SMFGs) where a leader aims to incentivize a large number of homogeneous self-interested followers to maximize her utility. Existing works focus on cases with a small number of heterog…

2024

True Knowledge Comes from Practice: Aligning Large Language Models with Embodied Environments via Reinforcement Learning

ICLR 2024poster

Despite the impressive performance across numerous tasks, large language models (LLMs) often fail in solving simple decision-making tasks due to the misalignment of the knowledge in LLMs with environments. On the contrary, reinforcement learning (RL) agents learn policies from scratch, which makes t…

Cited by 47SourcePDFScholar
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

An Efficient Deep Reinforcement Learning Algorithm for Solving Imperfect Information Extensive-Form Games

AAAI 2023technical

One of the most popular methods for learning Nash equilibrium (NE) in large-scale imperfect information extensive-form games (IIEFGs) is the neural variants of counterfactual regret minimization (CFR). CFR is a special case of Follow-The-Regularized-Leader (FTRL). At each iteration, the neural varia…

2023

Complex Contagion Influence Maximization: A Reinforcement Learning Approach

IJCAI 2023poster

In influence maximization (IM), the goal is to find a set of seed nodes in a social network that maximizes the influence spread. While most IM problems focus on classical influence cascades (e.g., Independent Cascade and Linear Threshold) which assume individual influence cascade probability is inde…

Cited by 1SourcePDFScholar
2023

Consistent Complementary-Label Learning via Order-Preserving Losses

AISTATS 2023poster

In contrast to ordinary supervised classification tasks that require massive data with high-quality labels, complementary-label learning (CLL) deals with the weakly-supervised learning scenario where each instance is equipped with a complementary label, which specifies a class the instance does not…

Cited by 17SourcePDFScholar
2023

Controlling Type Confounding in Ad Hoc Teamwork with Instance-wise Teammate Feedback Rectification

ICML 2023poster

Ad hoc teamwork requires an agent to cooperate with unknown teammates without prior coordination. Many works propose to abstract teammate instances into high-level representation of types and then pre-train the best response for each type. However, most of them do not consider the distribution of te…

Cited by 2SourcePDFScholar
2023

Exploring Leximin Principle for Fair Core-Selecting Combinatorial Auctions: Payment Rule Design and Implementation

IJCAI 2023poster

Core-selecting combinatorial auctions (CAs) restrict the auction result in the core such that no coalitions could improve their utilities by engaging in collusion. The minimum-revenue-core (MRC) rule is a widely used core-selecting payment rule to maximize the total utilities of all bidders. However…

2023

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

NeurIPS 2023poster

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

Cited by 6SourcePDFScholar
2023

In Defense of Softmax Parametrization for Calibrated and Consistent Learning to Defer

NeurIPS 2023poster

Enabling machine learning classifiers to defer their decision to a downstream expert when the expert is more accurate will ensure improved safety and performance. This objective can be achieved with the learning-to-defer framework which aims to jointly learn how to classify and how to defer to the e…

Cited by 21SourcePDFScholar
2023

Mitigating Memorization of Noisy Labels by Clipping the Model Prediction

ICML 2023poster

In the presence of noisy labels, designing robust loss functions is critical for securing the generalization performance of deep neural networks. Cross Entropy (CE) loss has been shown to be not robust to noisy labels due to its unboundedness. To alleviate this issue, existing works typically design…

Cited by 32SourcePDFScholar
2023

Offline RL with Discrete Proxy Representations for Generalizability in POMDPs

NeurIPS 2023poster

Offline Reinforcement Learning (RL) has demonstrated promising results in various applications by learning policies from previously collected datasets, reducing the need for online exploration and interactions. However, real-world scenarios usually involve partial observability, which brings crucial…

Cited by 0SourcePDFScholar
2023

On the Importance of Feature Separability in Predicting Out-Of-Distribution Error

NeurIPS 2023poster

Estimating the generalization performance is practically challenging on out-of-distribution (OOD) data without ground-truth labels. While previous methods emphasize the connection between distribution difference and OOD accuracy, we show that a large domain gap not necessarily leads to a low test ac…

Cited by 15SourcePDFScholar
2023

Population-size-Aware Policy Optimization for Mean-Field Games

ICLR 2023poster

In this work, we attempt to bridge the two fields of finite-agent and infinite-agent games, by studying how the optimal policies of agents evolve with the number of agents (population size) in mean-field games, an agent-centric perspective in contrast to the existing works focusing typically on the…

Cited by 6SourcePDFScholar
2023

RPM: Generalizable Multi-Agent Policies for Multi-Agent Reinforcement Learning

ICLR 2023poster

Despite the recent advancement in multi-agent reinforcement learning (MARL), the MARL agents easily overfit the training environment and perform poorly in evaluation scenarios where other agents behave differently. Obtaining generalizable policies for MARL agents is thus necessary but challenging ma…

Cited by 2SourcePDFScholar
2023

ResAct: Reinforcing Long-term Engagement in Sequential Recommendation with Residual Actor

ICLR 2023poster

Long-term engagement is preferred over immediate engagement in sequential recommendation as it directly affects product operational metrics such as daily active users (DAUs) and dwell time. Meanwhile, reinforcement learning (RL) is widely regarded as a promising framework for optimizing long-term en…

Cited by 30SourcePDFScholar
2023

Solving Large-Scale Pursuit-Evasion Games Using Pre-trained Strategies

AAAI 2023technical

Pursuit-evasion games on graphs model the coordination of police forces chasing a fleeing felon in real-world urban settings, using the standard framework of imperfect-information extensive-form games (EFGs). In recent years, solving EFGs has been largely dominated by the Policy-Space Response Oracl…

Cited by 12SourcePDFScholar
2023

State Regularized Policy Optimization on Data with Dynamics Shift

NeurIPS 2023poster

In many real-world scenarios, Reinforcement Learning (RL) algorithms are trained on data with dynamics shift, i.e., with different underlying environment dynamics. A majority of current methods address such issue by training context encoders to identify environment parameters. Data with dynamics shi…

Cited by 17SourcePDFScholar
2023

TradeMaster: A Holistic Quantitative Trading Platform Empowered by Reinforcement Learning

NeurIPS 2023poster

The financial markets, which involve over \$90 trillion market capitals, attract the attention of innumerable profit-seeking investors globally. Recent explosion of reinforcement learning in financial trading (RLFT) research has shown stellar performance on many quantitative trading tasks. However,…

2022

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

NeurIPS 2022accept

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

Cited by 10SourcePDFScholar
2022

Correlation-Based Algorithm for Team-Maxmin Equilibrium in Multiplayer Extensive-Form Games

IJCAI 2022poster

Efficient algorithms computing a Nash equilibrium have been successfully applied to large zero- sum two-player extensive-form games (e.g., poker). However, in multiplayer games, computing a Nash equilibrium is generally hard, and the equilibria are not exchangeable, which makes players face the prob…

2022

DO-GAN: A Double Oracle Framework for Generative Adversarial Networks

CVPR 2022poster

In this paper, we propose a new approach to train Generative Adversarial Networks (GANs) where we deploy a double-oracle framework using the generator and discriminator oracles. GAN is essentially a two-player zero-sum game between the generator and the discriminator. Training GANs is challenging as…

Cited by 5PDFScholar
2022

Deep Attentive Belief Propagation: Integrating Reasoning and Learning for Solving Constraint Optimization Problems

NeurIPS 2022accept

Belief Propagation (BP) is an important message-passing algorithm for various reasoning tasks over graphical models, including solving the Constraint Optimization Problems (COPs). It has been shown that BP can achieve state-of-the-art performance on various benchmarks by mixing old and new messages…

Cited by 7SourcePDFScholar
2022

GearNet: Stepwise Dual Learning for Weakly Supervised Domain Adaptation

AAAI 2022technical

This paper studies a weakly supervised domain adaptation (WSDA) problem, where we only have access to the source domain with noisy labels, from which we need to transfer useful information to the unlabeled target domain. Although there have been a few studies on this problem, most of them only explo…

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

Mitigating Neural Network Overconfidence with Logit Normalization

ICML 2022spotlight

Detecting out-of-distribution inputs is critical for the safe deployment of machine learning models in the real world. However, neural networks are known to suffer from the overconfidence issue, where they produce abnormally high confidence for both in- and out-of-distribution inputs. In this work,…

2022

NSGZero: Efficiently Learning Non-exploitable Policy in Large-Scale Network Security Games with Neural Monte Carlo Tree Search

AAAI 2022technical

How resources are deployed to secure critical targets in networks can be modelled by Network Security Games (NSGs). While recent advances in deep learning (DL) provide a powerful approach to dealing with large-scale NSGs, DL methods such as NSG-NFSP suffer from the problem of data inefficiency. Furt…

Cited by 10SourcePDFScholar
2022

Open-Sampling: Exploring Out-of-Distribution data for Re-balancing Long-tailed datasets

ICML 2022spotlight

Deep neural networks usually perform poorly when the training dataset suffers from extreme class imbalance. Recent studies found that directly training with out-of-distribution data (i.e., open-set samples) in a semi-supervised manner would harm the generalization performance. In this work, we theor…

Cited by 45SourcePDFScholar
2022

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

NeurIPS 2022accept

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

Cited by 24SourcePDFScholar
2022

Pretrained Cost Model for Distributed Constraint Optimization Problems

AAAI 2022technical

Distributed Constraint Optimization Problems (DCOPs) are an important subclass of combinatorial optimization problems, where information and controls are distributed among multiple autonomous agents. Previously, Machine Learning (ML) has been largely applied to solve combinatorial optimization probl…

2021

CFR-MIX: Solving Imperfect Information Extensive-Form Games with Combinatorial Action Space

IJCAI 2021poster

In many real-world scenarios, a team of agents must coordinate with each other to compete against an opponent. The challenge of solving this type of game is that the team's joint action space grows exponentially with the number of agents, which results in the inefficiency of the existing algorithms,…

Cited by 12SourcePDFScholar
2021

Commission Fee is not Enough: A Hierarchical Reinforced Framework for Portfolio Management

AAAI 2021technical

Portfolio management via reinforcement learning is at the forefront of fintech research, which explores how to optimally reallocate a fund into different financial assets over the long term by trial-and-error. Existing methods are impractical since they usually assume each reallocation can be finish…

Cited by 50SourcePDFScholar
2021

Complexity and Algorithms for Exploiting Quantal Opponents in Large Two-Player Games

AAAI 2021technical

Solution concepts of traditional game theory assume entirely rational players; therefore, their ability to exploit subrational opponents is limited. One type of subrationality that describes human behavior well is the quantal response. While there exist algorithms for computing solutions against qua…

Cited by 12SourcePDFScholar
2021

Computing Ex Ante Coordinated Team-Maxmin Equilibria in Zero-Sum Multiplayer Extensive-Form Games

AAAI 2021technical

Computational game theory has many applications in the modern world in both adversarial situations and the optimization of social good. While there exist many algorithms for computing solutions in two-player interactions, finding optimal strategies in multiplayer interactions efficiently remains an…

Cited by 31SourcePDFScholar
2021

Computing Quantal Stackelberg Equilibrium in Extensive-Form Games

AAAI 2021technical

Deployments of game-theoretic solution concepts in the real world have highlighted the necessity to consider human opponents' boundedly rational behavior. If subrationality is not addressed, the system can face significant losses in terms of expected utility. While there exist algorithms for computi…

Cited by 9SourcePDFScholar
2021

Contingency-aware influence maximization: A reinforcement learning approach

UAI 2021poster

The influence maximization (IM) problem aims at finding a subset of seed nodes in a social network that maximize the spread of influence. In this study, we focus on a sub-class of IM problems, where whether the nodes are willing to be the seeds when being invited is uncertain, called contingency-awa…

2021

Neural Regret-Matching for Distributed Constraint Optimization Problems

IJCAI 2021poster

Distributed constraint optimization problems (DCOPs) are a powerful model for multi-agent coordination and optimization, where information and controls are distributed among multiple agents by nature. Sampling-based algorithms are important incomplete techniques for solving medium-scale DCOPs. Howev…

Cited by 6SourcePDFScholar
2021

Open-set Label Noise Can Improve Robustness Against Inherent Label Noise

NeurIPS 2021poster

Learning with noisy labels is a practically challenging problem in weakly supervised learning. In the existing literature, open-set noises are always considered to be poisonous for generalization, similar to closed-set noises. In this paper, we empirically show that open-set noisy labels can be non-…

2021

Personalized Adaptive Meta Learning for Cold-start User Preference Prediction

AAAI 2021technical

A common challenge in personalized user preference prediction is the cold-start problem. Due to the lack of user-item interactions, directly learning from the new users' log data causes serious over-fitting problem. Recently, many existing studies regard the cold-start personalized preference predic…

Cited by 76SourcePDFScholar
2021

Pointwise Binary Classification with Pairwise Confidence Comparisons

ICML 2021spotlight

To alleviate the data requirement for training effective binary classifiers in binary classification, many weakly supervised learning settings have been proposed. Among them, some consider using pairwise but not pointwise labels, when pointwise labels are not accessible due to privacy, confidentiali…

Cited by 33SourcePDFScholar
2021

RMIX: Learning Risk-Sensitive Policies for Cooperative Reinforcement Learning Agents

NeurIPS 2021poster

Current value-based multi-agent reinforcement learning methods optimize individual Q values to guide individuals' behaviours via centralized training with decentralized execution (CTDE). However, such expected, i.e., risk-neutral, Q value is not sufficient even with CTDE due to the randomness of rew…

Cited by 59SourcePDFScholar
2021

Solving Large-Scale Extensive-Form Network Security Games via Neural Fictitious Self-Play

IJCAI 2021poster

Securing networked infrastructures is important in the real world. The problem of deploying security resources to protect against an attacker in networked domains can be modeled as Network Security Games (NSGs). Unfortunately, existing approaches, including the deep learning-based approaches, are in…

Cited by 19SourcePDFScholar
2020

Combating Noisy Labels by Agreement: A Joint Training Method with Co-Regularization

CVPR 2020poster

Deep Learning with noisy labels is a practically challenging problem in weakly-supervised learning. The state-of-the-art approaches "Decoupling" and "Co-teaching+" claim that the "disagreement" strategy is crucial for alleviating the problem of learning with noisy labels. In this paper, we start fro…

Cited by 706PDFcodeScholar
2020

Dinkelbach-Type Algorithm for Computing Quantal Stackelberg Equilibrium

IJCAI 2020poster

Stackelberg security games (SSGs) have been deployed in many real-world situations to optimally allocate scarce resource to protect targets against attackers. However, actual human attackers are not perfectly rational and there are several behavior models that attempt to predict subrational behavior…

Cited by 0SourcePDFScholar
2020

Enhancing Neural Models with Vulnerability via Adversarial Attack

COLING 2020main

Natural Language Sentence Matching (NLSM) serves as the core of many natural language processing tasks. 1) Most previous work develops a single specific neural model for NLSM tasks. 2) There is no previous work considering adversarial attack to improve the performance of NLSM tasks. 3) Adversarial a…

2020

I²HRL: Interactive Influence-based Hierarchical Reinforcement Learning

IJCAI 2020poster

Hierarchical reinforcement learning (HRL) is a promising approach to solve tasks with long time horizons and sparse rewards. It is often implemented as a high-level policy assigning subgoals to a low-level policy. However, it suffers the high-level non-stationarity problem since the low-level policy…

Cited by 0SourcePDFScholar
2020

Learning Efficient Multi-agent Communication: An Information Bottleneck Approach

ICML 2020poster

We consider the problem of the limited-bandwidth communication for multi-agent reinforcement learning, where agents cooperate with the assistance of a communication protocol and a scheduler. The protocol and scheduler jointly determine which agent is communicating what message and to whom. Under the…

2020

Learning Expensive Coordination: An Event-Based Deep RL Approach

ICLR 2020poster

Existing works in deep Multi-Agent Reinforcement Learning (MARL) mainly focus on coordinating cooperative agents to complete certain tasks jointly. However, in many cases of the real world, agents are self-interested such as employees in a company and clubs in a league. Therefore, the leader, i.e.,…

Cited by 11SourceScholar
2020

Speeding Up Incomplete GDL-based Algorithms for Multi-agent Optimization with Dense Local Utilities

IJCAI 2020poster

Incomplete GDL-based algorithms including Max-sum and its variants are important methods for multi-agent optimization. However, they face a significant scalability challenge as the computational overhead grows exponentially with respect to the arity of each utility function. Generic Domain Pruning (…

Cited by 0SourcePDFScholar
2019

Manipulating a Learning Defender and Ways to Counteract

NeurIPS 2019poster

In Stackelberg security games when information about the attacker's payoffs is uncertain, algorithms have been proposed to learn the optimal defender commitment by interacting with the attacker and observing their best responses. In this paper, we show that, however, these algorithms can be easily m…

Cited by 22SourcePDFScholar