vix.ing · top · new · best · stats · spec

Deep Distributional Sequence Embeddings Based on a Wasserstein Loss

2019/12/04 by Ahmed Abdel‐Wahab, Abdelwahab, Ahmed, Niels Landwehr +1 · 1 citation
Computer Science · #FOS: Computer and information sciences #Generative Adversarial Networks and Image Synthesis #Human Pose and Action Recognition #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Machine Learning in Healthcare

paper · pdf · doi:10.48550/arxiv.1912.01933

openalex publication_date 2019/12/04 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Deep metric learning employs deep neural networks to embed instances into a metric space such that distances between instances of the same class are small and distances between instances from different classes are large. In most existing deep metric learning techniques, the embedding of an instance is given by a feature vector produced by a deep neural network and Euclidean distance or cosine similarity defines distances between these vectors. In this paper, we study deep distributional embeddings of sequences, where the embedding of a sequence is given by the distribution of learned deep features across the sequence. This has the advantage of capturing statistical information about the distribution of patterns within the sequence in the embedding. When embeddings are distributions rather than vectors, measuring distances between embeddings involves comparing their respective distributions. We propose a distance metric based on Wasserstein distances between the distributions and a corresponding loss function for metric learning, which leads to a novel end-to-end trainable embedding model. We empirically observe that distributional embeddings outperform standard vector embeddings and that training with the proposed Wasserstein metric outperforms training with other distance functions.

Cited by

Related