🔬 Research Overview
My research centers around the Complete Lifecycle of Knowledge in AI systems, spanning three interconnected phases:
🏗️ Knowledge Graph Construction & Schema Induction
AutoSchemaKG: Autonomous Knowledge Graph Construction through Dynamic Schema Induction from Web-Scale Corpora
Jiaxin Bai, Wei Fan, Qi Hu, Qing Zong, Chunyang Li, Hong Ting Tsang, Hongyu Luo, Yauwai Yim, Haoyu Huang, Xiao Zhou, Feng Qin, Tianshi Zheng, Xi Peng, Xin Yao, Huiwen Yang, Leijie Wu, Yi Ji, Gong Zhang, Renhai Chen, Yangqiu Song
Arxiv [paper][code]
Intention Knowledge Graph Construction for User Intention Relation Modeling
Jiaxin Bai, Zhaobo Wang, Junfei Cheng, Dan Yu, Zerui Huang, Weiqi Wang, Xin Liu, Chen Luo, Yanming Zhu, Bo Li, Yangqiu Song
Arxiv [paper]
FolkScope: Intention Knowledge Graph Construction for E-commerce Commonsense Discovery
Changlong Yu, Weiqi Wang, Xin Liu, Jiaxin Bai, Yangqiu Song, Zheng Li, Yifan Gao, Tianyu Cao, Bing Yin
ACL-2023 (Findings) [paper] [code]
🧠 Knowledge Graph Reasoning & Neural Graph Databases
Top Ten Challenges Towards Agentic Neural Graph Databases
Jiaxin Bai, Zihao Wang, Yukun Zhou, Hang Yin, Weizhi Fei, Qi Hu, Zheye Deng, Jiayang Cheng, Tianshi Zheng, Hong Ting Tsang, Yisen Gao, Zhongwei Xie, Yufei Li, Lixin Fan, Binhang Yuan, Wei Wang, Lei Chen, Xiaofang Zhou, Yangqiu Song
IEEE Data Eng. Bulletin March 2025 issue [paper]
Federated Neural Graph Databases
Qi Hu, Weifeng Jiang, Haoran Li, Zihao Wang, Jiaxin Bai, Qianren Mao, Yangqiu Song, Lixin Fan, Jianxin Li
KDD-2024 [paper][code]
Enhancing Transformers for Generalizable First-Order Logical Entailment
Tianshi Zheng, Jiazheng Wang, Zihao Wang, Jiaxin Bai, Hang Yin, Zheye Deng, Yangqiu Song, Jianxin Li
ACL-2025 [paper][code]
Understanding Inter-Session Intentions via Complex Logical Reasoning
Jiaxin Bai, Chen Luo, Zheng Li, Qingyu Yin, Yangqiu Song
KDD-2024 [paper] [code]
Privacy-Preserved Neural Graph Databases
Qi Hu, Haoran Li, Jiaxin Bai, Zihao Wang, Yangqiu Song
KDD-2024 [paper][code]
Advancing Abductive Reasoning in Knowledge Graphs through Complex Logical Hypothesis Generation
Jiaxin Bai*, Yicheng Wang*, Tianshi Zheng, Yue Guo, Xin Liu, Yangqiu Song
ACL-2024 [paper] [code]
Complex Query Answering on Eventuality Knowledge Graph with Implicit Logical Constraints
Jiaxin Bai, Xin Liu, Weiqi Wang, Chen Luo, Yangqiu Song
NeurIPS-2023 [paper] [code]
Knowledge Graph Reasoning over Entities and Numerical Values
Jiaxin Bai, Chen Luo, Zheng Li, Qingyu Yin, Bing Yin, Yangqiu Song
KDD-2023 [paper] [code]
Sequential Query Encoding For Complex Query Answering on Knowledge Graphs
Jiaxin Bai*, Tianshi Zheng*, Yangqiu Song
Transactions of Machine Learning Research [paper] [code]
Query2Particles: Knowledge Graph Reasoning with Particle Embeddings
Jiaxin Bai, Zihao Wang, Hongming Zhang, Yangqiu Song
NAACL-2022 (Findings) [paper] [code]
🤖 Retrieval-Augmented Generation (RAG) & Language Model Applications
ZEN: Pre-training Chinese Text Encoder Enhanced by N-gram Representations
Shizhe Diao, Jiaxin Bai, Yan Song, Tong Zhang, Yonggang Wang
EMNLP-2020 (Findings) [paper] [code]
Assessing the Robustness of Retrieval-Augmented Generation Systems in K-12 Educational Question Answering with Knowledge Discrepancies
Tianshi Zheng, Weihan Li, Jiaxin Bai, Weiqi Wang, Yangqiu Song
ACL-2025 [paper][code]
CANDLE: Iterative Conceptualization and Instantiation Distillation from Large Language Models for Commonsense Reasoning
Weiqi Wang, Tianqing Fang, Chunyang Li, Haochen Shi, Wenxuan Ding, Baixuan Xu, Zhaowei Wang, Jiaxin Bai, Xin Liu, Jiayang Cheng, Chunkit Chan, Yangqiu Song
ACL-2024 [paper] [code]
EcomScriptBench: A Multi-task Benchmark for E-commerce Script Planning via Step-wise Intention-Driven Product Association
Weiqi Wang, Limeng Cui, Xin Liu, Sreyashi Nag, Wenju Xu, Chen Luo, Sheikh Muhammad Sarwar, Yang Li, Hansu Gu, Hui Liu, Changlong Yu, Jiaxin Bai, Yifan Gao, Haiyang Zhang, Qi He, Shuiwang Ji, Yangqiu Song
ACL-2025 [paper][code]
IntentionQA: A Benchmark for Evaluating Purchase Intention Comprehension Abilities of Language Models in E-commerce
Wenxuan Ding, Weiqi Wang, Sze Heng Douglas Kwok, Minghao Liu, Tianqing Fang, Jiaxin Bai, Xin Liu, Changlong Yu, Zheng Li, Chen Luo, Qingyu Yin, Bing Yin, Junxian He, Yangqiu Song
EMNLP-2024 (Findings) [paper][code]
MIND: Multimodal Shopping Intention Distillation from Large Vision-language Models for E-commerce Purchase Understanding
Baixuan Xu, Weiqi Wang, Haochen Shi, Wenxuan Ding, Huihao Jing, Tianqing Fang, Jiaxin Bai, Xin Liu, Changlong Yu, Zheng Li, Chen Luo, Qingyu Yin, Bing Yin, Long Chen, Yangqiu Song
EMNLP-2024 [paper][code]