Accelerated approximate nearest neighbors search through hierarchical product quantization
File(s)AmeerFPT2019.pdf (1.75 MB)
Accepted version
Author(s)
Abdelhadi, Ameer
Bouganis, Christos
Constantinides, George
Type
Conference Paper
Abstract
A fundamental recurring task in many machinelearning applications is the search for the Nearest Neighbor inhigh dimensional metric spaces. Towards answering queries inlarge scale problems, state-of-the-art methods employ Approx-imate Nearest Neighbors (ANN) search, a search that returnsthe nearest neighbor with high probability, as well as techniquesthat compress the dataset. Product-Quantization (PQ) based ANNsearch methods have demonstrated state-of-the-art performancein several problems, including classification, regression and infor-mation retrieval. The dataset is encoded into a Cartesian productof multiple low-dimensional codebooks, enabling faster searchand higher compression. Being intrinsically parallel, PQ-basedANN search approaches are amendable for hardware accelera-tion. This paper proposes a novel Hierarchical PQ (HPQ) basedANN search method as well as an FPGA-tailored architecturefor its implementation that outperforms current state of theart systems. HPQ gradually refines the search space, reducingthe number of data compares and enabling a pipelined search.The mapping of the architecture on a Stratix 10 FPGA devicedemonstrates over×250 speedups over current state-of-the-artsystems, opening the space for addressing larger datasets and/orimproving the query times of current systems.
Date Issued
2020-02-03
Date Acceptance
2019-10-07
Citation
2020, pp.1-9
Publisher
IEEE
Start Page
1
End Page
9
Copyright Statement
© 2020 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Sponsor
Royal Academy Of Engineering
Imagination Technologies Ltd
Engineering & Physical Science Research Council (EPSRC)
Engineering & Physical Science Research Council (EPSRC)
Identifier
https://ieeexplore.ieee.org/document/8977838
Grant Number
Prof Constantinides Chair
Prof Constantinides Chair
EP/P010040/1
EP/S030069/1
Source
2019 International Conference on Field-Programmable Technology
Subjects
Science & Technology
Technology
Computer Science, Interdisciplinary Applications
Engineering, Electrical & Electronic
Computer Science
Engineering
Approximate search
similarity search
nearest neighbor search
online indexing
high-dimensional indexing
product quantization
vector quantization
artificial intelligence
ALGORITHMS
Publication Status
Published
Start Date
2019-12-09
Finish Date
2019-12-13
Coverage Spatial
Tianjin, China
Date Publish Online
2020-02-03