arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2206.04763cs.LG

用于距离学习的神经Bregman散度

Neural Bregman Divergences for Distance Learning

University of Maryland, Baltimore County(马里兰大学巴尔的摩县分校);Booz Allen Hamilton(博兹艾伦汉密尔顿公司)

Fred Lu, Edward Raff, Francis Ferraro

AI总结:

提出一种通过输入凸神经网络以可微方式学习任意Bregman散度的新方法,克服了以往工作的限制,并在非对称回归、排序、聚类等任务中表现出色,展示了其在非对称学习中的通用性。

摘要:

许多度量学习任务,如三元组学习、最近邻检索和可视化,主要被视为嵌入任务,其中最终的度量是欧几里得距离的某种变体(例如余弦或马氏距离),并且算法必须学习将点嵌入到预先选择的空间中。非欧几里得几何的研究通常未被探索,我们认为这是由于缺乏学习非欧几里得距离度量的工具。最近的工作表明,Bregman散度可以从数据中学习,为学习非对称距离开辟了一种有前景的方法。我们提出了一种通过输入凸神经网络以可微方式学习任意Bregman散度的新方法,并表明它克服了以前工作的重大限制。我们还证明,我们的方法在一组新的和以前研究过的任务上更忠实地学习散度,包括非对称回归、排序和聚类。我们的测试进一步扩展到已知的非对称但非Bregman任务,尽管存在误设定,我们的方法仍然具有竞争力,显示了我们的方法在非对称学习中的通用性。

英文摘要:

Many metric learning tasks, such as triplet learning, nearest neighbor retrieval, and visualization, are treated primarily as embedding tasks where the ultimate metric is some variant of the Euclidean distance (e.g., cosine or Mahalanobis), and the algorithm must learn to embed points into the pre-chosen space. The study of non-Euclidean geometries is often not explored, which we believe is due to a lack of tools for learning non-Euclidean measures of distance. Recent work has shown that Bregman divergences can be learned from data, opening a promising approach to learning asymmetric distances. We propose a new approach to learning arbitrary Bergman divergences in a differentiable manner via input convex neural networks and show that it overcomes significant limitations of previous works. We also demonstrate that our method more faithfully learns divergences over a set of both new and previously studied tasks, including asymmetric regression, ranking, and clustering. Our tests further extend to known asymmetric, but non-Bregman tasks, where our method still performs competitively despite misspecification, showing the general utility of our approach for asymmetric learning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑