← Search

Yuxin Wu

15 accepted papers

2026

ST-TPP: Learning Semi-Transductive Temporal Point Processes with Gromov-Wasserstein Barycentric Regularization

AAAI 2026technical

The generative mechanisms behind real-world event sequences are often heterogeneous, leading to data that possesses inherent clustering structures. However, most existing temporal point processes (TPPs) treat different event sequences independently, without leveraging the clustering structures when

Cited by 0SourcePDFScholar
2026

Towards Robust Text-Attributed Federated Graph Learning: Multimodal Threats and Defense

AAAI 2026technical

Text-Attributed Graphs (TAGs) are graphs where both nodes and edges are associated with text attributes. To leverage their semantic richness, recent efforts have integrated large language models (LLMs) with graph neural networks, leading to the development of GraphLLMs. However, many real-world data

Cited by 0SourcePDFScholar
2025

A Plug-and-Play Bregman ADMM Module for Inferring Event Branches in Temporal Point Processes

AAAI 2025technical

An event sequence generated by a temporal point process is often associated with a hidden and structured event branching process that captures the triggering relations between its historical and current events. In this study, we design a new plug-and-play module based on the Bregman ADMM (BADMM) al…

2025

GHOST: Generalizable One-Shot Federated Graph Learning with Proxy-Based Topology Knowledge Retention

ICML 2025poster

Federated Graph Learning (FGL) proposes an effective approach to collaboratively training Graph Neural Networks (GNNs) while maintaining privacy. Nevertheless, communication efficiency becomes a critical bottleneck in environments with limited resources. In this context, one-shot FGL emerges as a pr…

2025

HYPERION: Fine-Grained Hypersphere Alignment for Robust Federated Graph Learning

NeurIPS 2025spotlight

Robust Federated Graph Learning (FGL) provides an effective decentralized framework for training Graph Neural Networks (GNNs) in noisy-label environments. However, the subtlety of noise during training presents formidable obstacles for developing robust FGL systems. Previous robust FL approaches nei…

Cited by 0SourceScholar
2025

MOTION: Multi-Sculpt Evolutionary Coarsening for Federated Continual Graph Learning

NeurIPS 2025poster

Graph neural networks (GNNs) have achieved remarkable success in various domains but typically rely on centralized, static graphs, which limits their applicability in distributed, evolving environments. To address this limitation, we define the task of Federated Continual Graph Learning (FCGL), a pa…

Cited by 0SourceScholar
2025

MoBA: Mixture of Block Attention for Long-Context LLMs

NeurIPS 2025spotlight

Scaling the effective context length is essential for advancing large language models (LLMs) toward artificial general intelligence (AGI). However, the quadratic increase in computational complexity inherent in traditional attention mechanisms presents a prohibitive overhead. Existing approaches eit…

Cited by 0SourcecodeScholar
2020

Momentum Contrast for Unsupervised Visual Representation Learning

CVPR 2020oral

We present Momentum Contrast (MoCo) for unsupervised visual representation learning. From a perspective on contrastive learning as dictionary look-up, we build a dynamic dictionary with a queue and a moving-averaged encoder. This enables building a large and consistent dictionary on-the-fly that fac…

Cited by 15734PDFcodeScholar
2019

Bayesian Relational Memory for Semantic Visual Navigation

ICCV 2019poster

We introduce a new memory architecture, Bayesian Relational Memory (BRM), to improve the generalization ability for semantic visual navigation agents in unseen environments, where an agent is given a semantic target to navigate towards. BRM takes the form of a probabilistic relation graph over seman…

Cited by 122PDFcodeScholar
2019

Feature Denoising for Improving Adversarial Robustness

CVPR 2019poster

Adversarial attacks to image classification systems present challenges to convolutional networks and opportunities for understanding them. This study suggests that adversarial perturbations on images lead to noise in the features constructed by these networks. Motivated by this observation, we devel…

Cited by 1130PDFcodeScholar
2018

Building Generalizable Agents with a Realistic and Rich 3D Environment

ICLR 2018workshop

Teaching an agent to navigate in an unseen 3D environment is a challenging task, even in the event of simulated environments. To generalize to unseen environments, an agent needs to be robust to low-level variations (e.g. color, texture, object changes), and also high-level variations (e.g. layout c…

Cited by 279SourcecodeScholar
2018

Group Normalization

ECCV 2018poster

Batch Normalization (BN) is a milestone technique in the development of deep learning, enabling various networks to train. However, normalizing along the batch dimension introduces problems --- BN's error increases rapidly when the batch size becomes smaller, caused by inaccurate batch statistics es…

2017

ELF: An Extensive, Lightweight and Flexible Research Platform for Real-time Strategy Games

NeurIPS 2017oral

In this paper, we propose ELF, an Extensive, Lightweight and Flexible platform for fundamental reinforcement learning research. Using ELF, we implement a highly customizable real-time strategy (RTS) engine with three game environments (Mini-RTS, Capture the Flag and Tower Defense). Mini-RTS, as a mi…