3D Solid Spherical Bispectrum CNNs for Biomedical Texture Analysis

Locally Rotation Invariant (LRI) operators have shown great potential in biomedical texture analysis where patterns appear at random positions and orientations. LRI operators can be obtained by computing the responses to the discrete rotation of local descriptors, such as Local Binary Patterns (LBP) or the Scale Invariant Feature Transform (SIFT). Other strategies achieve this invariance using Laplacian of Gaussian or steerable wavelets for instance, preventing the introduction of sampling errors during the discretization of the rotations. In this work, we obtain LRI operators via the local projection of the image on the spherical harmonics basis, followed by the computation of the bispectrum, which shares and extends the invariance properties of the spectrum. We investigate the benefits of using the bispectrum over the spectrum in the design of a LRI layer embedded in a shallow Convolutional Neural Network (CNN) for 3D image analysis. The performance of each design is evaluated on two datasets and compared against a standard 3D CNN. The first dataset is made of 3D volumes composed of synthetically generated rotated patterns, while the second contains malignant and benign pulmonary nodules in Computed Tomography (CT) images. The results indicate that bispectrum CNNs allows for a significantly better characterization of 3D textures than both the spectral and standard CNN. In addition, it can efficiently learn with fewer training examples and trainable parameters when compared to a standard convolutional layer.

[1]  Wanqing Li,et al.  Viewpoint invariants from three-dimensional data: The role of reflection in human activity understanding , 2011, CVPR 2011 WORKSHOPS.

[2]  Gabriel J. Brostow,et al.  CubeNet: Equivariance to 3D Rotation and Translation , 2018, ECCV.

[3]  P. Lockhart,et al.  Introduction to Geometry , 1940, The Mathematical Gazette.

[4]  M. Chaichian,et al.  Symmetries in quantum mechanics : from angular momentum to supersymmetry , 1998 .

[5]  Risi Kondor,et al.  On the Generalization of Equivariance and Convolution in Neural Networks to the Action of Compact Groups , 2018, ICML.

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

[7]  Taco S Cohen,et al.  Pulmonary nodule detection in CT scans with equivariant CNNs , 2019, Medical Image Anal..

[8]  Ronald M. Summers,et al.  Deep Learning in Medical Imaging: Overview and Future Promise of an Exciting New Technique , 2016 .

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

[10]  Edward H. Adelson,et al.  The Design and Use of Steerable Filters , 1991, IEEE Trans. Pattern Anal. Mach. Intell..

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

[12]  Eero P. Simoncelli,et al.  A Parametric Texture Model Based on Joint Statistics of Complex Wavelet Coefficients , 2000, International Journal of Computer Vision.

[13]  Vincent Andrearczyk,et al.  Local Rotation Invariance in 3D CNNs , 2020, Medical Image Anal..

[14]  Michael Unser,et al.  Principled Design and Implementation of Steerable Detectors , 2018, IEEE Transactions on Image Processing.

[15]  J. Flusser,et al.  Moments and Moment Invariants in Pattern Recognition , 2009 .

[16]  Jian Sun,et al.  Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

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

[18]  Rachid Deriche,et al.  A computational Framework for generating rotation invariant features and its application in diffusion MRI , 2019, Medical Image Anal..

[19]  Jocelyn Quaintance,et al.  Spherical Harmonics and Linear Representations of Lie Groups , 2009 .

[20]  Zhen Lin,et al.  Clebsch-Gordan Nets: a Fully Fourier Space Spherical Convolutional Neural Network , 2018, NeurIPS.

[21]  Ramakrishna Kakarala The Bispectrum as a Source of Phase-Sensitive Invariants for Fourier Descriptors: A Group-Theoretic Approach , 2012, Journal of Mathematical Imaging and Vision.

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

[23]  Taghi M. Khoshgoftaar,et al.  A survey on Image Data Augmentation for Deep Learning , 2019, Journal of Big Data.

[24]  Vincent Andrearczyk,et al.  Solid Spherical Energy (SSE) CNNs for Efficient 3D Medical Image Analysis , 2019 .

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

[26]  Yi Li,et al.  Is Rotation a Nuisance in Shape Recognition? , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

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

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

[29]  Michael Unser,et al.  Steerable Wavelet Machines (SWM): Learning Moving Frames for Texture Classification , 2017, IEEE Transactions on Image Processing.

[30]  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.

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

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

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

[34]  Max Welling,et al.  Spherical CNNs , 2018, ICLR.

[35]  J. Delft,et al.  A numerical algorithm for the explicit calculation of SU(N) and SL(N,C) Clebsch-Gordan coefficients , 2010, 1009.0437.

[36]  Pietro Perona Steerable-scalable kernels for edge detection and junction analysis , 1992, Image Vis. Comput..

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

[38]  Max Welling,et al.  Steerable CNNs , 2016, ICLR.

[39]  Michael Unser,et al.  A Unifying Parametric Framework for 2D Steerable Wavelet Transforms , 2013, SIAM J. Imaging Sci..

[40]  Maurice Weiler,et al.  A General Theory of Equivariant CNNs on Homogeneous Spaces , 2018, NeurIPS.

[41]  Steven W. Smith,et al.  The Scientist and Engineer's Guide to Digital Signal Processing , 1997 .

[42]  Erik J Bekkers B-Spline CNNs on Lie Groups , 2020, ICLR.

[43]  Paul F. Whelan,et al.  Using filter banks in Convolutional Neural Networks for texture classification , 2016, Pattern Recognit. Lett..

[44]  H. Coxeter,et al.  Introduction to Geometry. , 1961 .

[45]  Franco Vivaldi,et al.  The Arithmetic of Discretized Rotations , 2006 .

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