グループ同変畳み込みニューラルネットワーク(G-CNN)を一般化するPDEベースのフレームワークを紹介します。このフレームワークでは、ネットワーク層は、幾何学的に意味のあるPDE係数が層のトレーニング可能な重みになるPDEソルバーのセットと見なされます。等質空間でPDEを作成すると、CNNの標準的な変換同変に加えて、回転などの組み込みの対称性を使用してこれらのネットワークを設計できます。必要なすべての対称性を設計に含めることで、データ拡張などのコストのかかる手法を使用してそれらを含める必要がなくなります。 PDEベースのG-CNN(PDE-G-CNN)について、一般的な等質空間設定で説明すると同時に、主要な関心のあるケースである回転変換同変の詳細についても説明します。線形グループ畳み込みと非線形形態学的グループ畳み込みを、形式的定理で裏付けられた解析的カーネル近似と組み合わせることにより、対象の偏微分方程式を解きます。カーネル近似により、PDEソルバーの高速GPU実装が可能になります。この記事では、https://gitlab.com/bsmetsjr/lietorchで入手可能なPyTorchのLieTorch拡張機能の形式で実装をリリースしています。線形畳み込みの場合と同様に、形態学的畳み込みは、PDE-G-CNNでトレーニングするカーネルによって指定されます。 PDE-G-CNNでは、max / min-poolingやReLUなどの非線形性は、形態学的畳み込みにすでに含まれているため、使用しません。従来のCNNよりもはるかに少ないパラメータで深層学習ベースのイメージングアプリケーションのパフォーマンスを向上させる上で提案されたPDE-G-CNNの強さを実証するための一連の実験を提示します。
We present a PDE-based framework that generalizes Group equivariant Convolutional Neural Networks (G-CNNs). In this framework, a network layer is seen as a set of PDE-solvers where geometrically meaningful PDE-coefficients become the layer's trainable weights. Formulating our PDEs on homogeneous spaces allows these networks to be designed with built-in symmetries such as rotation in addition to the standard translation equivariance of CNNs. Having all the desired symmetries included in the design obviates the need to include them by means of costly techniques such as data augmentation. We will discuss our PDE-based G-CNNs (PDE-G-CNNs) in a general homogeneous space setting while also going into the specifics of our primary case of interest: roto-translation equivariance. We solve the PDE of interest by a combination of linear group convolutions and non-linear morphological group convolutions with analytic kernel approximations that we underpin with formal theorems. Our kernel approximations allow for fast GPU-implementation of the PDE-solvers, we release our implementation with this article in the form of the LieTorch extension to PyTorch, available at https://gitlab.com/bsmetsjr/lietorch . Just like for linear convolution a morphological convolution is specified by a kernel that we train in our PDE-G-CNNs. In PDE-G-CNNs we do not use non-linearities such as max/min-pooling and ReLUs as they are already subsumed by morphological convolutions. We present a set of experiments to demonstrate the strength of the proposed PDE-G-CNNs in increasing the performance of deep learning based imaging applications with far fewer parameters than traditional CNNs.