This package extends Thijs Kuipers' gconv to add support for Approximate/Relaxed Group Equivariant kernels as described in Wang et al. 2022 and Wang et al. 2023
With pip:
pip install git+https://github.com/dgcnz/gconv.gitWith poetry:
poetry add git+https://github.com/dgcnz/gconv.gitThe gconv modules are as straightforward to use as any regular Pytorch convolution module. The only difference is the output consisting of both the feature maps, as well as the group elements on which they are defined. See the example below:
import torch # 1
import gconv.gnn as gnn # 2
# 3
x1 = torch.randn(16, 3, 28, 28, 28) # 4
# 5
lifting_layer = gnn.GLiftingConvSE3(in_channels=3, out_channels=16, kernel_size=5) # 6
gconv_layer = gnn.GSeparableConvSE3(in_channels=16, out_channels=32, kernel_size=5) # 7
# 8
pool = gnn.GAvgGlobalPool() # 9
# 10
x2, H1 = lifting_layer(x1) # 11
x3, H2 = gconv_layer(x2, H1) # 12
# 13
y = pool(x3, H2) # 14In line 5, a random batch of three-channel
Furthermore, gconv offers all the necessary tools to build fully custom group convolutions. All that is required is implementing 5 (or less, depending on the type of convolution) group ops! For more details on how to implement custom group convolutions, see gconv_tutorial.ipynb.