(Left to right) Maps for Global attention, Toeplitz Attention (D=7), Toeplitz Attention (D=1), Toeplitz Attention (D=0) (Left to right) Maps for Global attention, Toeplitz Attention (D=7), Toeplitz Attention (D=1), Toeplitz Attention (D=0)

Delineation of Skin Strata in Reflectance Confocal Microscopy Images using Recurrent Convolutional Networks with Toeplitz Attention

Abstract

Reflectance confocal microscopy (RCM) is an effective, non-invasive pre-screening tool for skin cancer diagnosis, but it requires extensive training and experience to assess accurately. There are few quantitative tools available to standardize image acquisition and analysis, and the ones that are available are not interpretable. In this study, we use a recurrent neural network with attention on convolutional network features. We apply it to delineate skin strata in vertically-oriented stacks of transverse RCM image slices in an interpretable manner. We introduce a new attention mechanism called Toeplitz attention, which constrains the attention map to have a Toeplitz structure. Testing our model on an expert labeled dataset of 504 RCM stacks, we achieve 88.17% image-wise classification accuracy, which is the current state-of-art.

Publication
In Machine learning for Health (ML4H) Workshop at NIPS 2017

Related