← Search

Zhitao Ying

22 accepted papers

2026

FlatLand: Personalized Graph Federated Learning via Tailored Lorentz Space

ICML 2026oral

Personalization has become a pivotal field of study in contemporary intelligent systems. While large language models (LLMs) excel at general knowledge tasks, they often struggle with personalization, i.e., adapting their outputs to individual user expectations. Existing approaches that steer LLM beh…

Cited by 0SourceScholar
2026

HypRAG: Hyperbolic Dense Retrieval for Retrieval Augmented Generation

ICML 2026poster

Embedding geometry plays a fundamental role in retrieval quality, yet dense retrievers for retrieval-augmented generation (RAG) remain largely confined to Euclidean space. However, natural language exhibits hierarchical structure from broad topics to specific entities that Euclidean embeddings fail …

Cited by 0SourceScholar
2026

Hyperbolic Multimodal Continual Learning

ICML 2026poster

Hyperbolic geometry has recently emerged as a powerful representation space for multimodal learning, as it naturally captures hierarchical semantic structure across modalities. Despite this progress, how such representations behave under continual learning poses fundamentally different challenges th…

Cited by 0SourceScholar
2026

Platonic Transformers: A Solid Choice For Equivariance

ICML 2026poster

While widespread, Transformers lack inductive biases for geometric symmetries common in science and computer vision. Existing equivariant methods often sacrifice the efficiency and flexibility that make Transformers so effective through complex, computationally intensive designs. We introduce the Pl…

Cited by 0SourceScholar
2026

TelecomTS: A Multi-Modal Observability Dataset for Time Series and Language Analysis

ICML 2026poster

Modern enterprises generate vast streams of time series metrics when monitoring complex systems, known as observability data. Unlike conventional time series from domains such as climate, observability data are zero-inflated, highly stochastic, and exhibit minimal temporal structure. Despite their i…

Cited by 0SourceScholar
2026

Variational Learning for Insertion-based Generation

ICML 2026spotlight

Non-monotonic sequence generation methods, such as masked diffusion models, provide a flexible alternative to left-to-right autoregressive modeling by allowing tokens to be generated in non-fixed and prescribed orders. Despite their practical advantages, most existing non-monotonic models are order-…

Cited by 0SourceScholar
2024

THOUGHT PROPAGATION: AN ANALOGICAL APPROACH TO COMPLEX REASONING WITH LARGE LANGUAGE MODELS

ICLR 2024poster

Large Language Models (LLMs) have achieved remarkable success in reasoning tasks with the development of prompting methods. However, existing prompting approaches cannot reuse insights of solving similar problems and suffer from accumulated errors in multi-step reasoning, since they prompt LLMs to…

2023

D4Explainer: In-distribution Explanations of Graph Neural Network via Discrete Denoising Diffusion

NeurIPS 2023poster

The widespread deployment of Graph Neural Networks (GNNs) sparks significant interest in their explainability, which plays a vital role in model auditing and ensuring trustworthy graph learning. The objective of GNN explainability is to discern the underlying graph structures that have the most sign…

2023

FusionRetro: Molecule Representation Fusion via In-Context Learning for Retrosynthetic Planning

ICML 2023poster

Retrosynthetic planning aims to devise a complete multi-step synthetic route from starting materials to a target molecule. Current strategies use a decoupled approach of single-step retrosynthesis models and search algorithms, taking only the product as the input to predict the reactants for each pl…

2023

Hyperbolic Representation Learning: Revisiting and Advancing

ICML 2023poster

The non-Euclidean geometry of hyperbolic spaces has recently garnered considerable attention in the realm of representation learning. Current endeavors in hyperbolic representation largely presuppose that the underlying hierarchies can be automatically inferred and preserved through the adaptive opt…

2023

MuSe-GNN: Learning Unified Gene Representation From Multimodal Biological Graph Data

NeurIPS 2023poster

Discovering genes with similar functions across diverse biomedical contexts poses a significant challenge in gene representation learning due to data heterogeneity. In this study, we resolve this problem by introducing a novel model called Multimodal Similarity Learning Graph Neural Network, which c…

2023

Static and Sequential Malicious Attacks in the Context of Selective Forgetting

NeurIPS 2023poster

With the growing demand for the right to be forgotten, there is an increasing need for machine learning models to forget sensitive data and its impact. To address this, the paradigm of selective forgetting (a.k.a machine unlearning) has been extensively studied, which aims to remove the impact of re…

Cited by 19SourcePDFScholar
2023

TempME: Towards the Explainability of Temporal Graph Neural Networks via Motif Discovery

NeurIPS 2023poster

Temporal graphs are widely used to model dynamic systems with time-varying interactions. In real-world scenarios, the underlying mechanisms of generating future interactions in dynamic systems are typically governed by a set of recurring substructures within the graph, known as temporal motifs. Desp…

2021

Modeling Heterogeneous Hierarchies with Relation-specific Hyperbolic Cones

NeurIPS 2021poster

Hierarchical relations are prevalent and indispensable for organizing human knowledge captured by a knowledge graph (KG). The key property of hierarchical relations is that they induce a partial ordering over the entities, which needs to be modeled in order to allow for hierarchical reasoning. Howev…

2021

Neural Distance Embeddings for Biological Sequences

NeurIPS 2021poster

The development of data-dependent heuristics and representations for biological sequences that reflect their evolutionary distance is critical for large-scale biological research. However, popular machine learning approaches, based on continuous Euclidean spaces, have struggled with the discrete com…

2019

GNNExplainer: Generating Explanations for Graph Neural Networks

NeurIPS 2019poster

Graph Neural Networks (GNNs) are a powerful tool for machine learning on graphs.GNNs combine node feature information with the graph structure by recursively passing neural messages along edges of the input graph. However, incorporating both graph structure and feature information leads to complex…

2018

Graph Convolutional Policy Network for Goal-Directed Molecular Graph Generation

NeurIPS 2018spotlight

Generating novel graph structures that optimize given objectives while obeying some given underlying rules is fundamental for chemistry, biology and social science research. This is especially important in the task of molecular graph generation, whose goal is to discover novel molecules with desired…

2018

Hierarchical Graph Representation Learning with Differentiable Pooling

NeurIPS 2018spotlight

Recently, graph neural networks (GNNs) have revolutionized the field of graph representation learning through effectively learned node embeddings, and achieved state-of-the-art results in tasks such as node classification and link prediction. However, current GNN methods are inherently flat and do n…

Cited by 2077SourcePDFScholar