Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data.

Tongtian Zhu

dblp:323/5165 · DBLP profile ↗
← Back
9ranked-venue papers
3as first author
9since 2021 · last 2025
0000-0002-5503-9290ORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 8 · 3 first-author · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 3 · 3 since 2021Databases, data management, data science and information retrieval · 2 · 2 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
7 papers
Efficient and distributed learning · 26% Optimization for machine learning · 22% Reinforcement learning · 13%
Databases, data mining, and information retrieval
1 paper
Data mining · 100%
Computer architecture, parallel and distributed computing, and storage systems
1 paper
Distributed systems · 100%

Topics — the 22 heaviest of 23, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Efficient and distributed learning
distributed training
2.132025
Efficient Distributed Graph Neural Network Training With Source Chunking and Moving Aggregation · IEEE Trans. Knowl. Data Eng. 2025
Decentralized SGD and Average-direction SAM are Asymptotically Equivalent · ICML 2023
Topology-aware Generalization of Decentralized SGD · ICML 2022
Machine learning › Graph learning
graph neural network
1.522025
Efficient Distributed Graph Neural Network Training With Source Chunking and Moving Aggregation · IEEE Trans. Knowl. Data Eng. 2025
Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalization · KDD 2023
Machine learning › Optimization for machine learning › distributed optimization
decentralized SGD
1.222023
Decentralized SGD and Average-direction SAM are Asymptotically Equivalent · ICML 2023
Topology-aware Generalization of Decentralized SGD · ICML 2022
Machine learning › Learning theory
generalization bounds
1.222023
Decentralized SGD and Average-direction SAM are Asymptotically Equivalent · ICML 2023
Topology-aware Generalization of Decentralized SGD · ICML 2022
Machine learning › Efficient and distributed learning › distributed training › distributed DNN training
distributed graph neural network training
0.912025
Efficient Distributed Graph Neural Network Training With Source Chunking and Moving Aggregation · IEEE Trans. Knowl. Data Eng. 2025
Machine learning › Efficient and distributed learning › federated learning
incentive mechanism
0.912025
DICE: Data Influence Cascade in Decentralized Learning · ICLR 2025
Machine learning › Trustworthy machine learning › Data-centric AI › data influence
influence estimation
0.912025
DICE: Data Influence Cascade in Decentralized Learning · ICLR 2025
Distributed systems › distributed machine learning
decentralized learning
0.912025
DICE: Data Influence Cascade in Decentralized Learning · ICLR 2025
Data mining
anomaly detection
0.812024
SemanticMask: A Contrastive View Design for Anomaly Detection in Tabular Data · IJCAI 2024
Data mining › anomaly detection
tabular anomaly detection
0.812024
SemanticMask: A Contrastive View Design for Anomaly Detection in Tabular Data · IJCAI 2024
Machine learning › Reinforcement learning › multi-agent reinforcement learning
credit assignment
0.712023
Contrastive Identity-Aware Learning for Multi-Agent Value Decomposition · AAAI 2023
Machine learning › Optimization for machine learning
implicit regularization
0.712023
Decentralized SGD and Average-direction SAM are Asymptotically Equivalent · ICML 2023
Machine learning › Reinforcement learning
multi-agent reinforcement learning
0.712023
Contrastive Identity-Aware Learning for Multi-Agent Value Decomposition · AAAI 2023
Machine learning › Deep learning architectures and training
normalization
0.712023
Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalization · KDD 2023
Machine learning › Optimization for machine learning › gradient-based optimization
sharpness-aware minimization
0.712023
Decentralized SGD and Average-direction SAM are Asymptotically Equivalent · ICML 2023
Machine learning › Optimization for machine learning
stochastic gradient descent
0.712023
Lookaround Optimizer: k steps around, 1 step average · NeurIPS 2023
Machine learning › Reinforcement learning › multi-agent reinforcement learning › value-based multi-agent reinforcement learning
value decomposition
0.712023
Contrastive Identity-Aware Learning for Multi-Agent Value Decomposition · AAAI 2023
Machine learning › Deep learning architectures and training
weight averaging
0.712023
Lookaround Optimizer: k steps around, 1 step average · NeurIPS 2023
Machine learning › Learning theory › generalization bounds
algorithmic stability
0.612022
Topology-aware Generalization of Decentralized SGD · ICML 2022
Distributed systems
peer-to-peer systems
0.312025
DICE: Data Influence Cascade in Decentralized Learning · ICLR 2025
Machine learning › Representation and self-supervised learning
contrastive learning
0.212023
Contrastive Identity-Aware Learning for Multi-Agent Value Decomposition · AAAI 2023
Machine learning › Graph learning › graph neural network › deep graph neural network
over-smoothing
0.212023
Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalization · KDD 2023

Methods — techniques the papers use, named apart from their topics

loss landscape curvature · 1.7contrastive learning · 1.4source chunking · 0.9sampling parallelism · 0.9moving aggregation · 0.9influence functions · 0.9influence function · 0.9historical embeddings · 0.9non-convex optimization · 0.7mutual information maximization · 0.7asymptotic equivalence analysis · 0.7
YearPublicationVenuePosition
2025 DICE: Data Influence Cascade in Decentralized Learning
abstract
Decentralized learning offers a promising approach to crowdsource data consumptions and computational workloads across geographically distributed compute interconnected through peer-to-peer networks, accommodating the exponentially increasing demands. However, proper incentives are still in absence, considerably discouraging participation. Our vision is that a fair incentive mechanism relies on fair attribution of contributions to participating nodes, which faces non-trivial challenges arising from the localized connections making influence ``cascade'' in a decentralized network. To overcome this, we design the first method to estimate Data Influence CascadE (DICE) in a decentralized environment. Theoretically, the framework derives tractable approximations of influence cascade over arbitrary neighbor hops, suggesting the influence cascade is determined by an interplay of data, communication topology, and the curvature of loss landscape.DICE also lays the foundations for applications including selecting suitable collaborators and identifying malicious behaviors. Project page is available at https://raiden-zhu.github.io/blog/2025/DICE.
Tongtian Zhu, Can Wang 0001, Fengxiang He
ICLR1
2025 Efficient Distributed Graph Neural Network Training With Source Chunking and Moving Aggregation
abstract
Graph neural networks (GNNs) are effective models for analyzing graph-structured data, but encounter challenges when training on large distributed graphs. Existing GNN training frameworks use sampling parallelism and historical embedding methods to support distributed training and enhance efficiency. However, these methods suffer from issues like stale historical embeddings, imbalanced communication messages, and redundant storage and computation costs. In this paper, we present Emma, a distributed GNN training framework that incorporates source node centric chunking for frequent updates of embeddings and balanced communication, as well as a moving message aggregation technique to boost training efficiency and reduce storage costs. Experimental results show that Emma significantly enhances training efficiency by reducing computation and communication overhead, leading to a notable speedup while maintaining convergence accuracy compared to state-of-the-art distributed GNN training methods.
Tongya Zheng, Rui Wang 0076, Tongtian Zhu, Bingde Hu, Shuibing He, Mingli Song, Xinyu Wang 0001, Sai Wu, Chun Chen 0001
IEEE Trans. Knowl. Data Eng.4
2024 SemanticMask: A Contrastive View Design for Anomaly Detection in Tabular Data
Shuting Tao, Tongtian Zhu, Hongwei Wang 0001, Xiangming Meng
IJCAI2
2023 Contrastive Identity-Aware Learning for Multi-Agent Value Decomposition
abstract
Value Decomposition (VD) aims to deduce the contributions of agents for decentralized policies in the presence of only global rewards, and has recently emerged as a powerful credit assignment paradigm for tackling cooperative Multi-Agent Reinforcement Learning (MARL) problems. One of the main challenges in VD is to promote diverse behaviors among agents, while existing methods directly encourage the diversity of learned agent networks with various strategies. However, we argue that these dedicated designs for agent networks are still limited by the indistinguishable VD network, leading to homogeneous agent behaviors and thus downgrading the cooperation capability. In this paper, we propose a novel Contrastive Identity-Aware learning (CIA) method, explicitly boosting the credit-level distinguishability of the VD network to break the bottleneck of multi-agent diversity. Specifically, our approach leverages contrastive learning to maximize the mutual information between the temporal credits and identity representations of different agents, encouraging the full expressiveness of credit assignment and further the emergence of individualities. The algorithm implementation of the proposed CIA module is simple yet effective that can be readily incorporated into various VD architectures. Experiments on the SMAC benchmarks and across different VD backbones demonstrate that the proposed method yields results superior to the state-of-the-art counterparts. Our code is available at https://github.com/liushunyu/CIA.
Shunyu Liu 0001, Yihe Zhou, Jie Song 0011, Tongya Zheng, Kai-Xuan Chen 0001, Tongtian Zhu, Zunlei Feng, Mingli Song
AAAI6
2023 Adversarial Erasing with Pruned Elements: Towards Better Graph Lottery Tickets
abstract
Graph Lottery Ticket (GLT), a combination of core subgraph and sparse subnetwork, has been proposed to mitigate the computational cost of deep Graph Neural Networks (GNNs) on large input graphs while preserving original performance. However, the winning GLTs in exisiting studies are obtained by applying iterative magnitude-based pruning (IMP) without re-evaluating and re-considering the pruned information, which disregards the dynamic changes in the significance of edges/weights during graph/model structure pruning, and thus limits the appeal of the winning tickets. In this paper, we formulate a conjecture, i.e., existing overlooked valuable information in the pruned graph connections and model parameters which can be re-grouped into GLT to enhance the final performance. Specifically, we propose an adversarial complementary erasing (ACE) framework to explore the valuable information from the pruned components, thereby developing a more powerful GLT, referred to as the ACE-GLT. The main idea is to mine valuable information from pruned edges/weights after each round of IMP, and employ the ACE technique to refine the GLT processing. Finally, experimental results demonstrate that our ACE-GLT outperforms existing methods for searching GLT in diverse tasks. Our code is available at https://github.com/Wangyuwen0627/ACE-GLT.
Shunyu Liu 0001, Kai-Xuan Chen 0001, Tongtian Zhu, Ji Qiao, Mengjie Shi, Yuanyu Wan, Mingli Song
ECAI4
2023 Decentralized SGD and Average-direction SAM are Asymptotically Equivalent
abstract
Decentralized stochastic gradient descent (D-SGD) allows collaborative learning on massive devices simultaneously without the control of a central server. However, existing theories claim that decentralization invariably undermines generalization. In this paper, we challenge the conventional belief and present a completely new perspective for understanding decentralized learning. We prove that D-SGD implicitly minimizes the loss function of an average-direction Sharpness-aware minimization (SAM) algorithm under general non-convex non-$\beta$-smooth settings. This surprising asymptotic equivalence reveals an intrinsic regularization-optimization trade-off and three advantages of decentralization: (1) there exists a free uncertainty evaluation mechanism in D-SGD to improve posterior estimation; (2) D-SGD exhibits a gradient smoothing effect; and (3) the sharpness regularization effect of D-SGD does not decrease as total batch size increases, which justifies the potential generalization benefit of D-SGD over centralized SGD (C-SGD) in large-batch scenarios.
Tongtian Zhu, Fengxiang He, Kai-Xuan Chen 0001, Mingli Song, Dacheng Tao
ICML1
2023 Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalization
abstract
Graph Neural Networks~(GNNs) have emerged as a powerful category of learning architecture for handling graph-structured data. However, existing GNNs typically ignore crucial structural characteristics in node-induced subgraphs, which thus limits their expressiveness for various downstream tasks. In this paper, we strive to strengthen the representative capabilities of GNNs by devising a dedicated plug-and-play normalization scheme, termed as SUbgraph-sPEcific FactoR Embedded Normalization (SuperNorm), that explicitly considers the intra-connection information within each node-induced subgraph. To this end, we embed the subgraph-specific factor at the beginning and the end of the standard BatchNorm, as well as incorporate graph instance-specific statistics for improved distinguishable capabilities. In the meantime, we provide theoretical analysis to support that, with the elaborated SuperNorm, an arbitrary GNN is at least as powerful as the 1-WL test in distinguishing non-isomorphism graphs. Furthermore, the proposed SuperNorm scheme is also demonstrated to alleviate the over-smoothing phenomenon. Experimental results related to predictions of graph, node, and link properties on the eight popular datasets demonstrate the effectiveness of the proposed method. The code is available at https://github.com/chenchkx/SuperNorm.
Kai-Xuan Chen 0001, Shunyu Liu 0001, Tongtian Zhu, Ji Qiao, Yingjie Tian 0002, Tongya Zheng, Haofei Zhang, Zunlei Feng, Jingwen Ye, Mingli Song
KDD3
2023 Lookaround Optimizer: k steps around, 1 step average
abstract
Weight Average (WA) is an active research topic due to its simplicity in ensembling deep networks and the effectiveness in promoting generalization. Existing weight average approaches, however, are often carried out along only one training trajectory in a post-hoc manner (i.e., the weights are averaged after the entire training process is finished), which significantly degrades the diversity between networks and thus impairs the effectiveness. In this paper, inspired by weight average, we propose Lookaround, a straightforward yet effective SGD-based optimizer leading to flatter minima with better generalization. Specifically, Lookaround iterates two steps during the whole training period: the around step and the average step. In each iteration, 1) the around step starts from a common point and trains multiple networks simultaneously, each on transformed data by a different data augmentation, and 2) the average step averages these trained networks to get the averaged network, which serves as the starting point for the next iteration. The around step improves the functionality diversity while the average step guarantees the weight locality of these networks during the whole training, which is essential for WA to work. We theoretically explain the superiority of Lookaround by convergence analysis, and make extensive experiments to evaluate Lookaround on popular benchmarks including CIFAR and ImageNet with both CNNs and ViTs, demonstrating clear superiority over state-of-the-arts. Our code is available at https://github.com/Ardcy/Lookaround.
Shunyu Liu 0001, Jie Song 0011, Tongtian Zhu, Zhengqi Xu, Mingli Song
NeurIPS4
2022 Topology-aware Generalization of Decentralized SGD
abstract
This paper studies the algorithmic stability and generalizability of decentralized stochastic gradient descent (D-SGD). We prove that the consensus model learned by D-SGD is $\mathcal{O}{(m/N\unaryplus1/m\unaryplus\lambda^2)}$-stable in expectation in the non-convex non-smooth setting, where $N$ is the total sample size of the whole system, $m$ is the worker number, and $1\unaryminus\lambda$ is the spectral gap that measures the connectivity of the communication topology. These results then deliver an $\mathcal{O}{(1/N\unaryplus{({(m^{-1}\lambda^2)}^{\frac{\alpha}{2}}\unaryplus m^{\unaryminus\alpha})}/{N^{1\unaryminus\frac{\alpha}{2}}})}$ in-average generalization bound, which is non-vacuous even when $\lambda$ is closed to $1$, in contrast to vacuous as suggested by existing literature on the projected version of D-SGD. Our theory indicates that the generalizability of D-SGD has a positive correlation with the spectral gap, and can explain why consensus control in initial training phase can ensure better generalization. Experiments of VGG-11 and ResNet-18 on CIFAR-10, CIFAR-100 and Tiny-ImageNet justify our theory. To our best knowledge, this is the first work on the topology-aware generalization of vanilla D-SGD. Code is available at \url{https://github.com/Raiden-Zhu/Generalization-of-DSGD}.
Tongtian Zhu, Fengxiang He, Lan Zhang 0002, Zhengyang Niu, Mingli Song, Dacheng Tao
ICML1