arXiv reaDer
適応型多相アップサンプリングを使用した真にシフト同変の畳み込みニューラルネットワーク
Truly shift-equivariant convolutional neural networks with adaptive polyphase upsampling
畳み込みニューラルネットワークは、ダウンサンプリング層が存在するため、シフト同変性を欠いています。画像分類では、CNNを完全にシフト不変にするために、適応多相ダウンサンプリング(APS-D)が最近提案されました。ただし、画像再構成タスクに使用されるネットワークでは、それ自体ではシフト同変を復元できません。従来のアップサンプリングの非線形拡張である適応多相アップサンプリング(APS-U)を提案することでこの問題に対処します。これにより、対称エンコーダ-デコーダアーキテクチャ(U-Netなど)を備えたCNNが完全なシフト同変を示すことができます。 MRIおよびCT再構成実験により、APS-D / U層を含むネットワークが、画像再構成の品質を犠牲にすることなく、最先端の同変性能を示すことを示します。さらに、データ拡張やアンチエイリアシングなどの従来の方法とは異なり、APS-D / Uから得られる同変のゲインは、トレーニング分布外の画像にも拡張されます。
Convolutional neural networks lack shift equivariance due to the presence of downsampling layers. In image classification, adaptive polyphase downsampling (APS-D) was recently proposed to make CNNs perfectly shift invariant. However, in networks used for image reconstruction tasks, it can not by itself restore shift equivariance. We address this problem by proposing adaptive polyphase upsampling (APS-U), a non-linear extension of conventional upsampling, which allows CNNs with symmetric encoder-decoder architecture (for example U-Net) to exhibit perfect shift equivariance. With MRI and CT reconstruction experiments, we show that networks containing APS-D/U layers exhibit state of the art equivariance performance without sacrificing on image reconstruction quality. In addition, unlike prior methods like data augmentation and anti-aliasing, the gains in equivariance obtained from APS-D/U also extend to images outside the training distribution.
updated: Mon Dec 06 2021 22:33:18 GMT+0000 (UTC)
published: Sun May 09 2021 22:33:53 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト