about
Spherical Text Embedding (arxiv.org)
3 points by melissalobos on Jan 19, 2022 | hide | past | pdf | discuss on HN

In plain words: Text embeddings are usually trained in flat space but judged by direction, so this model trains words and paragraphs directly on a sphere, where only direction matters. It beat the best earlier unsupervised embeddings on word similarity and document clustering.

Abstract

Unsupervised text embedding has shown great power in a wide range of NLP tasks. While text embeddings are typically learned in the Euclidean space, directional similarity is often more effective in tasks such as word similarity and document clustering, which creates a gap between the training stage and usage stage of text embedding. To close this gap, we propose a spherical generative model based on which unsupervised word and paragraph embeddings are jointly learned. To learn text embeddings in the spherical space, we develop an efficient optimization algorithm with convergence guarantee based on Riemannian optimization. Our model enjoys high efficiency and achieves state-of-the-art performances on various text embedding tasks including word similarity and document clustering.

Yu Meng, Jiaxin Huang, Guangyuan Wang, Chao Zhang, Honglei Zhuang, Lance Kaplan, Jiawei Han
arXiv:1911.01196 · cs.CL, cs.LG, stat.ML · submitted Nov 4, 2019
abstract · pdf · html · NeurIPS 2019. (Code: https://github.com/yumeng5/Spherical-Text-Embedding)

add comment on HN