activity
20232026
most citedRevisiting LLM Reasoning via Information Bottleneck

1 citations · 2 across the 8 of their papers we have counts for

collaborators

8 papers

cs.AI2026

A Step Back: Prefix Importance Ratio Stabilizes Policy Optimization

Shiye Lei, Zhihao Cheng, Dacheng Tao

Reinforcement learning (RL) post-training has increasingly demonstrated strong ability to elicit reasoning behaviors in large language models (LLMs). For training efficiency, rollo…

cs.LG2025

Offline Behavioral Data Selection

Shiye Lei, Zhihao Cheng, Dacheng Tao

Behavioral cloning is a widely adopted approach for offline policy learning from expert demonstrations. However, the large scale of offline behavioral datasets often results in com…

cs.LG2025

State Diversity Matters in Offline Behavior Distillation

Shiye Lei, Zhihao Cheng, Dacheng Tao

Offline Behavior Distillation (OBD), which condenses massive offline RL data into a compact synthetic behavioral dataset, offers a promising approach for efficient policy training…

cs.LG2025

Drawback of Enforcing Equivariance and its Compensation via the Lens of Expressive Power

Yuzhu Chen, Tian Qin, Xinmei Tian +2

Equivariant neural networks encode the intrinsic symmetry of data as an inductive bias, which has achieved impressive performance in wide domains. However, the understanding to the…

cs.AI2025★ 1 cited

Revisiting LLM Reasoning via Information Bottleneck

Shiye Lei, Zhihao Cheng, Kai Jia +1

Large language models (LLMs) have recently demonstrated remarkable progress in reasoning capabilities through reinforcement learning with verifiable rewards (RLVR). By leveraging s…

cs.LG2024

Offline Behavior Distillation

Shiye Lei, Sen Zhang, Dacheng Tao

Massive reinforcement learning (RL) data are typically collected to train policies offline without the need for interactions, but the large data volume can cause training inefficie…

cs.GT2024

Inverse learning of black-box aggregator for robust Nash equilibrium

Guanpu Chen, Gehui Xu, Fengxiang He +3

In this note, we investigate the robustness of Nash equilibria (NE) in multi-player aggregative games with coupling constraints. There are many algorithms for computing an NE of an…

cs.CV2023★ 1 cited

Image Captions are Natural Prompts for Text-to-Image Models

Shiye Lei, Hao Chen, Sen Zhang +2

With the rapid development of Artificial Intelligence Generated Content (AIGC), it has become a common practice to train models on synthetic data due to data-scarcity and privacy l…