Bai, Jiaxin

6 publications

ICLR 2026 AtlasKV: Augmenting LLMs with Billion-Scale Knowledge Graphs in 20GB VRAM Haoyu Huang, Hong Ting Tsang, Jiaxin Bai, Xi Peng, Gong Zhang, Yangqiu Song

ICLR 2026 Controllable Logical Hypothesis Generation for Abductive Reasoning in Knowledge Graphs Yisen Gao, Jiaxin Bai, Tianshi Zheng, Ziwei Zhang, Qingyun Sun, Xingcheng Fu, Jianxin Li, Yangqiu Song

ICLR 2026 NewtonBench: Benchmarking Generalizable Scientific Law Discovery in LLM Agents Tianshi Zheng, Kelvin Kiu Wai Tam, Newt Nguyen Kim Hue Nam, Baixuan Xu, Zhaowei Wang, Cheng Jiayang, Hong Ting Tsang, Weiqi Wang, Jiaxin Bai, Tianqing Fang, Yangqiu Song, Ginny Wong, Simon See

TMLR 2025 Learning Federated Neural Graph Databases for Answering Complex Queries from Distributed Knowledge Graphs Qi Hu, Weifeng Jiang, Haoran Li, Zihao Wang, Jiaxin Bai, Qianren Mao, Yangqiu Song, Lixin Fan, Jianxin Li

NeurIPS 2023 Complex Query Answering on Eventuality Knowledge Graph with Implicit Logical Constraints Jiaxin Bai, Xin Liu, Weiqi Wang, Chen Luo, Yangqiu Song

TMLR 2023 Sequential Query Encoding for Complex Query Answering on Knowledge Graphs Jiaxin Bai, Tianshi Zheng, Yangqiu Song