← Search

Ke Xue

26 accepted papers

2026

Beyond Token-level Supervision: Unlocking the Potential of Decoding-based Regression via Reinforcement Learning

ICML 2026poster

Decoding-based regression, which reformulates regression as a sequence generation task, has emerged as a promising paradigm of applying large language models for numerical prediction. However, its progress is hindered by the misalignment between discrete token-level objectives (e.g., cross-entropy) …

Cited by 1SourceScholar
2026

Dual-View Predictive Diffusion: Lightweight Speech Enhancement via Spectrogram-Image Synergy

ICML 2026poster

Diffusion models have recently set new benchmarks in Speech Enhancement (SE). However, most existing score-based models treat speech spectrograms merely as generic 2D images, applying uniform processing that ignores the intrinsic structural sparsity of audio, which results in inefficient spectral re…

Cited by 0SourceScholar
2026

Expertise Can Be Helpful for Reinforcement Learning-based Macro Placement

ICLR 2026poster

Chip placement determines the locations of electronic components on a chip layout, which directly impacts performance, power, and area (PPA) metrics, and thus is a critical step in electronic design automation (EDA). As modern chips scale to accommodate millions of components, manual placement by hu…

Cited by 0SourceScholar
2025

Offline Model-Based Optimization by Learning to Rank

ICLR 2025poster

Offline model-based optimization (MBO) aims to identify a design that maximizes a black-box function using only a fixed, pre-collected dataset of designs and their corresponding scores. This problem has garnered significant attention from both scientific and industrial domains. A common approach in…

2025

Pareto Set Learning for Multi-Objective Reinforcement Learning

AAAI 2025technical

Multi-objective decision-making problems have emerged in numerous real-world scenarios, such as video games, navigation and robotics. Considering the clear advantages of Reinforcement Learning (RL) in optimizing decision-making processes, researchers have delved into the development of Multi-Objecti…

Cited by 1SourcePDFScholar
2025

Reinforced In-Context Black-Box Optimization

IJCAI 2025

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

2025

Sequential Multi-Agent Dynamic Algorithm Configuration

NeurIPS 2025poster

The performance of an algorithm often critically depends on its hyperparameter configuration. Dynamic algorithm configuration (DAC) is a recent trend in automated machine learning, which can dynamically adjust the algorithm’s configuration during the execution process and relieve users from tedious…

Cited by 0SourcecodeScholar
2025

Towards Universal Offline Black-Box Optimization via Learning Language Model Embeddings

ICML 2025poster

The pursuit of universal black-box optimization (BBO) algorithms is a longstanding goal. However, unlike domains such as language or vision, where scaling structured data has driven generalization, progress in offline BBO remains hindered by the lack of unified representations for heterogeneous nume…

2024

Monte Carlo Tree Search based Space Transfer for Black Box Optimization

NeurIPS 2024spotlight

Bayesian optimization (BO) is a popular method for computationally expensive black-box optimization. However, traditional BO methods need to solve new problems from scratch, leading to slow convergence. Recent studies try to extend BO to a transfer learning setup to speed up the optimization, where…

2024

Reinforcement Learning Policy as Macro Regulator Rather than Macro Placer

NeurIPS 2024poster

In modern chip design, placement aims at placing millions of circuit modules, which is an essential step that significantly influences power, performance, and area (PPA) metrics. Recently, reinforcement learning (RL) has emerged as a promising technique for improving placement quality, especially ma…

2024

Sample-Efficient Quality-Diversity by Cooperative Coevolution

ICLR 2024spotlight

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

Cited by 11SourcePDFScholar
2024

Stochastic Bayesian Optimization with Unknown Continuous Context Distribution via Kernel Density Estimation

AAAI 2024technical

Bayesian optimization (BO) is a sample-efficient method and has been widely used for optimizing expensive black-box functions. Recently, there has been a considerable interest in BO literature in optimizing functions that are affected by context variable in the environment, which is uncontrollable b…

2024

Towards Generalizable Neural Solvers for Vehicle Routing Problems via Ensemble with Transferrable Local Policy

IJCAI 2024poster

Machine learning has been adapted to help solve NP-hard combinatorial optimization problems. One prevalent way is learning to construct solutions by deep neural networks, which has been receiving more and more attention due to the high efficiency and less requirement for expert knowledge. However, m…

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

Multi-objective Optimization-based Selection for Quality-Diversity by Non-surrounded-dominated Sorting

IJCAI 2023poster

Quality-Diversity (QD) algorithms, a subset of evolutionary algorithms, maintain an archive (i.e., a set of solutions) and simulate the natural evolution process through iterative selection and reproduction, with the goal of generating a set of high-quality and diverse solutions. Though having found…

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…

2022

Evolutionary Diversity Optimization with Clustering-based Selection for Reinforcement Learning

ICLR 2022poster

Reinforcement Learning (RL) has achieved significant successes, which aims to obtain a single policy maximizing the expected cumulative rewards for a given task. However, in many real-world scenarios, e.g., navigating in complex environments and controlling robots, one may need to find a set of poli…

Cited by 24SourcePDFScholar
2022

Monte Carlo Tree Search based Variable Selection for High Dimensional Bayesian Optimization

NeurIPS 2022accept

Bayesian optimization (BO) is a class of popular methods for expensive black-box optimization, and has been widely applied to many scenarios. However, BO suffers from the curse of dimensionality, and scaling it to high-dimensional problems is still a challenge. In this paper, we propose a variable s…

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…