EMBEDDING SIGNALS ON GRAPHS WITH UNBALANCED DIFFUSION EARTH MOVER'S DISTANCE.

Autor: Tong A; Yale University, Dept. of Comp. Sci., New Haven, CT, USA., Huguet G; Université de Montréal, Dept. of Math. & Stat.; Mila - Quebec AI Institute, Montreal, QC, Canada., Shung D; Dept. of Medicine, New Haven, CT, USA., Natik A; Université de Montréal, Dept. of Math. & Stat.; Mila - Quebec AI Institute, Montreal, QC, Canada., Kuchroo M; Dept. of Genetics, New Haven, CT, USA., Lajoie G; Université de Montréal, Dept. of Math. & Stat.; Mila - Quebec AI Institute, Montreal, QC, Canada., Wolf G; Université de Montréal, Dept. of Math. & Stat.; Mila - Quebec AI Institute, Montreal, QC, Canada., Krishnaswamy S; Dept. of Genetics, New Haven, CT, USA.; Yale University, Dept. of Comp. Sci., New Haven, CT, USA.
Jazyk: angličtina
Zdroj: Proceedings of the ... IEEE International Conference on Acoustics, Speech, and Signal Processing. ICASSP (Conference) [Proc IEEE Int Conf Acoust Speech Signal Process] 2022 May; Vol. 2022, pp. 5647-5651. Date of Electronic Publication: 2022 Apr 27.
DOI: 10.1109/icassp43922.2022.9746556
Abstrakt: In modern relational machine learning it is common to encounter large graphs that arise via interactions or similarities between observations in many domains. Further, in many cases the target entities for analysis are actually signals on such graphs. We propose to compare and organize such datasets of graph signals by using an earth mover's distance (EMD) with a geodesic cost over the underlying graph. Typically, EMD is computed by optimizing over the cost of transporting one probability distribution to another over an underlying metric space. However, this is inefficient when computing the EMD between many signals. Here, we propose an unbalanced graph EMD that efficiently embeds the unbalanced EMD on an underlying graph into an L 1 space, whose metric we call unbalanced diffusion earth mover's distance (UDEMD) . Next, we show how this gives distances between graph signals that are robust to noise. Finally, we apply this to organizing patients based on clinical notes, embedding cells modeled as signals on a gene graph, and organizing genes modeled as signals over a large cell graph. In each case, we show that UDEMD-based embeddings find accurate distances that are highly efficient compared to other methods.
Databáze: MEDLINE