DeepLearningKit——基于 Metal 与 Swift 开发的面向 Apple iOS、OS X 和 tvOS 的 GPU 优化深度学习框架
机器学习
2016-05-17 v1 分布式、并行与集群计算
神经与进化计算
摘要
本文介绍 DeepLearningKit——一个支持在 iOS、OS X 和 tvOS 上使用预训练深度学习模型（卷积神经网络）的开源框架。DeepLearningKit 使用 Metal 开发以高效利用 GPU，并使用 Swift 以便与应用程序集成，例如基于 iOS 的 iPhone/iPad 移动应用、面向大屏幕的 tvOS 应用或 OS X 桌面应用。其目标是支持使用由流行框架（如 Caffe、Torch、TensorFlow、Theano、Pylearn、Deeplearning4J 和 Mocha）训练的深度学习模型。鉴于训练深度学习模型所需的大量 GPU 资源和时间，我们提出一种类似 App Store 的模型来分发和下载预训练且可复用的深度学习模型。
引用
@article{arxiv.1605.04614,
title = {DeepLearningKit - an GPU Optimized Deep Learning Framework for Apple's iOS, OS X and tvOS developed in Metal and Swift},
author = {Amund Tveit and Torbjørn Morland and Thomas Brox Røst},
journal= {arXiv preprint arXiv:1605.04614},
year = {2016}
}
备注
9 pages, 12 figures, open source documentation and code at deeplearningkit.org and github.com/deeplearningkit