← Search

Tong Zhao

50 accepted papers

2026

D-GARA: A Dynamic Benchmarking Framework for GUI Agent Robustness in Real-World Anomalies

AAAI 2026technical

Developing intelligent agents capable of operating a wide range of Graphical User Interfaces (GUIs) with human-level proficiency is a key milestone on the path toward Artificial General Intelligence. While most existing datasets and benchmarks for training and evaluating GUI agents are static and id

Cited by 5SourcePDFScholar
2026

DASFL: Dynamic Adaptive Split Federated Learning for Heterogeneous Clients

IJCAI 2026

Split Federated Learning (SFL) has emerged as a pivotal paradigm for privacy-preserving distributed training on resource-constrained edge devices by partitioning neural networks between clients and a server. A critical design choice in SFL is the split layer, which determines the computation distrib

Cited by 0Scholar
2026

Fast3Dcache: Training-free 3D Geometry Synthesis Acceleration

CVPR 2026

Diffusion models have achieved impressive generative quality across modalities like 2D images, videos, and 3D shapes, but their inference remains computationally expensive due to the iterative denoising process. While recent caching-based methods effectively reuse redundant computations to speed up

Cited by 0SourceScholar
2026

Few-Step Diffusion Sampling Through Instance-Aware Discretizations

CVPR 2026

Diffusion and flow matching models generate high-fidelity data by simulating paths defined by Ordinary or Stochastic Differential Equations (ODEs/SDEs), starting from a tractable prior distribution. The probability flow ODE formulation enables the use of advanced numerical solvers to accelerate samp

Cited by 0SourceScholar
2026

Improving Diffusion Generalization with Weak-to-Strong Segmented Guidance

CVPR 2026

Diffusion models generate synthetic images through an iterative refinement process. However, the misalignment between the simulation-free objective and the iterative process often causes accumulated gradient error along the sampling trajectory, which leads to unsatisfactory results and a failure to

Cited by 0SourcecodeScholar
2026

Plain Transformers are Surprisingly Powerful Link Predictors

ICML 2026poster

Link prediction is a core challenge in graph machine learning, demanding models that capture rich and complex topological dependencies. While Graph Neural Networks (GNNs) are the standard solution, state-of-the-art pipelines often rely on explicit structural heuristics or memory-intensive node embed…

Cited by 0SourceScholar
2026

Planning-Guided Diffusion Policy Learning for Contact-Rich Bimanual Object Reorientation

ICRA 2026poster

Contact-rich bimanual manipulation involves precise coordination of two arms to change object states through strategically selected contacts and motions. Due to the inherent complexity of these tasks, acquiring sufficient demonstration data and training policies that generalize to unseen scenarios r…

Cited by 0Scholar
2026

ST-VLM: A Spatial-to-Image Multimodal Spatial-Temporal Prediction Framework with Vision-Language Model

AAAI 2026technical

Spatial-temporal prediction plays a crucial role in various domains, including intelligent transportation and environmental monitoring. Although large language model has shown advantages in long-range dependency modeling and excellent generalization ability for forecasting, it has limited understand

Cited by 0SourcePDFScholar
2026

Taming Video Models for 3D and 4D Generation via Zero-Shot Camera Control

CVPR 2026

Video diffusion models have rich world priors, but their use in spatial tasks is limited by poor control, spatial-temporal inconsistent results, and entangled scene-camera dynamics. Current approaches, such as per-task fine-tuning or post-process warping, often introduce visual artifacts, fail to ge

Cited by 0SourcecodeScholar
2025

A Pre-training Framework for Relational Data with Information-theoretic Principles

NeurIPS 2025poster

Relational databases underpin critical infrastructure across a wide range of domains, yet the design of generalizable pre-training strategies for learning from relational databases remains an open challenge due to task heterogeneity. Specifically, there exist many possible downstream tasks, as tasks…

Cited by 0SourcecodeScholar
2025

CSAHFL:Clustered Semi-Asynchronous Hierarchical Federated Learning for Dual-layer Non-IID in Heterogeneous Edge Computing Networks

IJCAI 2025

Federated Learning (FL) enables collaborative model training across distributed devices without sharing raw data. Hierarchical Federated Learning (HFL) is a new paradigm of FL that leverages the Edge Servers (ESs) layer as an intermediary to perform partial local model aggregation in proximity, redu

Cited by 0SourcePDFScholar
2025

Distilling Parallel Gradients for Fast ODE Solvers of Diffusion Models

ICCV 2025poster

Diffusion models (DMs) have achieved state-of-the-art generative performance but suffer from high sampling latency due to their sequential denoising nature. Existing solver-based acceleration methods often face image quality degradation under a low-latency budget. In this paper, we propose the Ensem…

2025

Haste Makes Waste: A Simple Approach for Scaling Graph Neural Networks

ICML 2025poster

Graph neural networks (GNNs) have demonstrated remarkable success in graph representation learning and various sampling approaches have been proposed to scale GNNs to applications with large-scale graphs. A class of promising GNN training algorithms take advantage of historical embeddings to reduce…

2025

Is Linear Feedback on Smoothed Dynamics Sufficient for Stabilizing Contact-Rich Plans?

ICRA 2025

Designing planners and controllers for contact-rich manipulation is extremely challenging as contact violates the smoothness conditions that many gradient-based controller synthesis tools assume. Contact smoothing approximates a non-smooth system with a smooth one, allowing one to use these synthesi

Cited by 10SourceScholar
2025

Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning

ICML 2025poster

Backward compatible representation learning enables updated models to integrate seamlessly with existing ones, avoiding to reprocess stored data. Despite recent advances, existing compatibility approaches in Euclidean space neglect the uncertainty in the old embedding models and force the new model…

Cited by 0SourcePDFScholar
2025

Mosaic of Modalities: A Comprehensive Benchmark for Multimodal Graph Learning

CVPR 2025poster

Graph machine learning has made significant strides in recent years, yet the integration of visual information with graph structure and its potential for improving performance in downstream tasks remains an underexplored area. To address this critical gap, we introduce the Multimodal Graph Benchmark…

2025

Physics-Driven Data Generation for Contact-Rich Manipulation via Trajectory Optimization

RSS 2025poster

We present a low-cost data generation pipeline that integrates physics-based simulation, human demonstrations, and model-based planning to efficiently generate large-scale, high-quality datasets for contact-rich robotic manipulation tasks. Starting with a small number of embodiment-flexible human de…

Cited by 3PDFScholar
2025

Should We Learn Contact-Rich Manipulation Policies From Sampling-Based Planners?

RA-L 2025

The tremendous success of behavior cloning (BC) in robotic manipulation has been largely confined to tasks where demonstrations can be effectively collected through human teleoperation. However, demonstrations for contact-rich manipulation tasks that require complex coordination of multiple contacts

Cited by 13SourceScholar
2025

V2X-Radar: A Multi-modal Dataset with 4D Radar for Cooperative Perception

NeurIPS 2025spotlight

Modern autonomous vehicle perception systems often struggle with occlusions and limited perception range. Previous studies have demonstrated the effectiveness of cooperative perception in extending the perception range and overcoming occlusions, thereby enhancing the safety of autonomous driving. In…

Cited by 0SourceScholar
2024

A Topological Perspective on Demystifying GNN-Based Link Prediction Performance

ICLR 2024poster

Graph Neural Networks (GNNs) have shown great promise in learning node embeddings for link prediction (LP). While numerous studies improve the overall GNNs' LP performance, none have explored their varying performance across different nodes and the underlying reasons. To this end, we demystify which…

2024

Bias-aware Boolean Matrix Factorization Using Disentangled Representation Learning

UAI 2024poster

Boolean matrix factorization (BMF) has been widely utilized in fields such as recommendation systems, graph learning, text mining, and -omics data analysis. Traditional BMF methods decompose a binary matrix into the Boolean product of two lower-rank Boolean matrices plus homoscedastic random errors.…

2024

How Does Message Passing Improve Collaborative Filtering?

NeurIPS 2024poster

Collaborative filtering (CF) has exhibited prominent results for recommender systems and been broadly utilized for real-world applications. A branch of research enhances CF methods by message passing (MP) used in graph neural networks, due to its strong capabilities of extracting knowledge from grap…

2024

Position: Graph Foundation Models Are Already Here

ICML 2024spotlight

Graph Foundation Models (GFMs) are emerging as a significant research topic in the graph domain, aiming to develop graph models trained on extensive and diverse data to enhance their applicability across various tasks and domains. Developing GFMs presents unique challenges over traditional Graph Neu…

2024

Revisiting Link Prediction: a data perspective

ICLR 2024poster

Link prediction, a fundamental task on graphs, has proven indispensable in various applications, e.g., friend recommendation, protein analysis, and drug interaction prediction. However, since datasets span a multitude of domains, they could have distinct underlying mechanisms of link formation. Evid…

2023

Data-Centric Learning from Unlabeled Graphs with Diffusion Model

NeurIPS 2023poster

Graph property prediction tasks are important and numerous. While each task offers a small size of labeled examples, unlabeled graphs have been collected from various sources and at a large scale. A conventional approach is training a model with the unlabeled graphs on self-supervised tasks and then…

2023

Demystifying Structural Disparity in Graph Neural Networks: Can One Size Fit All?

NeurIPS 2023poster

Recent studies on Graph Neural Networks(GNNs) provide both empirical and theoretical evidence supporting their effectiveness in capturing structural patterns on both homophilic and certain heterophilic graphs. Notably, most real-world homophilic and heterophilic graphs are comprised of a mixture of…

2023

Efficient Deep Learning of Robust, Adaptive Policies using Tube MPC-Guided Data Augmentation

IROS 2023poster

The deployment of agile autonomous systems in challenging, unstructured environments requires adaptation capabilities and robustness to uncertainties. Existing robust and adaptive controllers, such as those based on model predictive control (MPC), can achieve impressive performance at the cost of he…

Cited by 4SourceScholar
2023

Empowering Graph Representation Learning with Test-Time Graph Transformation

ICLR 2023poster

As powerful tools for representation learning on graphs, graph neural networks (GNNs) have facilitated various applications from drug discovery to recommender systems. Nevertheless, the effectiveness of GNNs is immensely challenged by issues related to data quality, such as distribution shift, abnor…

2023

GraphPatcher: Mitigating Degree Bias for Graph Neural Networks via Test-time Augmentation

NeurIPS 2023poster

Recent studies have shown that graph neural networks (GNNs) exhibit strong biases towards the node degree: they usually perform satisfactorily on high-degree nodes with rich neighbor information but struggle with low-degree nodes. Existing works tackle this problem by deriving either designated GNN…

2023

Link Prediction with Non-Contrastive Learning

ICLR 2023poster

Graph neural networks (GNNs) are prominent in the graph machine learning domain, owing to their strong performance across various tasks. A recent focal area is the space of graph self-supervised learning (SSL), which aims to derive useful node representations without labeled data. Notably, many stat…

2023

Linkless Link Prediction via Relational Distillation

ICML 2023poster

Graph Neural Networks (GNNs) have shown exceptional performance in the task of link prediction. Despite their effectiveness, the high latency brought by non-trivial neighborhood data dependency limits GNNs in practical deployments. Conversely, the known efficient MLPs are much less effective than GN…

Cited by 64SourcePDFScholar
2023

MLPInit: Embarrassingly Simple GNN Training Acceleration with MLP Initialization

ICLR 2023poster

Training graph neural networks (GNNs) on large graphs is complex and extremely time consuming. This is attributed to overheads caused by sparse matrix multiplication, which are sidestepped when training multi-layer perceptrons (MLPs) with only node features. MLPs, by ignoring graph context, are simp…

2023

Multi-task Self-supervised Graph Neural Networks Enable Stronger Task Generalization

ICLR 2023poster

Self-supervised learning (SSL) for graph neural networks (GNNs) has attracted increasing attention from the graph machine learning community in recent years, owing to its capability to learn performant node embeddings without costly label information. One weakness of conventional SSL frameworks for…

2023

RLEKF: An Optimizer for Deep Potential with Ab Initio Accuracy

AAAI 2023technical

It is imperative to accelerate the training of neural network force field such as Deep Potential, which usually requires thousands of images based on first-principles calculation and a couple of days to generate an accurate potential energy surface. To this end, we propose a novel optimizer named re…

Cited by 5SourcePDFScholar
2022

Beyond the Limit Automated Driving with Performance Constrained Reachability Analysis

IROS 2022poster

Professional human drivers usually have more than one driving strategy to handle incoming traffic situations. These different strategies activate different performance characteristics of the vehicle, enabling the driver to minimize the risk in a variety of situations by optimizing the strategy selec…

Cited by 1SourceScholar
2022

Bias aware probabilistic Boolean matrix factorization

UAI 2022poster

Boolean matrix factorization (BMF) is a combinatorial problem arising from a wide range of applications including recommendation system, collaborative filtering, and dimensionality reduction. Currently, the noise model of existing BMF methods is often assumed to be homoscedastic; however, in real wo…

2022

Diversifying Content Generation for Commonsense Reasoning with Mixture of Knowledge Graph Experts

ACL 2022findings

Generative commonsense reasoning (GCR) in natural language is to reason about the commonsense while generating coherent text. Recent years have seen a surge of interest in improving the generation quality of commonsense reasoning tasks. Nevertheless, these approaches have seldom investigated diversi…

2022

Grape: Knowledge Graph Enhanced Passage Reader for Open-domain Question Answering

EMNLP 2022finding

A common thread of open-domain question answering (QA) models employs a retriever-reader pipeline that first retrieves a handful of relevant passages from Wikipedia and then peruses the passages to produce an answer. However, even state-of-the-art readers fail to capture the complex relationships be…

2022

Learning Personalized Item-to-Item Recommendation Metric via Implicit Feedback

AISTATS 2022poster

This paper studies the item-to-item recommendation problem in recommender systems from a new perspective of metric learning via implicit feedback. We develop and investigate a personalizable deep metric model that captures both the internal contents of items and how they were interacted with by user…

Cited by 5SourcePDFScholar
2022

Learning from Counterfactual Links for Link Prediction

ICML 2022spotlight

Learning to predict missing links is important for many graph-based applications. Existing methods were designed to learn the association between observed graph structure and existence of link between a pair of nodes. However, the causal relationship between the two variables was largely ignored for…

2022

Retrieval-Augmented Multilingual Keyphrase Generation with Retriever-Generator Iterative Training

NAACL 2022findings

Keyphrase generation is the task of automatically predicting keyphrases given a piece of long text. Despite its recent flourishing, keyphrase generation on non-English languages haven’t been vastly investigated. In this paper, we call attention to a new setting named multilingual keyphrase generatio…

2021

Data Augmentation for Graph Neural Networks

AAAI 2021technical

Data augmentation has been widely used to improve generalizability of machine learning models. However, comparatively little work studies data augmentation for graphs. This is largely due to the complex, non-Euclidean structure of graphs, which limits possible manipulation operations. Augmentation…

2021

End-to-End Conversational Search for Online Shopping with Utterance Transfer

EMNLP 2021main

Successful conversational search systems can present natural, adaptive and interactive shopping experience for online shopping customers. However, building such systems from scratch faces real word challenges from both imperfect product schema/knowledge and lack of training dialog data. In this work…

2021

Graph-based Multilingual Product Retrieval in E-Commerce Search

NAACL 2021industry

Nowadays, with many e-commerce platforms conducting global business, e-commerce search systems are required to handle product retrieval under multilingual scenarios. Moreover, comparing with maintaining per-country specific e-commerce search systems, having an universal system across countries can f…

Cited by 24SourcePDFScholar
2020

Geometric All-way Boolean Tensor Decomposition

NeurIPS 2020poster

Boolean tensor has been broadly utilized in representing high dimensional logical data collected on spatial, temporal and/or other relational domains. Boolean Tensor Decomposition (BTD) factorizes a binary tensor into the Boolean sum of multiple rank-1 tensors, which is an NP-hard problem. Existing…