| Literature DB >> 35464640 |
Mohammad Shifat-E-Rabbi1, Xuwang Yin2, Abu Hasnat Mohammad Rubaiyat2, Shiying Li1, Soheil Kolouri3, Akram Aldroubi4, Jonathan M Nichols5, Gustavo K Rohde6.
Abstract
We present a new supervised image classification method applicable to a broad class of image deformation models. The method makes use of the previously described Radon Cumulative Distribution Transform (R-CDT) for image data, whose mathematical properties are exploited to express the image data in a form that is more suitable for machine learning. While certain operations such as translation, scaling, and higher-order transformations are challenging to model in native image space, we show the R-CDT can capture some of these variations and thus render the associated image classification problems easier to solve. The method - utilizing a nearest-subspace algorithm in the R-CDT space - is simple to implement, non-iterative, has no hyper-parameters to tune, is computationally efficient, label efficient, and provides competitive accuracies to state-of-the-art neural networks for many types of classification problems. In addition to the test accuracy performances, we show improvements (with respect to neural network-based methods) in terms of computational efficiency (it can be implemented without the use of GPUs), number of training samples needed for training, as well as out-of-distribution generalization. The Python code for reproducing our results is available at [1].Entities:
Keywords: R-CDT; generative model; image classification; nearest subspace
Year: 2021 PMID: 35464640 PMCID: PMC9032314 DOI: 10.1007/s10851-021-01052-0
Source DB: PubMed Journal: J Math Imaging Vis ISSN: 0924-9907 Impact factor: 1.627