← Search

Yangyi Shen

3 accepted papers

2026

Relational Graph Transformer

ICLR 2026poster

Relational Deep Learning (RDL) is a promising approach for building state-of-the-art predictive models on multi-table relational data by representing it as a heterogeneous temporal graph. However, commonly used Graph Neural Network models suffer from fundamental limitations in capturing complex stru…

Cited by 0SourcecodeScholar
2026

Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Models

ICLR 2026poster

Diffusion large language models (dLLMs) generate text through iterative denoising, yet current decoding strategies discard rich intermediate predictions in favor of the final output. Our work here reveals a critical phenomenon, temporal oscillation, where correct answers often emerge in the middle p…

Cited by 0SourceScholar
2025

Zero-Shot Generalization of GNNs over Distinct Attribute Domains

ICML 2025poster

Traditional Graph Neural Networks (GNNs) cannot generalize to new graphs with node attributes different from the training ones, making zero-shot generalization across different node attribute domains an open challenge in graph machine learning. In this paper, we propose STAGE, which encodes *statis…

Cited by 6SourcePDFScholar