TF-IDF k-mer-based Classical and Hybrid Machine Learning Models for SARS-CoV-2 Variant Classification under Imbalanced Genomic Data
Nazmul HaqueiD, Abdul MazediD, Jannatul Nayeem Ankhi, Md Jamal UddiniD
Abstract
Accurate classification of SARS-CoV-2 genomic variants is essential for effective genomic surveillance, yet it is challenged by extreme class imbalance, limited representation of rare variants, and distribution shifts in real-world sequencing data. In this study, we employed hybrid RF-SVM framework designed for robust detection of rare SARS-CoV-2 variants. It integrates a random forest and a polynomial-kernel based support vector machine to enhance sensitivity to minority classes while maintaining overall predictive stability. We systematically compared classical machine learning models, deep learning approaches, and hybrid strategies under both standard and distribution-shifted evaluation settings. Our results show that classical models using TF-IDF-based k-mer features outperform deep learning methods on macro-averaged performance metrics. The Random Forest classifier using TF-IDF Feature achieved the best overall performance, with a macro-averaged F1-score of 0.8894 and an accuracy of 96.3%. The model also demonstrated strong generalization ability, as evidenced by stable cross-validation performance (CV accuracy = 0.9637). Hybrid RF-SVM model further improves rare variant detection under severe class imbalance. Calibration analysis indicates reliable probability estimates for common variants, although challenges persist for minority classes. Overall, this study highlights the limitations of deep learning in highly imbalanced genomic settings and demonstrates that carefully designed hybrid machine learning approaches provide an effective and interpretable solution for rare SARS-CoV-2 variant detection.
02 /Specification & Classification
Click any tag to browse matching papersReproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- A multi-class gene classifier for SARS-CoV-2 variants based on convolutional neural network
Junyu Fan, Chutao Chen, … Guifu Wu·bioRxiv·Nov 22, 2021
- Enhanced Deep Convolutional Neural Network for SARS-CoV-2 Variants Classification
Mike J. Mwanga, Hesborn Omwandho Obura, … Olaitan Igbagbo Awe·bioRxiv·Aug 9, 2023
- Classifying COVID-19 variants based on genetic sequences using deep learning models
Sayantani Basu, Roy H. Campbell·Springer series in reliability engineeringPeer-reviewed·Jul 25, 2022
- Genomic Evolution of SARS-CoV-2 Delta Variants Pre- and Post-Omicron Emergence using Alignment-free Machine Learning models
Sathish Sankar, Kaushika Anandharaman, … Esaki Muthu Shankar·PLoS ONEPeer-reviewed·Mar 19, 2026
- Machine learning-based approach KEVOLVE efficiently identifies SARS-CoV-2 variant-specific genomic signatures
Dylan Lebatteux, Hugo Soudeyns, … Abdoulaye Baniré Diallo·PLoS ONEPeer-reviewed·Jan 19, 2024
- Accurate and Fast Clade Assignment via Deep Learning and Frequency Chaos Game Representation
Jorge Avila Cartes, Santosh Kumar Anand, … Gianluca Della Vedova·GigaSciencePeer-reviewed·Dec 28, 2022