Unsupervised machine learning framework for discriminating major variants of concern during COVID-19.

Journal: PloS one
Published Date:

Abstract

Due to the high mutation rate of the virus, the COVID-19 pandemic evolved rapidly. Certain variants of the virus, such as Delta and Omicron emerged with altered viral properties leading to severe transmission and death rates. These variants burdened the medical systems worldwide with a major impact to travel, productivity, and the world economy. Unsupervised machine learning methods have the ability to compress, characterize, and visualize unlabelled data. This paper presents a framework that utilizes unsupervised machine learning methods to discriminate and visualize the associations between major COVID-19 variants based on their genome sequences. These methods comprise a combination of selected dimensionality reduction and clustering techniques. The framework processes the RNA sequences by performing a k-mer analysis on the data and further visualises and compares the results using selected dimensionality reduction methods that include principal component analysis (PCA), t-distributed stochastic neighbour embedding (t-SNE), and uniform manifold approximation projection (UMAP). Our framework also employs agglomerative hierarchical clustering to visualize the mutational differences among major variants of concern and country-wise mutational differences for selected variants (Delta and Omicron) using dendrograms. We also provide country-wise mutational differences for selected variants via dendrograms. We find that the proposed framework can effectively distinguish between the major variants and has the potential to identify emerging variants in the future.

Authors

  • Rohitash Chandra
  • Chaarvi Bansal
    Transitional Artificial Intelligence Research Group, School of Mathematics and Statistics, UNSW Sydney, Sydney, Australia.
  • Mingyue Kang
    Transitional Artificial Intelligence Research Group, School of Mathematics and Statistics, UNSW Sydney, Sydney, Australia.
  • Tom Blau
    Data 61, CSIRO, Sydney, Australia.
  • Vinti Agarwal
    Department of Computer Science and Information Systems, Birla Institute of Technology and Science Pilani, Rajasthan, India.
  • Pranjal Singh
    Department of Computer Science and Engineering, Indian Institute of Technology Guwathi, Assam, India.
  • Laurence O W Wilson
    1 Health and Biosecurity, CSIRO , Sydney, Australia .
  • Seshadri Vasan
    Department of Health Sciences, University of York, York, United Kingdom.