上海大学学报(自然科学版) ›› 2012, Vol. 18 ›› Issue (3): 305-310.doi: 10.3969/j.issn.1007-2861.2012.03.017

• 论文 • 上一篇    下一篇

邻居搜索问题在CUDA上基于KD-TRIE方法的优化与实现

包南森1,李正杰1,柴亚辉1,2,徐炜民1   

  1.  (1.上海大学 计算机工程与科学学院,上海 200072; 2.华东交通大学 信息工程学院,南昌 330013)
  • 出版日期:2012-06-30 发布日期:2012-06-30
  • 通讯作者: 徐炜民(1949~),男,教授,博士生导师,研究方向为计算机系统结构、并行处理等. E-mail:wmxu@staff.shu.edu.cn
  • 基金资助:

    国家高技术研究发展计划(863计划)资助项目(20009AA012201);上海市教委重点学科建设资助项目(J50103)

Optimization of Neighbor Searching Problem on  CUDA Based on KD-TRIE and Its Application

BAO Nan-sen1,LI Zheng-jie1,CHAI Ya-hui1,2,XU Wei-min1   

  1. (1. School of Computer Engineering and Science, Shanghai University, Shanghai 200072, China;2. School of Information Engineering Department, East China Jiaotong University, Nanchang 330013, China)
  • Online:2012-06-30 Published:2012-06-30

摘要: 介绍如何在CUDA上搭建KD-TRIE,并对其进行搜索,使其能适应解决邻居搜索问题.实验结果表明,当搜索半径较小(如整个空间直径的0.01和0.001),数据规模较大(如106)时,使用KD-TRIE进行搜索的效果最佳,与蛮力算法相比可以达到加速比5 000~15 000倍的效果;当搜索半径较大时,加速比会相应减少.采取优化措施,可以提高加速比.

关键词: CUDA, KD-TRIE, 图形处理器, k-最邻近结点算法

Abstract: This paper describes the construction of a KD-TRIE on CUDA, and the search and preprocessing for solving particle neighbor searching problem. The experimental results show that it is efficient to use KD-TRIE to search the CUDA when the search radius is not very large (e.g., 0.01 and 0.001 of the entire problem space diameter) and the data scale is relatively large (e.g., 106). Compare to the brute force method, the speedup factor is about 5 000 to 15 000. However, if the search radius is too large, the speedup factor will drop. Various optimization methods discussed can be used to raise the speedup factor.

Key words: CUDA, graphic processing unit (GPU), KD-TRIE, k-nearest neighbor algorithm (kNN)

中图分类号: