Supervised learning of sheared distributions using linearized optimal transport
Supervised learning of sheared distributions using linearized optimal transport
复制标题
DOI:
10.1007/s43670-022-00038-2
复制
发表时间:
2022-01
期刊:
影响因子:
--
通讯作者:
Varun Khurana;Harish Kannan;A. Cloninger;Caroline Moosmüller
中科院分区:
文献类型:
--
作者:
Varun Khurana;Harish Kannan;A. Cloninger;Caroline Moosmüller
In this paper we study supervised learning tasks on the space of probability measures. We approach this problem by embedding the space of probability measures intospaces using the optimal transport framework. In the embedding spaces, regular machine learning techniques are used to achieve linear separability. This idea has proved successful in applications and when the classes to be separated are generated by shifts and scalings of a fixed measure. This paper extends the class of elementary transformations suitable for the framework to families of shearings, describing conditions under which two classes of sheared distributions can be linearly separated. We furthermore give necessary bounds on the transformations to achieve a pre-specified separation level, and show how multiple embeddings can be used to allow for larger families of transformations. We demonstrate our results on image classification tasks.