Solid Spherical Energy (SSE) CNNs for Efficient 3D Medical Image Analysis

Invariance to local rotation, to differentiate from the global rotation of images and objects, is required in various texture analysis problems. It has led to several breakthrough methods such as local binary patterns, maximum response and steerable filterbanks. In particular, textures in medical images often exhibit local structures at arbitrary orientations. Locally Rotation Invariant (LRI) Convolutional Neural Networks (CNN) were recently proposed using 3D steerable filters to combine LRI with Directional Sensitivity (DS). The steerability avoids the expensive cost of convolutions with rotated kernels and comes with a parametric representation that results in a drastic reduction of the number of trainable parameters. Yet, the potential bottleneck (memory and computation) of this approach lies in the necessity to recombine responses for a set of predefined discretized orientations. In this paper, we propose to calculate invariants from the responses to the set of spherical harmonics projected onto 3D kernels in the form of a lightweight Solid Spherical Energy (SSE) CNN. It offers a compromise between the high kernel specificity of the LRI-CNN and a low memory/operations requirement. The computational gain is evaluated on 3D synthetic and pulmonary nodule classification experiments. The performance of the proposed approach is compared with steerable LRI-CNNs and standard 3D CNNs, showing competitive results with the state of the art.

[1]  Matti Pietikäinen,et al.  Multiresolution Gray-Scale and Rotation Invariant Texture Classification with Local Binary Patterns , 2002, IEEE Trans. Pattern Anal. Mach. Intell..

[2]  R. Gillies,et al.  Quantitative imaging in cancer evolution and ecology. , 2013, Radiology.

[3]  D. Mattis Quantum Theory of Angular Momentum , 1981 .

[4]  Mitko Veta,et al.  Roto-Translation Covariant Convolutional Networks for Medical Image Analysis , 2018, MICCAI.

[5]  Adrien Depeursinge,et al.  Biomedical Texture Operators and Aggregation Functions: A Methodological Review and User's Guide , 2017 .

[6]  Vincent Andrearczyk,et al.  Rotational 3D Texture Classification Using Group Equivariant CNNs , 2018, ArXiv.

[7]  Stéphane Mallat,et al.  Solid Harmonic Wavelet Scattering: Predicting Quantum Molecular Energy from Invariant Descriptors of 3D Electronic Densities , 2017, NIPS.

[8]  D. Healy,et al.  Computing Fourier Transforms and Convolutions on the 2-Sphere , 1994 .

[9]  Andrew Zisserman,et al.  A Statistical Approach to Texture Classification from Single Images , 2004, International Journal of Computer Vision.

[10]  Max Welling,et al.  3D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Data , 2018, NeurIPS.

[11]  Henning Müller,et al.  3D Solid Texture Classification Using Locally-Oriented Wavelet Transforms , 2017, IEEE Transactions on Image Processing.

[12]  Taco Cohen,et al.  3D G-CNNs for Pulmonary Nodule Detection , 2018, ArXiv.

[13]  Ramakrishna Kakarala,et al.  A theory of phase-sensitive rotation invariance with spherical harmonic and moment-based representations , 2010, 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition.

[14]  Vincent Andrearczyk,et al.  Exploring local rotation invariance in 3D CNNs with steerable filters , 2018, MIDL.

[15]  Stephan J. Garbin,et al.  Harmonic Networks: Deep Translation and Rotation Equivariance , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[16]  Maurice Weiler,et al.  Learning Steerable Filters for Rotation Equivariant CNNs , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[17]  Michael Unser,et al.  Rotation Invariance and Directional Sensitivity: Spherical Harmonics versus Radiomics Features , 2018, MLMI@MICCAI.

[18]  Max Welling,et al.  Group Equivariant Convolutional Networks , 2016, ICML.