Shen, Ke

9 publications

ICLR 2026 Joint Selection for Large-Scale Pre-Training Data via Policy Gradient-Based Mask Learning Ziqing Fan, Yuqiao Xian, Yan Sun, Ke Shen, Li Shen

ICLR 2026 Reformulation for Pretraining Data Augmentation Xintong Hao, Rui-Jie Zhu, Ge Zhang, Ke Shen, Chenggang Li

ICLR 2026 Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models Yinjie Wang, Ling Yang, Bowen Li, Ye Tian, Ke Shen, Mengdi Wang

ICLR 2026 Unveiling Downstream Performance Scaling of LLMs: A Clustering-Based Perspective Chengyin Xu, Kaiyuan Chen, Xiao Li, Ke Shen, Chenggang Li

NeurIPS 2025 Co-Evolving LLM Coder and Unit Tester via Reinforcement Learning Yinjie Wang, Ling Yang, Ye Tian, Ke Shen, Mengdi Wang

NeurIPS 2025 MMaDA: Multimodal Large Diffusion Language Models Ling Yang, Ye Tian, Bowen Li, Xinchen Zhang, Ke Shen, Yunhai Tong, Mengdi Wang

NeurIPS 2025 ReasonFlux-PRM: Trajectory-Aware PRMs for Long Chain-of-Thought Reasoning in LLMs Jiaru Zou, Ling Yang, Jingwen Gu, Jiahao Qiu, Ke Shen, Jingrui He, Mengdi Wang

AAAI 2024 Multipartite Entity Resolution: Motivating a K-Tuple Perspective (Student Abstract) Adin Aberbach, Mayank Kejriwal, Ke Shen

AAAI 2024 The Generalization and Robustness of Transformer-Based Language Models on Commonsense Reasoning Ke Shen