Using transfer learning approaches to predict RNA-Seq gene expression data for cancer classification
Abstract
There is a great need to categorize cancer types for early cancer detection and treatment. RNA-Seq data is essential for getting insight into the differentially expressed genes. Due to its high dimensionality and complexity, performing an analysis on RNA-Seq data is quite challenging. In the past, RNA-Seq data were analyzed for a single cancer type as a two-class problem (either positive or negative) and did not contain information from other classes of cancer types. To classify different cancer types and discover the most promising genes, RNA-Seq data for different types of cancer should be examined. Multiple repositories offer RNA-Seq-based cancer types data. The present study incorporates a dataset from the Mendeley repository for classification. RNA-Seq values are then converted to their respective 2D images using some transformations. The classification problem is handled by five Transfer Learning (TL) algorithms (VGG16, VGG19, Resnet50, Resnet101, and Resnet152). Four different splitting strategies are applied for each classifier presented in the results and discussion section. A comparative analysis is also carried out with and without data augmentation. Results show that classifiers perform best at a split of 70-30. VGG16 attained the best position on overall results by achieving an accuracy of 95%. Hence, VGG16 is the leading TL algorithm for classification among all the accessible models and is not difficult to execute and easy to comprehend.
Fields of Study & Subjects
Gene expression and cancer classification
Machine Learning in Bioinformatics
Cancer-related molecular mechanisms research
Publication details
- Journal / source
- Frontiers in Artificial Intelligence
- Publisher
- Frontiers Media SA
- Volume
- 9
- Pages
- 1790249–1790249
- Publication date
- 19 March 2026
- Language
- EN