We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Autoencoders as Weight Initialization of Deep Classification Networks for Cancer versus Cancer Studies

Abstract: Cancer is still one of the most devastating diseases of our time. One way of automatically classifying tumor samples is by analyzing its derived molecular information (i.e., its genes expression signatures). In this work, we aim to distinguish three different types of cancer: thyroid, skin, and stomach. For that, we compare the performance of a Denoising Autoencoder (DAE) used as weight initialization of a deep neural network. Although we address a different domain problem in this work, we have adopted the same methodology of Ferreira et al.. In our experiments, we assess two different approaches when training the classification model: (a) fixing the weights, after pre-training the DAE, and (b) allowing fine-tuning of the entire classification network. Additionally, we apply two different strategies for embedding the DAE into the classification network: (1) by only importing the encoding layers, and (2) by inserting the complete autoencoder. Our best result was the combination of unsupervised feature learning through a DAE, followed by its full import into the classification network, and subsequent fine-tuning through supervised training, achieving an F1 score of 98.04% +/- 1.09 when identifying cancerous thyroid samples.
Comments: 5 pages, 2 figures (each one with 6 sub-figures), 2 tables. Special Session of Machine Learning in Healthcare Informatics and Medical Biology of the 16th International Conference on Computational Intelligence methods for Bioinformatics and Biostatistics (CIBB2019)
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:2001.05253 [cs.LG]
  (or arXiv:2001.05253v1 [cs.LG] for this version)

Submission history

From: Mafalda Falcão Ferreira [view email]
[v1] Wed, 15 Jan 2020 11:49:41 GMT (155kb,D)

Link back to: arXiv, form interface, contact.