|
Jiaxi Liu | 刘家熙
I am Jiaxi Liu, a first-year Master's student in the Department of Computer Science at Shanghai Jiao Tong University. Since 2024, I have been a member of ReThinkLab, advised by Prof. Junchi Yan(严骏驰).
I received my Bachelor of Engineering from the Department of Artificial Intelligence at Nanjing University in 2025. During my undergraduate studies, I had the privilege of working with Prof. Yang Yu in the LAMDA Group. I was also fortunate to serve as a visiting researcher at Yale University, collaborating with Prof. Zhuoran Yang. Additionally, I previously worked as an algorithm engineering intern at Tencent, Guangzhou.
My research interests lie in Large Language Models (LLMs), self-evolving systems, agentic RL, and machine learning on graphs. My research goal is to build highly autonomous AI agents by synergizing generative modeling with principled decision-making frameworks, ultimately pushing the boundaries of self-evolving artificial intelligence.
Email /
Google Scholar /
Github
|
|
Publications (* indicates equal contribution)
|
|
|
TodoEvolve: Learning to Architect Agent Planning Systems
Jiaxi Liu*,
Guibin Zhang*, Yanzuo Jiang, Zihan Zhang, Heng Chang, Zhenfei Yin, Qibing Ren, Junchi Yan
ICML 2026
CCF A
PDF | Code
We propose TodoEvolve, a meta-planning paradigm that autonomously synthesizes and dynamically revises task-specific LLM agent planning architectures via Impedance-Guided Preference Optimization (IGPO), consistently outperforming carefully engineered static planning frameworks.
|
|
|
Design Linear Constrained Neural Layers with Implicit Convex Optimization
Junchi Yan,
Jiaxi Liu,
Liangliang Shi, Fangyuan Zhou, Wenzheng Pan, Zhongteng Gui, Yihui Tu
ICML 2026
CCF A
PDF
We introduce LinConLayer, a plug-in differentiable layer for enforcing linear constraints in neural networks through implicit convex optimization. The framework unifies classic constraint layers from an optimization perspective and instantiates two efficient variants, BLCLayer and GLCLayer, enabling stable, hard-constrained prediction for linear programming, partial graph matching, and portfolio allocation.
|
|
|
Towards Practical Large-scale Dynamical Heterogeneous Graph Embedding: Cold-start Resilient Recommendation
Mabiao Long*,
Jiaxi Liu*,
Yufeng Li, Hao Xiong, Junchi Yan, Kefan Wang, Yi Cao, Jiandong Ding
TKDE
CCF A
PDF
We introduce a practical, two-stage framework combining HetSGFormer for scalable static learning and Incremental Locally Linear Embedding (ILLE) for real-time, GPU-free updates, addressing scalability, data freshness, and cold-start challenges in billion-scale recommendation systems.
|
- 2025: Outstanding Graduate, Nanjing University
- 2025: Academic Research Scholarship, School of Artificial Intelligence, Nanjing University
- 2024: Top Outstanding Student Award, Nanjing University (Top 35 University-wide)
- 2024: Zheng Gang Scholarship for Overseas Exchange, Nanjing University
- 2023: Huawei Scholarship
- 2023: Outstanding Volunteer, Nanjing University
- 2022: Outstanding Individual in Social Practice, Jiangsu Province
|
-
2025.09 - Present: Master's student in the Department of Computer Science, Shanghai Jiao Tong University.
-
2021.09 - 2025.06: Undergraduate in the School of Artificial Intelligence, Nanjing University.
-
2018.09 - 2021.06: Guangzhou No.2 High School, Guangdong.
|
-
Travel: I love exploring new places and capturing the beauty of the world. So far, I have visited over 50 cities across China and explored more than 10 countries and regions overseas. 🌍✈️📸
-
Fantasy Literature: I am deeply fascinated by imaginative world-building and epic narratives, with a particular love for The Lord of the Rings and A Song of Ice and Fire series. 🐉📚✨
-
Sports: I enjoy playing football to stay active after a long day of research. I am also a fan of FC Bayern Munich! ⚽🏃♂️🏟️
-
Musical Theater: I am a huge fan of live performances. My absolute favorites are Hamilton and Les Misérables, and I have had the unforgettable experience of watching Hamilton live twice at the Richard Rodgers Theatre in New York. 🎭🎶✨
|
The source code is stolen from Jon Barron. Thanks for his sharing! 🙏🏻
|
|