کد مقاله کد نشریه سال انتشار مقاله انگلیسی نسخه تمام متن
6873104 1440629 2018 10 صفحه PDF دانلود رایگان
عنوان انگلیسی مقاله ISI
T1000: Mitigating the memory footprint of convolution neural networks with decomposition and re-fusion
موضوعات مرتبط
مهندسی و علوم پایه مهندسی کامپیوتر نظریه محاسباتی و ریاضیات
پیش نمایش صفحه اول مقاله
T1000: Mitigating the memory footprint of convolution neural networks with decomposition and re-fusion
چکیده انگلیسی
In this work, we propose T1000 to re-fuse the convolutions across tensors after applying the canonical polyadic decomposition to conventional convolution layers so that we can receive the benefit of reduced computation complexity, in the meanwhile mitigate the memory occupancy of the intermediate data. We demonstrate the effectiveness of our approach by applying canonical polyadic decomposition and re-fusion to the convolution layers of two well-known convolution neural networks, AlexNet and VGG-19 implemented with Caffe. Compared to the default canonical polyadic decomposition, our approach reduces the memory occupancy of the intermediate data by 84.6% and 77.4% for AlexNet and VGG-19 respectively. In addition, our approach improves the performance of AlexNet and VGG-19 by 1.77× and 1.4× respectively.
ناشر
Database: Elsevier - ScienceDirect (ساینس دایرکت)
Journal: Future Generation Computer Systems - Volume 84, July 2018, Pages 1-10
نویسندگان
, , , , ,