Accelerated tensor_toolbox version 1.0

来源:互联网 发布:索引超出数据界限 编辑:程序博客网 时间:2024/05/21 17:23

  基于张量分解的MATLAB工具箱已经来到了2.6版,但是在使用过程中仍有许多不尽如人意的地方,比如说,计算速度比较慢,而且作者似乎一开始就没有考虑大规模的问题。为了解决这些问题的一个或几个,笔者花了将近两个月的时间研究CUDA和MATLAB的混合编程,目前终于是做出了1.0版本的tensor_toolbox的加速工具箱。虽然我很想开源供大家使用,但是基于版权的问题还是没有办法将它做到开源(但是相信读了我的博客的读者很容易能按照tensor_toolbox的程序写出自己的工具箱)。由于是1.0的初版,这个工具箱还是有相当多不完备的地方,现在将其优点和缺点一一列举:
  优点:
  1、在计算能力仅仅为2.1的GPU上,加速比能达到百分之40,而且MATLAB的版本还是最新的2015b。
  2、比tensor_toolbox减少了调用ttm的次数(但是实验显示这个好像和程序的速度关系不大),使程序看起来更简洁。
  3、支持数组形式的输入,而无需转化为张量
  缺点:
  1、不支持大规模的张量问题,比如(1000,1000,200)以上就计算不了。
  2、精度问题,为了加快运算牺牲了精度,导致奇异值分解部分比较慢。
  3、不支持稀疏的数据

  希望在2.0版本的时候能够解决稀疏的问题,至于精度和大规模的问题可能要去查查论文了,不是那么快能够解决的,但是还是希望大家能够给我提出宝贵的意见和建议。

0 0
原创粉丝点击