公开(公告)号：CN115713628A

公开(公告)日：2023-02-24

申请号：CN202211552678.1

申请日：2022-12-05

Applicant: 西南科技大学

Inventor： 王耀彬 ,  申超 ,  唐苹苹 ,  刘欢 ,  杨梁 ,  从明 ,  陈俊仕 ,  安虹

IPC: G06V10/40 ,  G06V10/82 ,  G06V10/94 ,  G06F16/51

Abstract: 本发明公开了一种基于全局索引表的快速特征采集方法，该方法包括如下步骤：剖析内存使用，确定可用于缓存分配的最大GPU内存量；在离线状态下，按出度数对顶点进行预排序。并在运行时选择高出度的顶点来填充GPU缓存；确定mini‑batch输入的顶点数；确定全局索引表中局部顶点的位置；从GPU缓存数据中获取输入特征；当前批次特征采集结束。本发明针对目前图神经网络中的特征采集机制均无法达到快速查找特征数据的现状，提出一种基于全局索引表的快速特征采集方法，需要在GPU内存中有一个连续的空间用于缓存特征数据，同时将顶点元数据组织到一个全局索引表中（元数据远小于缓存的特征数据），用于答复查询到的顶点位于什么位置，以便完成快速查找特征。