中文

可转向 CNN 的隐式卷积核

机器学习 2023-10-30 v3 人工智能 计算机视觉与模式识别

摘要

可转向卷积神经网络(CNN)提供了一个通用框架,用于构建对平移及保原点群 GG(如反射和旋转)的变换等变的神经网络。它们依赖于标准卷积与 GG-可转向核,这些核通过解析求解施加于核空间的群特定等变约束获得。由于该解是针对特定群 GG 定制的,实现核基并不能推广到其他对称变换,这给通用群等变模型的开发带来了困难。我们提出使用通过多层感知机(MLP)的隐式神经表示来参数化 GG-可转向核。所得框架提供了一种简单灵活的方式来实现可转向 CNN,并可推广到任何能够构建 GG-等变 MLP 的群 GG。我们在多个任务上证明了方法的有效性,包括 N 体模拟、点云分类和分子性质预测。

关键词

引用

@article{arxiv.2212.06096,
  title  = {Implicit Convolutional Kernels for Steerable CNNs},
  author = {Maksim Zhdanov and Nico Hoffmann and Gabriele Cesa},
  journal= {arXiv preprint arXiv:2212.06096},
  year   = {2023}
}

备注

Accepted to 37th Conference on Neural Information Processing Systems (NeurIPS 2023)