English
Related papers

Related papers: Torch Geometric Pool: the PyTorch library for pool…

200 papers

Graph Neural Networks (GNNs) have shown advantages in various graph-based applications. Most existing GNNs assume strong homophily of graph structure and apply permutation-invariant local aggregation of neighbors to learn a representation…

Machine Learning · Computer Science 2022-01-04 Tianmeng Yang , Yujing Wang , Zhihan Yue , Yaming Yang , Yunhai Tong , Jing Bai

Deep graph clustering has recently received significant attention due to its ability to enhance the representation learning capabilities of models in unsupervised scenarios. Nevertheless, deep clustering for temporal graphs, which could…

Machine Learning · Computer Science 2024-04-12 Meng Liu , Yue Liu , Ke Liang , Wenxuan Tu , Siwei Wang , Sihang Zhou , Xinwang Liu

We consider the graph link prediction task, which is a classic graph analytical problem with many real-world applications. With the advances of deep learning, current link prediction methods commonly compute features from subgraphs centered…

Machine Learning · Computer Science 2020-10-21 Lei Cai , Jundong Li , Jie Wang , Shuiwang Ji

Graph Neural Networks (GNNs) have recently caught great attention and achieved significant progress in graph-level applications. In this paper, we propose a framework for graph neural networks with multiresolution Haar-like wavelets, or…

Machine Learning · Computer Science 2021-01-26 Xuebin Zheng , Bingxin Zhou , Ming Li , Yu Guang Wang , Junbin Gao

We introduce torchbearer, a model fitting library for pytorch aimed at researchers working on deep learning or differentiable programming. The torchbearer library provides a high level metric and callback API that can be used for a wide…

Machine Learning · Computer Science 2018-09-11 Ethan Harris , Matthew Painter , Jonathon Hare

As Graph Neural Networks (GNNs) become popular, libraries like PyTorch-Geometric (PyG) and Deep Graph Library (DGL) are proposed; these libraries have emerged as the de facto standard for implementing GNNs because they provide…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-02-29 Yi-Chien Lin , Yuyang Chen , Sameh Gobriel , Nilesh Jain , Gopi Krishna Jha , Viktor Prasanna

Graph Convolutional Networks (GCNs) can capture non-Euclidean spatial dependence between different brain regions. The graph pooling operator, a crucial element of GCNs, enhances the representation learning capability and facilitates the…

Computer Vision and Pattern Recognition · Computer Science 2024-03-01 Cheng Zhu , Jiayi Zhu , Xi Wu , Lijuan Zhang , Shuqi Yang , Ping Liang , Honghan Chen , Ying Tan

The problem of accurately measuring the similarity between graphs is at the core of many applications in a variety of disciplines. Graph kernels have recently emerged as a promising approach to this problem. There are now many kernels, each…

The $\texttt{torch-choice}$ is an open-source library for flexible, fast choice modeling with Python and PyTorch. $\texttt{torch-choice}$ provides a $\texttt{ChoiceDataset}$ data structure to manage databases flexibly and…

Machine Learning · Computer Science 2025-06-05 Tianyu Du , Ayush Kanodia , Susan Athey

Numerous approaches have been explored for graph clustering, including those which optimize a global criteria such as modularity. More recently, Graph Neural Networks (GNNs), which have produced state-of-the-art results in graph analysis…

Social and Information Networks · Computer Science 2023-08-21 Co Tran , Mo Badawy , Tyler McDonnell

Sampling graphs is an important task in data mining. In this paper, we describe Little Ball of Fur a Python library that includes more than twenty graph sampling algorithms. Our goal is to make node, edge, and exploration-based network…

Social and Information Networks · Computer Science 2020-08-12 Benedek Rozemberczki , Oliver Kiss , Rik Sarkar

Graph neural networks achieve high accuracy in link prediction by jointly leveraging graph topology and node attributes. Topology, however, is represented indirectly; state-of-the-art methods based on subgraph classification label nodes…

Machine Learning · Computer Science 2022-03-17 Liming Pan , Cheng Shi , Ivan Dokmanić

Graph pooling compresses graphs and summarises their topological properties and features in a vectorial representation. It is an essential part of deep graph representation learning and is indispensable in graph-level tasks like…

Machine Learning · Computer Science 2025-05-16 Jan von Pichowski , Christopher Blöcker , Ingo Scholtes

Graph neural networks are a versatile machine learning architecture that received a lot of attention recently. In this technical report, we present an implementation of convolution and pooling layers for TensorFlow-Keras models, which…

Machine Learning · Computer Science 2023-10-12 Patrick Reiser , Andre Eberhard , Pascal Friederich

Graph neural networks (GNNs) extends the functionality of traditional neural networks to graph-structured data. Similar to CNNs, an optimized design of graph convolution and pooling is key to success. Borrowing ideas from physics, we…

Machine Learning · Computer Science 2022-01-12 Zheng Ma , Junyu Xuan , Yu Guang Wang , Ming Li , Pietro Lio

Graph neural networks (GNNs) have been widely used to learn vector representation of graph-structured data and achieved better task performance than conventional methods. The foundation of GNNs is the message passing procedure, which…

Machine Learning · Computer Science 2022-01-31 Takeshi D. Itoh , Takatomi Kubo , Kazushi Ikeda

The constant introduction of standardized benchmarks in the literature has helped accelerating the recent advances in meta-learning research. They offer a way to get a fair comparison between different algorithms, and the wide range of…

Machine Learning · Computer Science 2019-09-17 Tristan Deleu , Tobias Würfl , Mandana Samiei , Joseph Paul Cohen , Yoshua Bengio

This paper introduces torchsom, an open-source Python library that provides a reference implementation of the Self-Organizing Map (SOM) in PyTorch. This package offers three main features: (i) dimensionality reduction, (ii) clustering, and…

Machine Learning · Statistics 2025-10-14 Louis Berthier , Ahmed Shokry , Maxime Moreaud , Guillaume Ramelet , Eric Moulines

We introduce pygrank, an open source Python package to define, run and evaluate node ranking algorithms. We provide object-oriented and extensively unit-tested algorithm components, such as graph filters, post-processors, measures,…

Machine Learning · Computer Science 2021-10-19 Emmanouil Krasanakis , Symeon Papadopoulos , Ioannis Kompatsiaris , Andreas Symeonidis

Rule-based fine-grained IP geolocation methods are hard to generalize in computer networks which do not follow hypothetical rules. Recently, deep learning methods, like multi-layer perceptron (MLP), are tried to increase generalization…

Machine Learning · Computer Science 2023-04-18 Shichang Ding , Xiangyang Luo , Jinwei Wang , Xiaoming Fu