# Convolutional neural network models for cancer type prediction based on gene expression

arXiv: https://arxiv.org/abs/1906.07794 · HTML: https://mlsanity.com/paper/1906.07794/

## Key facts

- Cited by 22 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 18 Jun 2019.

## Built on by (most cited first, 22 of 22)

- [Do Feature Attribution Methods Correctly Attribute Features?](https://mlsanity.com/paper/2104.14403/) (2021)
- [A Multimodal Knowledge-enhanced Whole-slide Pathology Foundation Model](https://mlsanity.com/paper/2407.15362/) (2024)
- [Transcriptomics-guided Slide Representation Learning in Computational Pathology](https://mlsanity.com/paper/2405.11618/) (2024)
- [Machine Learning Methods for Cancer Classification Using Gene Expression Data: A Review](https://mlsanity.com/paper/2301.12222/) (2023)
- [DeepGene Transformer: Transformer for the gene expression-based classification of cancer subtypes](https://mlsanity.com/paper/2108.11833/) (2021)
- [Breast cancer detection using artificial intelligence techniques: A systematic literature review](https://mlsanity.com/paper/2203.04308/) (2022)
- [Interpretable Graph Kolmogorov-Arnold Networks for Multi-Cancer Classification and Biomarker Identification using Multi-Omics Data](https://mlsanity.com/paper/2503.22939/) (2025)
- [Multiple Instance Neural Networks Based on Sparse Attention for Cancer Detection using T-cell Receptor Sequences](https://mlsanity.com/paper/2208.04524/) (2022)
- [PINNet: a deep neural network with pathway prior knowledge for Alzheimer's disease](https://mlsanity.com/paper/2211.15669/) (2022)
- [Precision Cancer Classification and Biomarker Identification from mRNA Gene Expression via Dimensionality Reduction and Explainable AI](https://mlsanity.com/paper/2410.07260/) (2024)
- [Can Large Language Models Replace Data Scientists in Biomedical Research?](https://mlsanity.com/paper/2410.21591/) (2024)
- [Application of Machine Learning in Early Recommendation of Cardiac Resynchronization Therapy](https://mlsanity.com/paper/2109.06139/) (2021)
- [LASSO-MOGAT: A Multi-Omics Graph Attention Framework for Cancer Classification](https://mlsanity.com/paper/2408.17384/) (2024)
- [A Study of Feature Selection and Extraction Algorithms for Cancer Subtype Prediction](https://mlsanity.com/paper/2109.14648/) (2021)
- [Can a face tell us anything about an NBA prospect? -- A Deep Learning approach](https://mlsanity.com/paper/2212.06804/) (2022)
- [Supervised Machine Learning for Breast Cancer Risk Factors Analysis and Survival Prediction](https://mlsanity.com/paper/2304.07299/) (2023)
- [Gene-MOE: A sparsely gated prognosis and classification framework exploiting pan-cancer genomic information](https://mlsanity.com/paper/2311.17401/) (2023)
- [Comparative Analysis of Multi-Omics Integration Using Advanced Graph Neural Networks for Cancer Classification](https://mlsanity.com/paper/2410.05325/) (2024)
- [TSEML: A task-specific embedding-based method for few-shot classification of cancer molecular subtypes](https://mlsanity.com/paper/2412.13228/) (2024)
- [A Pan-cancer Classification Model using Multi-view Feature Selection Method and Ensemble Classifier](https://mlsanity.com/paper/2501.06805/) (2025)
- [Transformer-Based Representation Learning for Robust Gene Expression Modeling and Cancer Prognosis](https://mlsanity.com/paper/2504.09704/) (2025)
- [Hierarchical Pooling and Explainability in Graph Neural Networks for Tumor and Tissue-of-Origin Classification Using RNA-seq Data](https://mlsanity.com/paper/2601.06381/) (2026)

## Abstract (arXiv)

Background Precise prediction of cancer types is vital for cancer diagnosis and therapy. Important cancer marker genes can be inferred through predictive model. Several studies have attempted to build machine learning models for this task however none has taken into consideration the effects of tissue of origin that can potentially bias the identification of cancer markers. Results In this paper, we introduced several Convolutional Neural Network (CNN) models that take unstructured gene expression inputs to classify tumor and non-tumor samples into their designated cancer types or as normal. Based on different designs of gene embeddings and convolution schemes, we implemented three CNN models: 1D-CNN, 2D-Vanilla-CNN, and 2D-Hybrid-CNN. The models were trained and tested on combined 10,340 samples of 33 cancer types and 731 matched normal tissues of The Cancer Genome Atlas (TCGA). Our models achieved excellent prediction accuracies (93.9-95.0%) among 34 classes (33 cancers and normal). Furthermore, we interpreted one of the models, known as 1D-CNN model, with a guided saliency technique and identified a total of 2,090 cancer markers (108 per class). The concordance of differential expression of these markers between the cancer type they represent and others is confirmed. In breast cancer, for instance, our model identified well-known markers, such as GATA3 and ESR1. Finally, we extended the 1D-CNN model for prediction of breast cancer subtypes and achieved an average accuracy of 88.42% among 5 subtypes. The codes can be found at https://github.com/chenlabgccri/CancerTypePrediction.