Yunchong Song

Researcher, Shanghai Artificial Intelligence Laboratory

Joined

May 2022

Names

Yunchong Song (Preferred)

Emails

****@sjtu.edu.cn (Confirmed)

****@gmail.com (Confirmed)

****@pjlab.org.cn (Confirmed)

Personal Links

Homepage

Google Scholar

Career & Education History

Researcher

Shanghai Artificial Intelligence Laboratory (pjlab.org.cn)

2025 – Present

PhD student

Shanghai Jiao Tong University (sjtu.edu.cn)

2020 – 2025

Undergrad student

Xidian University (xidian.edu.cn)

2016 – 2020

Advisors, Relations & Conflicts

No relations added

Expertise

Machine Learning

Present

Publications

Towards Compressive and Scalable Recurrent Memory
Yunchong Song, Jushi Kai, Liming Lu, Kaixi Qiu, Zhouhan Lin
- Submitted to ICLR 2026
- Readers: Everyone
$\alpha$-GFN: Generalized Mixing in GFlowNets for Better Exploration-Exploitation Trade-off
Lin Chen, Samuel Drapeau, Fanghao Shao, Xuekai Zhu, Bo Xue, Yunchong Song, Mathieu Lauriere, Zhouhan Lin
- Submitted to ICLR 2026
- Readers: Everyone
Flow of Spans: Generalizing Language Models to Dynamic Span-Vocabulary via GFlowNets
Bo Xue, Yunchong Song, Fanghao Shao, Xuekai Zhu, Lin Chen, Luoyi Fu, Xinbing Wang, Zhouhan Lin
- ICLR 2026 Poster
- Readers: Everyone
FlowRL: Matching Reward Distributions for LLM Reasoning
Xuekai Zhu, Daixuan Cheng, Dinghuai Zhang, Hengli Li, Kaiyan Zhang, Che Jiang, Youbang Sun, Ermo Hua, Yuxin Zuo, Xingtai Lv, Qizheng Zhang, Lin Chen, Fanghao Shao, Bo Xue, Yunchong Song, Zhenjie Yang, Ganqu Cui, Ning Ding, Jianfeng Gao, Xiaodong Liu et al. (3 additional authors not shown)
- ICLR 2026 Poster
- Readers: Everyone
Cluster-wise Graph Transformer with Dual-granularity Kernelized Attention
Siyuan Huang, Yunchong Song, Jiayue Zhou, Zhouhan Lin
- NeurIPS 2024 spotlight
- Readers: Everyone
Unlock the Power of Frozen LLMs in Knowledge Graph Completion
Bo Xue, Yi Xu, Yunchong Song, Yiming Pang, Yuyang Ren, Jiaxin Ding, Luoyi Fu, Xinbing Wang
- CoRR 2024
- Readers: Everyone
Graph Parsing Networks
Yunchong Song, Siyuan Huang, Xinbing Wang, Chenghu Zhou, Zhouhan Lin
- ICLR 2024
- Readers: Everyone
GeoGalactica: A Scientific Large Language Model in Geoscience
Zhouhan Lin, Cheng Deng, Le Zhou, Tianhang Zhang, Yi Xu, Yutong Xu, Zhongmou He, Yuanyuan Shi, Beiya Dai, Yunchong Song, Boyi Zeng, Qiyuan Chen, Tao Shi, Tianyu Huang, Yiwei Xu, Shu Wang, Luoyi Fu, Weinan Zhang, Junxian He, Chao Ma et al. (3 additional authors not shown)
- Published: 01 Jan 2024, Last Modified: 29 Jan 2024
- CoRR 2024
- Readers: Everyone
S4G: Breaking the Bottleneck on Graphs with Structured State Spaces
Yunchong Song, Siyuan Huang, Jiacheng Cai, Xinbing Wang, Chenghu Zhou, Zhouhan Lin
- Submitted to ICLR 2024
- Readers: Everyone
Graph Parsing Networks
Yunchong Song, Siyuan Huang, Xinbing Wang, Chenghu Zhou, Zhouhan Lin
- ICLR 2024 poster
- Readers: Everyone

View all 14 publications

Co-Authors

View all 54 co-authors