利用张量核心单元的相似性搜索
数据结构与算法
2020-06-24 v1 分布式、并行与集群计算
信息检索
摘要
张量核心单元（TCU）是为深度神经网络开发的硬件加速器，可高效支持两个稠密 矩阵的乘法，其中 为给定的硬件参数。本文中我们表明 TCU 也能加速相似性搜索问题。我们提出了 Johnson-Lindenstrauss 降维和相似性连接的算法，通过利用 TCU，相对于传统方法实现了 倍的加速。
引用
@article{arxiv.2006.12608,
title = {Similarity Search with Tensor Core Units},
author = {Thomas D. Ahle and Francesco Silvestri},
journal= {arXiv preprint arXiv:2006.12608},
year = {2020}
}