Representation of features as images with neighborhood dependencies for compatibility with convolutional neural networks.

Journal: Nature communications
Published Date:

Abstract

Deep learning with Convolutional Neural Networks has shown great promise in image-based classification and enhancement but is often unsuitable for predictive modeling using features without spatial correlations. We present a feature representation approach termed REFINED (REpresentation of Features as Images with NEighborhood Dependencies) to arrange high-dimensional vectors in a compact image form conducible for CNN-based deep learning. We consider the similarities between features to generate a concise feature map in the form of a two-dimensional image by minimizing the pairwise distance values following a Bayesian Metric Multidimensional Scaling Approach. We hypothesize that this approach enables embedded feature extraction and, integrated with CNN-based deep learning, can boost the predictive accuracy. We illustrate the superior predictive capabilities of the proposed framework as compared to state-of-the-art methodologies in drug sensitivity prediction scenarios using synthetic datasets, drug chemical descriptors as predictors from NCI60, and both transcriptomic information and drug descriptors as predictors from GDSC.

Authors

  • Omid Bazgir
    Department of Electrical and Computer Engineering, Texas Tech University, 1012 Boston Ave, Lubbock, TX, 79409, USA.
  • Ruibo Zhang
    Department of Electrical and Computer Engineering, Texas Tech University, 1012 Boston Ave, Lubbock, TX, 79409, USA.
  • Saugato Rahman Dhruba
    Department of Electrical and Computer Engineering, Texas Tech University, 1012 Boston Ave, Lubbock, TX, 79409, USA.
  • Raziur Rahman
    Department of Electrical and Computer Engineering, Texas Tech University, 1012 Boston Ave, Lubbock, TX, 79409, USA.
  • Souparno Ghosh
    Department of Mathematics and Statistics, Texas Tech University, 1108 Memorial Circle, Lubbock, TX, 79409, USA.
  • Ranadip Pal
    Department of Electrical and Computer Engineering, Texas Tech University, Lubbock, TX, 79409, USA. Ranadip.Pal@ttu.edu.