Jiang, Song

9 publications

ICLR 2026 SPG: Sandwiched Policy Gradient for Masked Diffusion Language Models Chenyu Wang, Paria Rashidinejad, DiJia Su, Song Jiang, Sid Wang, Siyan Zhao, Cai Zhou, Shannon Zejiang Shen, Feiyu Chen, Tommi Jaakkola, Yuandong Tian, Bo Liu

NeurIPS 2025 NaturalReasoning: Reasoning in the Wild with 2.8m Challenging Questions Weizhe Yuan, Jane Yu, Song Jiang, Karthik Padthe, Yang Li, Dong Wang, Ilia Kulikov, Kyunghyun Cho, Yuandong Tian, Jason E Weston, Xian Li

NeurIPS 2025 SmartCache: Context-Aware Semantic Cache for Efficient Multi-Turn LLM Inference Chengye Yu, Tianyu Wang, Zili Shao, Song Jiang

IJCAI 2024 A Single Vector Is Not Enough: Taxonomy Expansion via Box Embeddings (Extended Abstract) Song Jiang, Qiyue Yao, Qifan Wang, Yizhou Sun

TMLR 2024 On the Equivalence of Graph Convolution and Mixup Xiaotian Han, Hanqing Zeng, Yu Chen, Shaoliang Nie, Jingzhou Liu, Kanika Narang, Zahra Shakeri, Karthik Abinav Sankararaman, Song Jiang, Madian Khabsa, Qifan Wang, Xia Hu

NeurIPSW 2024 Towards Full Delegation: Designing Ideal Agentic Behaviors for Travel Planning Song Jiang, Da Ju, Andrew Cohen, Sasha Mitts, Aaron Foss, Justine T Kao, Xian Li, Yuandong Tian

NeurIPS 2023 CARE: Modeling Interacting Dynamics Under Temporal Environmental Variation Xiao Luo, Haixin Wang, Zijie Huang, Huiyu Jiang, Abhijeet Gangan, Song Jiang, Yizhou Sun

NeurIPSW 2022 Unit Selection: Learning Benefit Function from Finite Population Data Ang Li, Song Jiang, Yizhou Sun, Judea Pearl

NeurIPS 2019 Layer-Dependent Importance Sampling for Training Deep and Large Graph Convolutional Networks Difan Zou, Ziniu Hu, Yewen Wang, Song Jiang, Yizhou Sun, Quanquan Gu