Abstract:
|
One key challenge encountered in single-cell-data clustering is to combine clustering results of datasets acquired from multiple sources. We propose to represent the clustering result of each dataset by a Gaussian mixture model (GMM) and produce an integrated result based on the notion of Wasserstein barycenter. However, the precise barycenter of GMMs is computationally infeasible. Importantly, it may not be a GMM with a reasonable number of components. We thus propose to use the Minimized Aggregated Wasserstein (MAW) barycenter. With significantly improved tractability, the MAW distance for GMMs approximates the Wasserstein metric, and recent theoretical advances further justify its usage. In this talk, we develop a new algorithm for computing the barycenter of GMMs under MAW. We also prove that the MAW barycenter has the same expectation as the Wasserstein barycenter. Our proposed algorithm for clustering integration scales well with the data dimension and the number of mixture components, with complexity independent of data size. We demonstrate that the new method achieves better clustering results on several single-cell RNA-seq datasets than a few popular methods.
|