IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0092409.html
   My bibliography  Save this article

Fast k-NNG Construction with GPU-Based Quick Multi-Select

Author

Listed:
  • Ivan Komarov
  • Ali Dashti
  • Roshan M D'Souza

Abstract

In this paper, we describe a new brute force algorithm for building the -Nearest Neighbor Graph (k-NNG). The k-NNG algorithm has many applications in areas such as machine learning, bio-informatics, and clustering analysis. While there are very efficient algorithms for data of low dimensions, for high dimensional data the brute force search is the best algorithm. There are two main parts to the algorithm: the first part is finding the distances between the input vectors, which may be formulated as a matrix multiplication problem; the second is the selection of the k-NNs for each of the query vectors. For the second part, we describe a novel graphics processing unit (GPU)-based multi-select algorithm based on quick sort. Our optimization makes clever use of warp voting functions available on the latest GPUs along with user-controlled cache. Benchmarks show significant improvement over state-of-the-art implementations of the k-NN search on GPUs.

Suggested Citation

  • Ivan Komarov & Ali Dashti & Roshan M D'Souza, 2014. "Fast k-NNG Construction with GPU-Based Quick Multi-Select," PLOS ONE, Public Library of Science, vol. 9(5), pages 1-9, May.
  • Handle: RePEc:plo:pone00:0092409
    DOI: 10.1371/journal.pone.0092409
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0092409
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0092409&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0092409?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Ahmed Shamsul Arefin & Carlos Riveros & Regina Berretta & Pablo Moscato, 2012. "GPU-FS-kNN: A Software Tool for Fast and Scalable kNN Computation Using GPUs," PLOS ONE, Public Library of Science, vol. 7(8), pages 1-13, August.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Ali Dashti & Ivan Komarov & Roshan M D’Souza, 2013. "Efficient Computation of k-Nearest Neighbour Graphs for Large High-Dimensional Data Sets on GPU Clusters," PLOS ONE, Public Library of Science, vol. 8(9), pages 1-12, September.

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0092409. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.