← Search

Ziwei Chai

5 accepted papers

2025

Cypher-RI: Reinforcement Learning for Integrating Schema Selection into Cypher Generation

NeurIPS 2025poster

The increasing utilization of graph databases across various fields stems from their capacity to represent intricate interconnections. Nonetheless, exploiting the full capabilities of graph databases continues to be a significant hurdle, largely because of the inherent difficulty in translating natu…

Cited by 0SourceScholar
2024

An Expert is Worth One Token: Synergizing Multiple Expert LLMs as Generalist via Expert Token Routing

ACL 2024long

We present Expert-Token-Routing, a unified generalist framework that facilitates seamless integration of multiple expert LLMs. Our framework represents expert LLMs as special expert tokens within the vocabulary of a meta LLM. The meta LLM can route to an expert LLM like generating new tokens. Expert…

2024

InfiAgent-DABench: Evaluating Agents on Data Analysis Tasks

ICML 2024poster

In this paper, we introduce InfiAgent-DABench, the first benchmark specifically designed to evaluate LLM-based agents on data analysis tasks. Agents need to solve these tasks end-to-end by interacting with an execution environment. This benchmark contains DAEval, a dataset consisting of 603 data ana…

2023

Towards Learning to Discover Money Laundering Sub-network in Massive Transaction Network

AAAI 2023technical

Anti-money laundering (AML) systems play a critical role in safeguarding global economy. As money laundering is considered as one of the top group crimes, there is a crucial need to discover money laundering sub-network behind a particular money laundering transaction for a robust AML system. Howeve…

Cited by 17SourcePDFScholar
2022

Can Abnormality be Detected by Graph Neural Networks?

IJCAI 2022poster

Anomaly detection in graphs has attracted considerable interests in both academia and industry due to its wide applications in numerous domains ranging from finance to biology. Meanwhile, graph neural networks (GNNs) is emerging as a powerful tool for modeling graph data. A natural and fundamental q…