ML Anthology

You, Jiaxuan

31 publications

TMLR 2025 AcademicEval: Live Long-Context LLM Benchmark Haozhen Zhang, Tao Feng, Pengrui Han, Jiaxuan You
ICLR 2025 DeFT: Decoding with Flash Tree-Attention for Efficient Tree-Structured LLM Inference Jinwei Yao, Kaiqi Chen, Kexun Zhang, Jiaxuan You, Binhang Yuan, Zeke Wang, Tao Lin
ICML 2025 Graph World Model Tao Feng, Yexin Wu, Guanyu Lin, Jiaxuan You
ICLR 2025 GraphEval: A Lightweight Graph-Based LLM Framework for Idea Evaluation Tao Feng, Yihang Sun, Jiaxuan You
ICLR 2025 GraphRouter: A Graph-Based Router for LLM Selections Tao Feng, Yanzhen Shen, Jiaxuan You
TMLR 2025 PersonalizedRouter: Personalized LLM Routing via Graph-Based User Preference Modeling Zhongjie Dai, Tao Feng, Jiaxuan You
ICML 2025 ResearchTown: Simulator of Human Research Community Haofei Yu, Zhaochen Hong, Zirui Cheng, Kunlun Zhu, Keyang Xuan, Jinwei Yao, Tao Feng, Jiaxuan You
NeurIPS 2025 Router-R1: Teaching LLMs Multi-Round Routing and Aggregation via Reinforcement Learning Haozhen Zhang, Tao Feng, Jiaxuan You
ICLRW 2024 DeFT: Flash Tree-Attention with IO-Awareness for Efficient Tree-Search-Based LLM Inference Jinwei Yao, Kexun Zhang, Kaiqi Chen, Jiaxuan You, Zeke Wang, Binhang Yuan, Tao Lin
ICLR 2024 Deep Reinforcement Learning for Modelling Protein Complexes Ziqi Gao, Tao Feng, Jiaxuan You, Chenyi Zi, Yan Zhou, Chen Zhang, Jia Li
TMLR 2024 How Far Are We from AGI: Are LLMs All We Need? Tao Feng, Chuanyang Jin, Jingyu Liu, Kunlun Zhu, Haoqin Tu, Zirui Cheng, Guanyu Lin, Jiaxuan You
ICMLW 2024 In-Context Learning May Not Elicit Trustworthy Reasoning: A-Not-B Errors in Pretrained Language Models Pengrui Han, Peiyang Song, Haofei Yu, Jiaxuan You
NeurIPS 2024 InstructG2I: Synthesizing Images from Multimodal Attributed Graphs Bowen Jin, Ziqi Pang, Bingjun Guo, Yu-Xiong Wang, Jiaxuan You, Jiawei Han
ICML 2024 Position: Relational Deep Learning - Graph Representation Learning on Relational Databases Matthias Fey, Weihua Hu, Kexin Huang, Jan Eric Lenssen, Rishabh Ranjan, Joshua Robinson, Rex Ying, Jiaxuan You, Jure Leskovec
NeurIPS 2024 RankRAG: Unifying Context Ranking with Retrieval-Augmented Generation in LLMs Yue Yu, Wei Ping, Zihan Liu, Boxin Wang, Jiaxuan You, Chao Zhang, Mohammad Shoeybi, Bryan Catanzaro
ICLRW 2024 Thought-Retriever: Don’t Just Retrieve Raw Data, Retrieve Thoughts Tao Feng, Pengrui Han, Guanyu Lin, Ge Liu, Jiaxuan You
ICLR 2023 AutoTransfer: AutoML with Knowledge Transfer - An Application to Graph Neural Networks Kaidi Cao, Jiaxuan You, Jiaju Liu, Jure Leskovec
NeurIPSW 2022 AutoTransfer: AutoML with Knowledge Transfer - An Application to Graph Neural Networks Kaidi Cao, Jiaxuan You, Jiaju Liu, Jure Leskovec
NeurIPSW 2022 Efficient Automatic Machine Learning via Design Graphs Ying-Xin Wu, Jiaxuan You, Jure Leskovec, Zhitao Ying
ICLR 2022 Relational Multi-Task Learning: Modeling Relations Between Data and Tasks Kaidi Cao, Jiaxuan You, Jure Leskovec
AAAI 2021 Identity-Aware Graph Neural Networks Jiaxuan You, Jonathan Michael Gomes Selman, Rex Ying, Jure Leskovec
NeurIPS 2020 Design Space for Graph Neural Networks Jiaxuan You, Zhitao Ying, Jure Leskovec
ICML 2020 Graph Structure of Neural Networks Jiaxuan You, Jure Leskovec, Kaiming He, Saining Xie
NeurIPS 2020 Handling Missing Data with Graph Representation Learning Jiaxuan You, Xiaobai Ma, Yi Ding, Mykel J Kochenderfer, Jure Leskovec
NeurIPS 2019 G2SAT: Learning to Generate SAT Formulas Jiaxuan You, Haoze Wu, Clark Barrett, Raghuram Ramanujan, Jure Leskovec
NeurIPS 2019 GNNExplainer: Generating Explanations for Graph Neural Networks Zhitao Ying, Dylan Bourgeois, Jiaxuan You, Marinka Zitnik, Jure Leskovec
ICML 2019 Position-Aware Graph Neural Networks Jiaxuan You, Rex Ying, Jure Leskovec
NeurIPS 2018 Graph Convolutional Policy Network for Goal-Directed Molecular Graph Generation Jiaxuan You, Bowen Liu, Zhitao Ying, Vijay Pande, Jure Leskovec
ICML 2018 GraphRNN: Generating Realistic Graphs with Deep Auto-Regressive Models Jiaxuan You, Rex Ying, Xiang Ren, William Hamilton, Jure Leskovec
NeurIPS 2018 Hierarchical Graph Representation Learning with Differentiable Pooling Zhitao Ying, Jiaxuan You, Christopher Morris, Xiang Ren, Will Hamilton, Jure Leskovec
AAAI 2017 Deep Gaussian Process for Crop Yield Prediction Based on Remote Sensing Data Jiaxuan You, Xiaocheng Li, Melvin Low, David B. Lobell, Stefano Ermon