Song, Lei

24 publications

ICLR 2025 Efficient Discovery of Pareto Front for Multi-Objective Reinforcement Learning Ruohong Liu, Yuxin Pan, Linjie Xu, Lei Song, Pengcheng You, Yize Chen, Jiang Bian

AAAI 2025 Exploring Rationale Learning for Continual Graph Learning Lei Song, Jiaxing Li, Qinghua Si, Shihan Guan, Youyong Kong

ICML 2025 From Complex to Atomic: Enhancing Augmented Generation via Knowledge-Aware Dual Rewriting and Reasoning Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian

AAAI 2025 HePa: Heterogeneous Graph Prompting for All-Level Classification Tasks Jia Jinghong, Lei Song, Jiaxing Li, Youyong Kong

TMLR 2025 Knowing What Not to Do: Leverage Language Model Insights for Action Space Pruning in Multi-Agent Reinforcement Learning Zhihao Liu, Xianliang Yang, Zichuan Liu, Yifan Xia, Wei Jiang, Yuanyu Zhang, Lijuan Li, Guoliang Fan, Lei Song, Jiang Bian

L4DC 2025 Multi-Constraint Safe Reinforcement Learning via Closed-Form Solution for Log-Sum-Exp Approximation of Control Barrier Functions Chenggang Wang, Xinyi Wang, Yutong Dong, Lei Song, Xinping Guan

NeurIPS 2025 NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data Haolong Qian, Xianliang Yang, Ling Zhang, Lei Song, Jiang Bian, Chun Yuan

IJCAI 2025 Reinforced In-Context Black-Box Optimization Lei Song, Chen-Xiao Gao, Ke Xue, Chenyang Wu, Dong Li, Jianye Hao, Zongzhang Zhang, Chao Qian

ICML 2025 Unveiling Markov Heads in Pretrained Language Models for Offline Reinforcement Learning Wenhao Zhao, Qiushui Xu, Linjie Xu, Lei Song, Jinyu Wang, Chunlai Zhou, Jiang Bian

IJCAI 2024 Diversification of Adaptive Policy for Effective Offline Reinforcement Learning Yunseon Choi, Li Zhao, Chuheng Zhang, Lei Song, Jiang Bian, Kee-Eung Kim

TMLR 2024 Mildly Constrained Evaluation Policy for Offline Reinforcement Learning Linjie Xu, Zhengyao Jiang, Jinyu Wang, Lei Song, Jiang Bian

NeurIPS 2024 Monte Carlo Tree Search Based Space Transfer for Black Box Optimization Shukuan Wang, Ke Xue, Lei Song, Xiaobin Huang, Chao Qian

ICML 2024 Position: Rethinking Post-Hoc Search-Based Neural Approaches for Solving Large-Scale Traveling Salesman Problems Yifan Xia, Xianliang Yang, Zichuan Liu, Zhihao Liu, Lei Song, Jiang Bian

NeurIPS 2024 Protecting Your LLMs with Information Bottleneck Zichuan Liu, Zefan Wang, Linjie Xu, Jinyu Wang, Lei Song, Tianchun Wang, Chunlin Chen, Wei Cheng, Jiang Bian

AAAI 2024 Stochastic Bayesian Optimization with Unknown Continuous Context Distribution via Kernel Density Estimation Xiaobin Huang, Lei Song, Ke Xue, Chao Qian

ICML 2024 TimeX++: Learning Time-Series Explanations with Information Bottleneck Zichuan Liu, Tianchun Wang, Jimeng Shi, Xu Zheng, Zhuomin Chen, Lei Song, Wenqian Dong, Jayantha Obeysekera, Farhad Shirani, Dongsheng Luo

ICLR 2024 Whittle Index with Multiple Actions and State Constraint for Inventory Management Chuheng Zhang, Xiangsen Wang, Wei Jiang, Xianliang Yang, Siwei Wang, Lei Song, Jiang Bian

AAAI 2023 H-TSP: Hierarchically Solving the Large-Scale Traveling Salesman Problem Xuanhao Pan, Yan Jin, Yuandong Ding, Mingxiao Feng, Li Zhao, Lei Song, Jiang Bian

AAAI 2023 Pointerformer: Deep Reinforced Multi-Pointer Transformer for the Traveling Salesman Problem Yan Jin, Yuandong Ding, Xuanhao Pan, Kun He, Li Zhao, Tao Qin, Lei Song, Jiang Bian

ICML 2023 Robust Situational Reinforcement Learning in Face of Context Disturbances Jinpeng Zhang, Yufeng Zheng, Chuheng Zhang, Li Zhao, Lei Song, Yuan Zhou, Jiang Bian

NeurIPS 2022 Monte Carlo Tree Search Based Variable Selection for High Dimensional Bayesian Optimization Lei Song, Ke Xue, Xiaobin Huang, Chao Qian

IJCAI 2018 Model Checking Probabilistic Epistemic Logic for Probabilistic Multiagent Systems Chen Fu, Andrea Turrini, Xiaowei Huang, Lei Song, Yuan Feng, Lijun Zhang

IJCAI 2015 A Simple Probabilistic Extension of Modal Mu-Calculus Wanwei Liu, Lei Song, Ji Wang, Lijun Zhang

IJCAI 2015 Planning for Stochastic Games with Co-Safe Objectives Lei Song, Yuan Feng, Lijun Zhang