Song, Kaitao

21 publications

NeurIPS 2025 Chain-of-Model Learning for Language Model Xiaohua Wang, Kaitao Song, Xu Tan, Huiqiang Jiang, Chengruidong Zhang, Yongliang Shen, Cen Lu, Zihao Li, Zifan Song, Caihua Shan, Yansen Wang, Kan Ren, Xiaoqing Zheng, Tao Qin, Yuqing Yang, Dongsheng Li, Lili Qiu

NeurIPS 2025 Let LRMs Break Free from Overthinking via Self-Braking Tuning Haoran Zhao, Yuchen Yan, Yongliang Shen, Haolei Xu, Wenqi Zhang, Kaitao Song, Jian Shao, Weiming Lu, Jun Xiao, Yueting Zhuang

NeurIPS 2025 Mind the Gap: Bridging Thought Leap for Improved Chain-of-Thought Tuning Haolei Xu, Yuchen Yan, Yongliang Shen, Wenqi Zhang, Guiyang Hou, Shengpei Jiang, Kaitao Song, Weiming Lu, Jun Xiao, Yueting Zhuang

NeurIPS 2024 Can Graph Learning Improve Planning in LLM-Based Agents? Xixi Wu, Yifei Shen, Caihua Shan, Kaitao Song, Siwei Wang, Bohang Zhang, Jiarui Feng, Hong Cheng, Wei Chen, Yun Xiong, Dongsheng Li

ICLR 2024 Connecting Large Language Models with Evolutionary Algorithms Yields Powerful Prompt Optimizers Qingyan Guo, Rui Wang, Junliang Guo, Bei Li, Kaitao Song, Xu Tan, Guoqing Liu, Jiang Bian, Yujiu Yang

ICLRW 2024 EASYTOOL: Enhancing LLM-Based Agents with Concise Tool Instruction Siyu Yuan, Kaitao Song, Jiangjie Chen, Xu Tan, Yongliang Shen, Kan Ren, Dongsheng Li, Deqing Yang

NeurIPSW 2024 EvoAgent: Towards Automatic Multi-Agent Generation via Evolutionary Algorithms Siyu Yuan, Kaitao Song, Jiangjie Chen, Xu Tan, Dongsheng Li, Deqing Yang

ICML 2024 NaturalSpeech 3: Zero-Shot Speech Synthesis with Factorized Codec and Diffusion Models Zeqian Ju, Yuancheng Wang, Kai Shen, Xu Tan, Detai Xin, Dongchao Yang, Eric Liu, Yichong Leng, Kaitao Song, Siliang Tang, Zhizheng Wu, Tao Qin, Xiangyang Li, Wei Ye, Shikun Zhang, Jiang Bian, Lei He, Jinyu Li, Sheng Zhao

ICLR 2024 PromptTTS 2: Describing and Generating Voices with Text Prompt Yichong Leng, Zhifang Guo, Kai Shen, Zeqian Ju, Xu Tan, Eric Liu, Yufei Liu, Dongchao Yang, Leying Zhang, Kaitao Song, Lei He, Xiangyang Li, Sheng Zhao, Tao Qin, Jiang Bian

NeurIPS 2024 TaskBench: Benchmarking Large Language Models for Task Automation Yongliang Shen, Kaitao Song, Xu Tan, Wenqi Zhang, Kan Ren, Siyu Yuan, Weiming Lu, Dongsheng Li, Yueting Zhuang

ICLRW 2024 TaskBench: Benchmarking Large Language Models for Task Automation Yongliang Shen, Kaitao Song, Xu Tan, Wenqi Zhang, Kan Ren, Siyu Yuan, Weiming Lu, Dongsheng Li, Yueting Zhuang

ICML 2023 CircuitNet: A Generic Neural Network to Realize Universal Circuit Motif Modeling Yansen Wang, Xinyang Jiang, Kan Ren, Caihua Shan, Xufang Luo, Dongqi Han, Kaitao Song, Yifei Shen, Dongsheng Li

NeurIPS 2023 HuggingGPT: Solving AI Tasks with ChatGPT and Its Friends in Hugging Face Yongliang Shen, Kaitao Song, Xu Tan, Dongsheng Li, Weiming Lu, Yueting Zhuang

AAAI 2023 SoftCorrect: Error Correction with Soft Detection for Automatic Speech Recognition Yichong Leng, Xu Tan, Wenjie Liu, Kaitao Song, Rui Wang, Xiang-Yang Li, Tao Qin, Edward Lin, Tie-Yan Liu

ICML 2022 Analyzing and Mitigating Interference in Neural Architecture Search Jin Xu, Xu Tan, Kaitao Song, Renqian Luo, Yichong Leng, Tao Qin, Tie-Yan Liu, Jian Li

NeurIPS 2022 Transcormer: Transformer for Sentence Scoring with Sliding Language Modeling Kaitao Song, Yichong Leng, Xu Tan, Yicheng Zou, Tao Qin, Dongsheng Li

ICCV 2021 Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction Without Convolutions Wenhai Wang, Enze Xie, Xiang Li, Deng-Ping Fan, Kaitao Song, Ding Liang, Tong Lu, Ping Luo, Ling Shao

AAAI 2021 SongMASS: Automatic Song Writing with Pre-Training and Alignment Constraint Zhonghao Sheng, Kaitao Song, Xu Tan, Yi Ren, Wei Ye, Shikun Zhang, Tao Qin

NeurIPS 2020 MPNet: Masked and Permuted Pre-Training for Language Understanding Kaitao Song, Xu Tan, Tao Qin, Jianfeng Lu, Tie-Yan Liu

IJCAI 2020 Neural Machine Translation with Error Correction Kaitao Song, Xu Tan, Jianfeng Lu

ICML 2019 MASS: Masked Sequence to Sequence Pre-Training for Language Generation Kaitao Song, Xu Tan, Tao Qin, Jianfeng Lu, Tie-Yan Liu