4 citations · 6 across the 13 of their papers we have counts for
9 papers · 1 filter
Textual Belief States for World Models: Identifiable Representation Learning Under Strict Mediation
Xiang Gao, Kaiwen Dong, Yuguang Yao +2
World models in partially observed environments rely on latent representations that summarize interaction history, but in many modern LLM-based architectures predictive performance…
Goal-Conditioned Supervised Learning for LLM Fine-Tuning
Shijun Li, Kaiwen Dong, Xiang Gao +1
Large language models often require fine-tuning to better align their behavior with user intent at deployment. Existing approaches are commonly divided into online and offline para…
CORE: Data Augmentation for Link Prediction via Information Bottleneck
Kaiwen Dong, Zhichun Guo, Nitesh V. Chawla
Link prediction (LP) is a fundamental task in graph representation learning, with numerous applications in diverse domains. However, the generalizability of LP models is often comp…
You do not have to train Graph Neural Networks at all on text-attributed graphs
Kaiwen Dong, Zhichun Guo, Nitesh V. Chawla
Graph structured data, specifically text-attributed graphs (TAG), effectively represent relationships among varied entities. Such graphs are essential for semi-supervised node clas…
Universal Link Predictor By In-Context Learning on Graphs
Kaiwen Dong, Haitao Mao, Zhichun Guo +1
Link prediction is a crucial task in graph machine learning, where the goal is to infer missing or future links within a graph. Traditional approaches leverage heuristic methods ba…
Node Duplication Improves Cold-start Link Prediction
Zhichun Guo, Tong Zhao, Yozen Liu +5
Graph Neural Networks (GNNs) are prominent in graph machine learning and have shown state-of-the-art performance in Link Prediction (LP) tasks. Nonetheless, recent studies show tha…
Pure Message Passing Can Estimate Common Neighbor for Link Prediction
Kaiwen Dong, Zhichun Guo, Nitesh V. Chawla
Message Passing Neural Networks (MPNNs) have emerged as the {\em de facto} standard in graph representation learning. However, when it comes to link prediction, they often struggle…
FakeEdge: Alleviate Dataset Shift in Link Prediction
Kaiwen Dong, Yijun Tian, Zhichun Guo +2
Link prediction is a crucial problem in graph-structured data. Due to the recent success of graph neural networks (GNNs), a variety of GNN-based models were proposed to tackle the…
Heterogeneous Graph Masked Autoencoders
Yijun Tian, Kaiwen Dong, Chunhui Zhang +2
Generative self-supervised learning (SSL), especially masked autoencoders, has become one of the most exciting learning paradigms and has shown great potential in handling graph da…