← Search

Ke Shen

13 accepted papers

2026

AutoTool: Dynamic Tool Selection and Integration for Agentic Reasoning

ICML 2026poster

Agentic reinforcement learning has advanced large language models (LLMs) to reason through long chain-of-thought trajectories while interleaving external tool use. Existing approaches assume a fixed inventory of tools, which limits the adaptability of LLM agents to new or evolving toolsets. We prese…

Cited by 0SourceScholar
2026

Joint Selection for Large-Scale Pre-Training Data via Policy Gradient-based Mask Learning

ICLR 2026poster

A fine-grained data recipe is crucial for pre-training large language models (LLMs), as it can significantly enhance training efficiency and model performance. One important ingredient in the recipe is to select samples based on scores produced by defined rules, LLM judgment, or statistical informat…

Cited by 0SourcecodeScholar
2026

Latent Collaboration in Multi-Agent Systems

ICML 2026spotlight

Multi-agent systems (MAS) extend large language models (LLMs) from independent single-model reasoning to coordinative system-level intelligence. While existing LLM agents depend on text-based mediation for reasoning and communication, we take a step forward by enabling models to collaborate directly…

Cited by 0SourceScholar
2026

RLAnything: Forge Environment, Policy, and Reward Model in Completely Dynamic RL System

ICML 2026poster

The quality of both the environment and the reward model fundamentally governs the effectiveness of reinforcement learning. Accordingly, we propose RLAnything, a reinforcement learning framework that dynamically optimizes each component through closed-loop optimization, amplifying learning signals a…

Cited by 0SourceScholar
2026

Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models

ICLR 2026poster

The extension of diffusion models to language tasks has shown promising results, but their post-training methods remain largely unexplored. We highlight the importance of aligning a diffusion language model’s preference-inference trajectory with its post-training objective. To this end, we propose T…

Cited by 0SourcecodeScholar
2026

Unveiling Downstream Performance Scaling of LLMs: A Clustering-Based Perspective

ICLR 2026poster

The escalating scale and cost of Large Language Models (LLMs) training necessitate accurate pre-training prediction of downstream task performance for efficient resource allocation. This is challenged by: 1) the emergence phenomenon, where unpredictable capabilities appearing suddenly at critical mo…

Cited by 0SourceScholar
2026

When to Memorize and When to Stop: Gated Recurrent Memory for Long-Context Reasoning

ICML 2026poster

While reasoning over long context is crucial for various real-world applications, it remains challenging for large language models (LLMs) as they suffer from performance degradation as the context length grows. Recent work MemAgent has tried to tackle this by processing context chunk-by-chunk in an …

Cited by 0SourceScholar
2025

CURE: Co-Evolving Coders and Unit Testers via Reinforcement Learning

NeurIPS 2025spotlight

Mathematical reasoning in large language models has been successfully incentivized through reinforcement learning with verifiable rewards, leading to improved one-shot precision. In this work, we turn our focus to the coding domain. Beyond one-shot precision, we highlight unit test generation as ano…

Cited by 0SourceScholar
2025

MMaDA: Multimodal Large Diffusion Language Models

NeurIPS 2025poster

We introduce MMaDA, a novel class of multimodal diffusion foundation models designed to achieve superior performance across diverse domains such as textual reasoning, multimodal understanding, and text-to-image generation. The approach is distinguished by three key innovations: (i) MMaDA adopts a un…

Cited by 0SourcecodeScholar
2025

ReasonFlux-PRM: Trajectory-Aware PRMs for Long Chain-of-Thought Reasoning in LLMs

NeurIPS 2025poster

Process Reward Models (PRMs) have recently emerged as a powerful framework for supervising intermediate reasoning steps in large language models (LLMs). Previous PRMs are primarily trained on model final output responses and struggle to evaluate intermediate thinking trajectories robustly, especiall…

Cited by 0SourcecodeScholar
2024

Multipartite Entity Resolution: Motivating a K-Tuple Perspective (Student Abstract)

AAAI 2024technical

Entity Resolution (ER) is the problem of algorithmically matching records, mentions, or entries that refer to the same underlying real-world entity. Traditionally, the problem assumes (at most) two datasets, between which records need to be matched. There is considerably less research in ER when k >…

Cited by 0SourcePDFScholar