Lei Song

Principal Researcher

Publications

View by:

- SIT-Graph: State Integrated Tool Graph for Multi-Turn Agents
  
  Sijia Li, Yuchen Huang, Zifan Liu, Zijian Li, Jingjing Fu, Lei Song, Jiang Bian, Jun Zhang, Rui Wang
  
  December 2025
  
  Publication
- NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data
  
  Haolong Qian, Xianliang Yang, Ling Zhang, Lei Song, Jiang Bian, Chun Yuan
  
  NeurIPS 2025 | October 2025
  
  Publication
- PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
  
  Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
  
  February 2025
  
  Publication Github
- Diversification of Adaptive Policy for Effective Offline Reinforcement Learning
  
  Yunseon Choi, Li Zhao, Chuheng Zhang, Lei Song, Jiang Bian, Kee-Eung Kim
  
  IJCAI 2024 | July 2024
  
  DOI
- 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
  
  2024 International Conference on Machine Learning | June 2024
  
  PDF
- TimeX++: Learning Time-Series Explanations with Information Bottleneck
  
  Zichuan Liu, Tianchun Wang, Jimeng Shi, Xu Zheng, Zhuomin Chen, Lei Song, Wenqian Dong, J. Obeysekera, Farhad Shirani, Dongsheng Luo
  
  ICML 2024 | May 2024
  
  Publication
- 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
  
  2024 International Conference on Learning Representations | May 2024
  
  Publication
- Protecting Your LLMs with Information Bottleneck
  
  Zichuan Liu, Zefan Wang, Linjie Xu, Jinyu Wang, Lei Song, Tianchun Wang, Chunlin Chen, Wei Cheng, Jiang Bian
  
  NeurIPS 2024 | April 2024
  
  DOI Publication Publication
- Robust Situational Reinforcement Learning in Face of Context Disturbances
  
  Jinpeng Zhang, Yufeng Zheng, Chuheng Zhang, Li Zhao, Lei Song, Yuan Zhou, Jiang Bian
  
  Proceedings of the 40th International Conference on Machine Learning (ICML) | September 2023
  
  Publication
- Pre-Trained Large Language Models for Industrial Control
  
  Lei Song, Chuheng Zhang, Li Zhao, Jiang Bian
  
  ArXiv | August 2023, Vol abs/2308.03028
  
  DOI Publication Publication Video
- 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
  
  AAAI 2023 | February 2023
  
  Publication
- TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets
  
  Yuanying Cai, Chuheng Zhang, Li Zhao, Wei Shen, Xuyun Zhang, Lei Song, Jiang Bian, Tao Qin, Tie-Yan Liu
  
  IEEE International Conference on Data Mining (Best Student Paper Runner-Up Awards) | January 2023
  
  Publication

- SIT-Graph: State Integrated Tool Graph for Multi-Turn Agents
  
  Sijia Li, Yuchen Huang, Zifan Liu, Zijian Li, Jingjing Fu, Lei Song, Jiang Bian, Jun Zhang, Rui Wang
  
  December 2025
  
  Publication
- NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data
  
  Haolong Qian, Xianliang Yang, Ling Zhang, Lei Song, Jiang Bian, Chun Yuan
  
  NeurIPS 2025 | October 2025
  
  Publication
- PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
  
  Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
  
  February 2025
  
  Publication Github
- Diversification of Adaptive Policy for Effective Offline Reinforcement Learning
  
  Yunseon Choi, Li Zhao, Chuheng Zhang, Lei Song, Jiang Bian, Kee-Eung Kim
  
  IJCAI 2024 | July 2024
  
  DOI
- TimeX++: Learning Time-Series Explanations with Information Bottleneck
  
  Zichuan Liu, Tianchun Wang, Jimeng Shi, Xu Zheng, Zhuomin Chen, Lei Song, Wenqian Dong, J. Obeysekera, Farhad Shirani, Dongsheng Luo
  
  ICML 2024 | May 2024
  
  Publication
- 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
  
  2024 International Conference on Learning Representations | May 2024
  
  Publication
- Protecting Your LLMs with Information Bottleneck
  
  Zichuan Liu, Zefan Wang, Linjie Xu, Jinyu Wang, Lei Song, Tianchun Wang, Chunlin Chen, Wei Cheng, Jiang Bian
  
  NeurIPS 2024 | April 2024
  
  DOI Publication Publication
- Robust Situational Reinforcement Learning in Face of Context Disturbances
  
  Jinpeng Zhang, Yufeng Zheng, Chuheng Zhang, Li Zhao, Lei Song, Yuan Zhou, Jiang Bian
  
  Proceedings of the 40th International Conference on Machine Learning (ICML) | September 2023
  
  Publication
- Pre-Trained Large Language Models for Industrial Control
  
  Lei Song, Chuheng Zhang, Li Zhao, Jiang Bian
  
  ArXiv | August 2023, Vol abs/2308.03028
  
  DOI Publication Publication Video
- 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
  
  AAAI 2023 | February 2023
  
  Publication
- TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets
  
  Yuanying Cai, Chuheng Zhang, Li Zhao, Wei Shen, Xuyun Zhang, Lei Song, Jiang Bian, Tao Qin, Tie-Yan Liu
  
  IEEE International Conference on Data Mining (Best Student Paper Runner-Up Awards) | January 2023
  
  Publication
- PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
  
  Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
  
  February 2025
  
  Publication Github
- PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
  
  Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
  
  February 2025
  
  Publication Github
- 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
  
  2024 International Conference on Machine Learning | June 2024
  
  PDF

- SIT-Graph: State Integrated Tool Graph for Multi-Turn Agents
  
  Sijia Li, Yuchen Huang, Zifan Liu, Zijian Li, Jingjing Fu, Lei Song, Jiang Bian, Jun Zhang, Rui Wang
  
  December 2025
  
  Publication
- PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
  
  Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
  
  February 2025
  
  Publication Github
- NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data
  
  Haolong Qian, Xianliang Yang, Ling Zhang, Lei Song, Jiang Bian, Chun Yuan
  
  NeurIPS 2025 | October 2025
  
  Publication
- Diversification of Adaptive Policy for Effective Offline Reinforcement Learning
  
  Yunseon Choi, Li Zhao, Chuheng Zhang, Lei Song, Jiang Bian, Kee-Eung Kim
  
  IJCAI 2024 | July 2024
  
  DOI
- 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
  
  2024 International Conference on Machine Learning | June 2024
  
  PDF
- TimeX++: Learning Time-Series Explanations with Information Bottleneck
  
  Zichuan Liu, Tianchun Wang, Jimeng Shi, Xu Zheng, Zhuomin Chen, Lei Song, Wenqian Dong, J. Obeysekera, Farhad Shirani, Dongsheng Luo
  
  ICML 2024 | May 2024
  
  Publication
- 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
  
  2024 International Conference on Learning Representations | May 2024
  
  Publication
- Protecting Your LLMs with Information Bottleneck
  
  Zichuan Liu, Zefan Wang, Linjie Xu, Jinyu Wang, Lei Song, Tianchun Wang, Chunlin Chen, Wei Cheng, Jiang Bian
  
  NeurIPS 2024 | April 2024
  
  DOI Publication Publication
- Robust Situational Reinforcement Learning in Face of Context Disturbances
  
  Jinpeng Zhang, Yufeng Zheng, Chuheng Zhang, Li Zhao, Lei Song, Yuan Zhou, Jiang Bian
  
  Proceedings of the 40th International Conference on Machine Learning (ICML) | September 2023
  
  Publication
- 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
  
  AAAI 2023 | February 2023
  
  Publication
- TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets
  
  Yuanying Cai, Chuheng Zhang, Li Zhao, Wei Shen, Xuyun Zhang, Lei Song, Jiang Bian, Tao Qin, Tie-Yan Liu
  
  IEEE International Conference on Data Mining (Best Student Paper Runner-Up Awards) | January 2023
  
  Publication
- Pre-Trained Large Language Models for Industrial Control
  
  Lei Song, Chuheng Zhang, Li Zhao, Jiang Bian
  
  ArXiv | August 2023, Vol abs/2308.03028
  
  DOI Publication Publication Video

Lei Song

Publications

SIT-Graph: State Integrated Tool Graph for Multi-Turn Agents

NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data

PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation

Diversification of Adaptive Policy for Effective Offline Reinforcement Learning

Position: Rethinking Post-Hoc Search-Based Neural Approaches for Solving Large-Scale Traveling Salesman Problems

TimeX++: Learning Time-Series Explanations with Information Bottleneck

Whittle Index with Multiple Actions and State Constraint for Inventory Management

Protecting Your LLMs with Information Bottleneck

Robust Situational Reinforcement Learning in Face of Context Disturbances

Pre-Trained Large Language Models for Industrial Control

Pointerformer: Deep Reinforced Multi-Pointer Transformer for the Traveling Salesman Problem

TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets

SIT-Graph: State Integrated Tool Graph for Multi-Turn Agents

NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data

PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation

Diversification of Adaptive Policy for Effective Offline Reinforcement Learning

TimeX++: Learning Time-Series Explanations with Information Bottleneck

Whittle Index with Multiple Actions and State Constraint for Inventory Management

Protecting Your LLMs with Information Bottleneck

Robust Situational Reinforcement Learning in Face of Context Disturbances

Pre-Trained Large Language Models for Industrial Control

Pointerformer: Deep Reinforced Multi-Pointer Transformer for the Traveling Salesman Problem

TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets

PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation

PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation

Position: Rethinking Post-Hoc Search-Based Neural Approaches for Solving Large-Scale Traveling Salesman Problems

SIT-Graph: State Integrated Tool Graph for Multi-Turn Agents

PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation

NaDRO: Leveraging Dual-Reward Strategies for LLMs Training on Noisy Data

Diversification of Adaptive Policy for Effective Offline Reinforcement Learning

Position: Rethinking Post-Hoc Search-Based Neural Approaches for Solving Large-Scale Traveling Salesman Problems

TimeX++: Learning Time-Series Explanations with Information Bottleneck

Whittle Index with Multiple Actions and State Constraint for Inventory Management

Protecting Your LLMs with Information Bottleneck

Robust Situational Reinforcement Learning in Face of Context Disturbances

Pointerformer: Deep Reinforced Multi-Pointer Transformer for the Traveling Salesman Problem

TD3 with Reverse KL Regularizer for Offline Reinforcement Learning from Mixed Datasets

Pre-Trained Large Language Models for Industrial Control