PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
February 2025
Lei Song
Principal Researcher
Papiers de Collaboration
-
-
-
-
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
-
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
-
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
-
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
-
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
-
-
-
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
-
Pre-Trained Large Language Models for Industrial Control
Lei Song, Chuheng Zhang, Li Zhao, Jiang Bian
ArXiv | August 2023, Vol abs/2308.03028
-
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
-
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
-
-
-
PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generation
Jinyu Wang, Jingjing Fu, Rui Wang, Lei Song, Jiang Bian
February 2025
-
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
-
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
-
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
-
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
-
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
-
Pre-Trained Large Language Models for Industrial Control
Lei Song, Chuheng Zhang, Li Zhao, Jiang Bian
ArXiv | August 2023, Vol abs/2308.03028
-
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
-
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
-
-
-
-
-
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
-
-
-
-
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
-
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
-
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
-
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
-
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
-
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
-
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
-
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
-
-
-
Pre-Trained Large Language Models for Industrial Control
Lei Song, Chuheng Zhang, Li Zhao, Jiang Bian
ArXiv | August 2023, Vol abs/2308.03028
-