一种面向硬件实现的GBDT分类模型压缩方法与流程

文档序号:13915599阅读:来源:国知局

技术特征:

技术总结
一种面向硬件实现的GBDT分类模型压缩方法,该方法引入自定义剪枝、自定义编码以及自定义寻址来实现模型结构上、参数上及寻址上的压缩,以此来降低模型的存储空间。在提取构建模型的必须参数后,对其中的预测值value数据进行近似,并在此基础上对于模型中的每棵决策树进行剪枝;对剪枝后的模型进行自定义寻址,并将寻址信息记入相应节点的参数信息当中;同时,对于模型中的每一组节点信息进行二进制组合编码后按序存入其类别对应的数组即实现了对GBDT分类模型的压缩。利用位运算读取节点编码中存储的参数信息即可实现模型重建。对原始模型的剪枝、模型参数的编码以及模型寻址过程的改变,使模型在硬件系统中的存储占用得到大幅降低。

技术研发人员:梅魁志;王晓;吴杰;赵海波;薛建儒;郑南宁;赵英海;高凡
受保护的技术使用者:西安交通大学
技术研发日:2017.09.27
技术公布日:2018.03.13
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1