works on

From the 2 of 31 linked papers with an AI index.

activity
20242026
collaborators
Showing cs.LGShow all

10 papers · 1 filter

cs.LG2026

Branching Policy Optimization: Sandbox-Native Language Agent Reinforcement Learning

Bowei He, Yankai Chen, Xiaokun Zhang +1

The paper proposes Branching Policy Optimization (BPO), a reinforcement learning method for large language model agents operating in deterministic, snapshottable sandboxes, which l…

cs.LG2026

Discrete Diffusion Models: A Unified Framework from Tokenization to Generation

Ye Yuan, Weien Li, Rui Song +20

The paper proposes a unified framework for discrete denoising diffusion models that ties together tokenization, vocabulary design, and generation methods, showing how existing appr…

cs.LG2026

Distributionally Robust Set Representation Learning Under Inference-Time Element Corruption

Yankai Chen, Hanrong Zhang, Bowei He +2

Standard Set Representation Learning methods typically excel on curated data but often overlook the challenge of inference-time element corruption. This refers to scenarios where d…

cs.LG2026

Support-Proximity Augmented Diffusion Estimation for Offline Black-Box Optimization

Yonghan Yang, Ye Yuan, Zipeng Sun +5

Offline black-box optimization aims to discover novel designs with high property scores using only a static dataset, a task fundamentally challenged by the out-of-distribution (OOD…

cs.LG2026

MINER: Mining Multimodal Internal Representation for Efficient Retrieval

Weien Li, Rui Song, Zeyu Li +8

Visual document retrieval has become essential for accessing information in visually rich documents. Existing approaches fall into two camps. Late-interaction retrievers achieve st…

cs.LG2026

The Workload-Router-Pool Architecture for LLM Inference Optimization: A Vision Paper from the vLLM Semantic Router Project

Huamin Chen, Xunzhuo Liu, Bowei He +5

Over the past year, the vLLM Semantic Router project has released a series of work spanning: (1) core routing mechanisms -- signal-driven routing, context-length pool routing, rout…