Learning Permutation Invariant Representations using Memory Networks

11/18/2019
by   Shivam Kalra, et al.
0

Many real world tasks such as 3D object detection and high-resolution image classification involve learning from a set of instances. In these cases, only a group of instances, a set, collectively contains meaningful information and therefore only the sets have labels, and not individual data instances. In this work, we present a permutation invariant neural network called a Memory-based Exchangeable Model (MEM) for learning set functions. The model consists of memory units that embed an input sequence to high-level features (memories) enabling the model to learn inter-dependencies among instances of the set in the form of attention vectors. To demonstrate its learning ability, we evaluated our model on test datasets created using MNIST, point cloud classification, and population estimation. We also tested the model for classifying histopathology whole slide images to discriminate between two subtypes of Lung cancer—Lung Adenocarcinoma, and Lung Squamous Cell Carcinoma. We systematically extracted patches from lung cancer images from The Cancer Genome Atlas (TCGA) dataset, the largest public repository of histopathology images. The proposed method achieved a competitive classification accuracy of 84.84%. The results on other datasets are promising and demonstrate the efficacy of our model.

READ FULL TEXT
research
04/16/2020

Representation Learning of Histopathology Images using Graph Neural Networks

Representation learning for Whole Slide Images (WSIs) is pivotal in deve...
research
08/24/2018

Using Apple Machine Learning Algorithms to Detect and Subclassify Non-Small Cell Lung Cancer

Lung cancer continues to be a major healthcare challenge with high morbi...
research
04/11/2023

Artificial intelligence based prediction on lung cancer risk factors using deep learning

In this proposed work, we identified the significant research issues on ...
research
04/22/2018

A Deep Convolutional Neural Network for Lung Cancer Diagnostic

In this paper, we examine the strength of deep learning technique for di...
research
12/15/2015

Hyper-Heuristic Algorithm for Finding Efficient Features in Diagnose of Lung Cancer Disease

Background: Lung cancer was known as primary cancers and the survival ra...
research
10/11/2017

Lung Cancer Screening Using Adaptive Memory-Augmented Recurrent Networks

In this paper, we investigate the effectiveness of deep learning techniq...
research
09/10/2020

Why I'm not Answering

Safe deployment of deep learning systems in critical real world applicat...

Please sign up or login with your details

Forgot password? Click here to reset