KNN exato em GPU

  • Michel B. Cordeiro UFPR
  • Bruno H. Meyer UFPR
  • Wagner M. Nunan Zola UFPR

Resumo


Este artigo apresenta duas implementações do KNN em GPU. Cada implementação é otimizada para diferentes quantidades de pontos no conjunto de consultas. Essas versões são comparadas com algoritmo RSFK, que é uma aproximação do KNN. As implementações mostraram estar bem otimizado quando a quantidade de pontos é pequena, alcançando uma aceleração de até 80 vezes em relação ao RSFK, também em GPU, para consultas de 128 pontos.

Referências

Johnson, J., Douze, M., and Jégou, H. (2019). Billion-scale similarity search with gpus. IEEE Transactions on Big Data, 7(3):535-547.

Meyer, B., Pozo, A., and Nunan Zola, W. M. (2021). Warp-centric k-nearest neighbor graphs construction on GPU. In 50th International Conference on Parallel Processing Workshop, ICPP Workshops '21, New York, NY, USA. Pub. ACM.

Meyer, B., Pozo, A., and Zola, W. (2022). ANN-RSFK: Busca genérica de similaridade em GPU. In Anais da XXII Escola Regional de Alto Desempenho da Região Sul, pages 89-90, Porto Alegre, RS, Brasil. SBC.
Publicado
10/05/2023
Como Citar

Selecione um Formato
CORDEIRO, Michel B.; MEYER, Bruno H.; ZOLA, Wagner M. Nunan. KNN exato em GPU. In: ESCOLA REGIONAL DE ALTO DESEMPENHO DA REGIÃO SUL (ERAD-RS), 23. , 2023, Porto Alegre/RS. Anais [...]. Porto Alegre: Sociedade Brasileira de Computação, 2023 . p. 17-20. ISSN 2595-4164. DOI: https://doi.org/10.5753/eradrs.2023.229273.