English
Related papers

Related papers: 3D Steerable CNNs: Learning Rotationally Equivaria…

200 papers

We developed a convolution neural network (CNN) on semi-regular triangulated meshes whose vertices have 6 neighbours. The key blocks of the proposed CNN, including convolution and down-sampling, are directly defined in a vertex domain. By…

Machine Learning · Computer Science 2019-04-16 Caoqiang Liu , Hui Ji , Anqi Qiu

The principle of translation equivariance (if an input image is translated an output image should be translated by the same amount), led to the development of convolutional neural networks that revolutionized machine vision. Other…

Computer Vision and Pattern Recognition · Computer Science 2025-05-29 Zachary Schlamowitz , Andrew Bennecke , Daniel J. Tward

Motivated by objects such as electric fields or fluid streams, we study the problem of learning stochastic fields, i.e. stochastic processes whose samples are fields like those occurring in physics and engineering. Considering general…

Machine Learning · Computer Science 2021-07-20 Peter Holderrieth , Michael Hutchinson , Yee Whye Teh

In this work, we focus on using convolution neural networks (CNN) to perform object recognition on the event data. In object recognition, it is important for a neural network to be robust to the variations of the data during testing. For…

Computer Vision and Pattern Recognition · Computer Science 2019-12-02 Ziyun Wang

Automatic tumor segmentation is a crucial step in medical image analysis for computer-aided diagnosis. Although the existing methods based on convolutional neural networks (CNNs) have achieved the state-of-the-art performance, many…

Image and Video Processing · Electrical Eng. & Systems 2020-05-11 Shuchao Pang , Anan Du , Mehmet A. Orgun , Yan Wang , Quanzheng Sheng , Shoujin Wang , Xiaoshui Huang , Zhemei Yu

Modeling 3D dynamics is a fundamental problem in multi-body systems across scientific and engineering domains and has important practical implications in object trajectory prediction and simulation. While recent GNN-based approaches have…

Machine Learning · Computer Science 2026-05-12 Kai Yang , Yuqi Huang , Junheng Tao , Wanyu Wang , Qitian Wu

Accurate prediction of epileptic seizures allows patients to take preventive measures in advance to avoid possible injuries. In this work, a novel convolutional neural network (CNN) is proposed to analyze time, frequency, and channel…

Machine Learning · Computer Science 2021-05-07 Ziyu Wang , Jie Yang , Mohamad Sawan

Deploying 3D graph neural networks (GNNs) that are equivariant to 3D rotations (the group SO(3)) on edge devices is challenging due to their high computational cost. This paper addresses the problem by compressing and accelerating an…

Machine Learning · Computer Science 2026-03-04 Haoyu Zhou , Ping Xue , Hao Zhang , Tianfan Fu

This paper introduces a new model to learn graph neural networks equivariant to rotations, translations, reflections and permutations called E(n)-Equivariant Graph Neural Networks (EGNNs). In contrast with existing methods, our work does…

Machine Learning · Computer Science 2022-02-17 Victor Garcia Satorras , Emiel Hoogeboom , Max Welling

Incorporating either rotation equivariance or scale equivariance into CNNs has proved to be effective in improving models' generalization performance. However, jointly integrating rotation and scale equivariance into CNNs has not been…

Computer Vision and Pattern Recognition · Computer Science 2023-04-11 Yilong Yang , Srinandan Dasmahapatra , Sasan Mahmoodi

Point cloud registration is a foundational task for 3D alignment and reconstruction applications. While both traditional and learning-based registration approaches have succeeded, leveraging the intrinsic symmetry of point cloud data,…

Computer Vision and Pattern Recognition · Computer Science 2024-10-10 Xueyang Kang , Zhaoliang Luan , Kourosh Khoshelham , Bing Wang

The convolutional neural network (CNN) is one of the most commonly used architectures for computer vision tasks. The key building block of a CNN is the convolutional kernel that aggregates information from the pixel neighborhood and shares…

Image and Video Processing · Electrical Eng. & Systems 2022-02-08 Tianyu Ma , Alan Q. Wang , Adrian V. Dalca , Mert R. Sabuncu

We study how group symmetry helps improve data efficiency and generalization for end-to-end differentiable planning algorithms when symmetry appears in decision-making tasks. Motivated by equivariant convolution networks, we treat the path…

Machine Learning · Computer Science 2023-05-02 Linfeng Zhao , Xupeng Zhu , Lingzhi Kong , Robin Walters , Lawson L. S. Wong

In this paper, we propose a novel formulation to extend CNNs to two-dimensional (2D) manifolds using orthogonal basis functions, called Zernike polynomials. In many areas, geometric features play a key role in understanding scientific…

Computer Vision and Pattern Recognition · Computer Science 2023-05-12 Zhiyu Sun , Ethan Rooke , Jerome Charton , Yusen He , Jia Lu , Stephen Baek

Augmenting transformation knowledge onto a convolutional neural network's weights has often yielded significant improvements in performance. For rotational transformation augmentation, an important element to recent approaches has been the…

Computer Vision and Pattern Recognition · Computer Science 2019-06-11 Rohan Ghosh , Anupam K. Gupta

The state of the art in many computer vision tasks is represented by Convolutional Neural Networks (CNNs). Although their hierarchical organization and local feature extraction are inspired by the structure of primate visual systems, the…

Computer Vision and Pattern Recognition · Computer Science 2019-10-21 Noemi Montobbio , Laurent Bonnasse-Gahot , Giovanna Citti , Alessandro Sarti

We introduce a novel equivariant graph neural network (GNN) architecture designed to predict the tensorial response properties of molecules. Unlike traditional frameworks that focus on regressing scalar quantities and derive tensorial…

Machine Learning · Computer Science 2025-11-11 Jean Philip Filling , Felix Post , Michael Wand , Denis Andrienko

Equivariant diffusion models have achieved impressive performance in 3D molecule generation. These models incorporate Euclidean symmetries of 3D molecules by utilizing an SE(3)-equivariant denoising network. However, specialized equivariant…

Machine Learning · Computer Science 2025-07-01 Yuhui Ding , Thomas Hofmann

Group convolutional neural networks (G-CNNs) have been shown to increase parameter efficiency and model accuracy by incorporating geometric inductive biases. In this work, we investigate the properties of representations learned by regular…

Computer Vision and Pattern Recognition · Computer Science 2022-04-05 David M. Knigge , David W. Romero , Erik J. Bekkers

Like other applications in computer vision, medical image segmentation has been most successfully addressed using deep learning models that rely on the convolution operation as their main building block. Convolutions enjoy important…

Image and Video Processing · Electrical Eng. & Systems 2022-04-05 Davood Karimi , Serge Vasylechko , Ali Gholipour