International Conference on Computer and Knowledge Engineering

Home / 12th International Conference on Computer and Knowledge Engineering

Spatial-channel attention-based stochastic neighboring embedding pooling and long short term memory for lung nodules classification

Authors :

AHMED SAIHOOD¹ HOSSEIN KARSHENAS² AHMADREZA NAGHSH NILCHI³

1- Artificial Intelligence Department Faculty of Computer Engineering University of Isfahan Isfahan, Iran 2- Artificial Intelligence Department Faculty of Computer Engineering University of Isfahan Isfahan, Iran 3- Artificial Intelligence Department Faculty of Computer Engineering University of Isfahan Isfahan, Iran

Keywords :

SNE-pooling،CNN،attention-based pooling،lung nodules

Abstract :

Handling lesion size and location variance in lung nodules are one of the main shortcomings of traditional convolutional neural networks (CNNs). The pooling layer within CNNs reduces the resolution of the feature maps causing small local details loss that needs processing by the following layers. In this article, we proposed a new pooling-based stochastic neighboring embedding method (SNE-pooling) that is able to handle the long-range dependencies property of the lung nodules. Further, an attention-based SNE pooling model is proposed that could perform spatially and channel attention. The experimental results conducted on LIDC and LUNGx datasets show that the attention-based SNE pooling model significantly improves the performance for the state of the art.