English
Related papers

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

200 papers

Various methods to deal with graph data have been proposed in recent years. However, most of these methods focus on graph feature aggregation rather than graph pooling. Besides, the existing top-k selection graph pooling methods have a few…

Social and Information Networks · Computer Science 2020-02-06 Liang Zhang , Xudong Wang , Hongsheng Li , Guangming Zhu , Peiyi Shen , Ping Li , Xiaoyuan Lu , Syed Afaq Ali Shah , Mohammed Bennamoun

Graph neural networks have achieved great success in learning node representations for graph tasks such as node classification and link prediction. Graph representation learning requires graph pooling to obtain graph representations from…

Machine Learning · Computer Science 2020-07-22 Zhengyang Wang , Shuiwang Ji

TorchOptics is an open-source Python library for differentiable Fourier optics simulations, developed using PyTorch to enable GPU-accelerated tensor computations and automatic differentiation. It provides a comprehensive framework for…

Optics · Physics 2024-11-28 Matthew J. Filipovich , A. I. Lvovsky

In graph neural networks (GNNs), pooling operators compute local summaries of input graphs to capture their global properties, and they are fundamental for building deep GNNs that learn hierarchical representations. In this work, we propose…

Machine Learning · Computer Science 2024-04-23 Filippo Maria Bianchi , Daniele Grattarola , Lorenzo Livi , Cesare Alippi

We introduce tf_geometric, an efficient and friendly library for graph deep learning, which is compatible with both TensorFlow 1.x and 2.x. tf_geometric provides kernel libraries for building Graph Neural Networks (GNNs) as well as…

Machine Learning · Computer Science 2021-01-28 Jun Hu , Shengsheng Qian , Quan Fang , Youze Wang , Quan Zhao , Huaiwen Zhang , Changsheng Xu

Graph Neural Networks achieve state-of-the-art performance on a plethora of graph classification tasks, especially due to pooling operators, which aggregate learned node embeddings hierarchically into a final graph representation. However,…

Machine Learning · Computer Science 2022-09-09 Alexandre Duval , Fragkiskos Malliaros

How to utilize deep learning methods for graph classification tasks has attracted considerable research attention in the past few years. Regarding graph classification tasks, the graphs to be classified may have various graph sizes (i.e.,…

Machine Learning · Computer Science 2020-04-16 Yanyan Liang , Yanfeng Zhang , Dechao Gao , Qian Xu

This paper presents TorchNWP, a compilation library tool for the efficient coupling of artificial intelligence components and traditional numerical models. It aims to address the issues of poor cross-language compatibility, insufficient…

Mathematical Software · Computer Science 2026-03-19 Sa Xiao , Hao Jing , Honglu Sun , Haoyu Li

We design and implement a Python library to help the non-expert using all these powerful tools in a way that is efficient, extensible, and simple to incorporate into the workflow of the data scientist, practitioner, and applied researcher.…

Optimization and Control · Mathematics 2022-04-13 Mario Lezcano-Casado

Differentiable sparse linear algebra is foundational for scientific machine learning, yet PyTorch lacks a unified library for it: \texttt{torch.sparse} provides only low-level kernels and a non-differentiable, CPU-only \texttt{spsolve}, and…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-07 Mingyuan Chi , Shizheng Wen

Networks are ubiquitous in many real-world applications (e.g., social networks encoding trust/distrust relationships, correlation networks arising from time series data). While many networks are signed or directed, or both, there is a lack…

Machine Learning · Computer Science 2023-12-04 Yixuan He , Xitong Zhang , Junjie Huang , Benedek Rozemberczki , Mihai Cucuringu , Gesine Reinert

Pooling operations have shown to be effective on computer vision and natural language processing tasks. One challenge of performing pooling operations on graph data is the lack of locality that is not well-defined on graphs. Previous…

Machine Learning · Computer Science 2020-10-21 Hongyang Gao , Yi Liu , Shuiwang Ji

Inspired by the conventional pooling layers in convolutional neural networks, many recent works in the field of graph machine learning have introduced pooling operators to reduce the size of graphs. The great variety in the literature stems…

Machine Learning · Computer Science 2024-03-26 Daniele Grattarola , Daniele Zambon , Filippo Maria Bianchi , Cesare Alippi

The Web Geometry Laboratory (WGL) is a collaborative and adaptive e-learning Web platform integrating a well known dynamic geometry system. Thousands of Geometric problems for Geometric Theorem Provers (TGTP) is a Web-based repository of…

Computers and Society · Computer Science 2018-03-06 Pedro Quaresma , Vanda Santos , Nuno Baeta

Graph neural networks (GNNs) have led to major breakthroughs in a variety of domains such as drug discovery, social network analysis, and travel time estimation. However, they lack interpretability which hinders human trust and thereby…

Machine Learning · Computer Science 2023-12-05 Jonas Jürß , Lucie Charlotte Magister , Pietro Barbiero , Pietro Liò , Nikola Simidjievski

We introduce BN-Pool, the first clustering-based pooling method for Graph Neural Networks that adaptively determines the number of supernodes in a coarsened graph. BN-Pool leverages a generative model based on a Bayesian nonparametric…

Machine Learning · Computer Science 2026-04-02 Daniele Castellana , Filippo Maria Bianchi

Deep Graph Neural Networks (GNNs) are useful models for graph classification and graph-based regression tasks. In these tasks, graph pooling is a critical ingredient by which GNNs adapt to input graphs of varying size and structure. We…

Machine Learning · Computer Science 2020-06-25 Yu Guang Wang , Ming Li , Zheng Ma , Guido Montufar , Xiaosheng Zhuang , Yanan Fan

Following the success of convolution on non-Euclidean space, the corresponding pooling approaches have also been validated on various tasks regarding graphs. However, because of the fixed compression quota and stepwise pooling design, these…

Machine Learning · Computer Science 2022-06-29 Junran Wu , Xueyuan Chen , Ke Xu , Shangzhe Li

Deep learning methods for graphs have seen rapid progress in recent years with much focus awarded to generalising Convolutional Neural Networks (CNN) to graph data. CNNs are typically realised by alternating convolutional and pooling layers…

Machine Learning · Computer Science 2020-06-04 Yaniv Shulman

Graph neural networks (GNN) extends deep learning to graph-structure dataset. Similar to Convolutional Neural Networks (CNN) using on image prediction, convolutional and pooling layers are the foundation to success for GNN on graph…

Machine Learning · Computer Science 2023-02-28 Lingjie Kong , Yun Liao