activity
20212026
most citedMagNet: A Neural Network for Directed Graphs

48 citations · 57 across the 7 of their papers we have counts for

collaborators

7 papers

cs.LG2026

How Far Can Sharpness and Complexity Jointly Explain Generalization?

Ziyu Cheng, Xitong Zhang, Longxiu Huang +1

Large-scale empirical studies have revealed that many widely used generalization measures correlate only weakly with generalization, suggesting that relying on a single factor is i…

cs.CV2024★ 1 cited

Optimal Eye Surgeon: Finding Image Priors through Sparse Generators at Initialization

Avrajit Ghosh, Xitong Zhang, Kenneth K. Sun +3

We introduce Optimal Eye Surgeon (OES), a framework for pruning and training deep image generator networks. Typically, untrained deep convolutional networks, which include image sa…

cs.LG2024

Structure-Preserving Network Compression Via Low-Rank Induced Training Through Linear Layers Composition

Xitong Zhang, Ismail R. Alkhouri, Rongrong Wang

Deep Neural Networks (DNNs) have achieved remarkable success in addressing many previously unsolvable tasks. However, the storage and computational requirements associated with DNN…

cs.LG2023★ 1 cited

Harnessing Trust in Directed Graphs: Redefining Robustness of Graph Learning

Zhichao Hou, Xitong Zhang, Wei Wang +2

Existing research on robust Graph Neural Networks (GNNs) focuses predominantly on undirected graphs, overlooking the trustworthiness inherent in the link directions of directed gra…

cs.LG2023★ 1 cited

Implicit regularization in Heavy-ball momentum accelerated stochastic gradient descent

Avrajit Ghosh, He Lyu, Xitong Zhang +1

It is well known that the finite step-size () in Gradient Descent (GD) implicitly regularizes solutions to flatter minima. A natural question to ask is "Does the momentum parame…

cs.LG2022★ 6 cited

PyTorch Geometric Signed Directed: A Software Package on Graph Neural Networks for Signed and Directed Graphs

Yixuan He, Xitong Zhang, Junjie Huang +3

Networks are ubiquitous in many real-world applications (e.g., social networks encoding trust/distrust relationships, correlation networks arising from time series data). While man…

cs.LG2021★ 48 cited

MagNet: A Neural Network for Directed Graphs

Xitong Zhang, Yixuan He, Nathan Brugnone +2

The prevalence of graph-based data has spurred the rapid development of graph neural networks (GNNs) and related machine learning algorithms. Yet, despite the many datasets natural…