Wednesday, March 4, 2015

BIDMach: faster machine learning on a single machine

I got this from my colleague Yucheng Low. BIDMach is a new open source project (BSD3 license) which speeds up basic machine learning algorithms using GPU.

Some impressive performance numbers on Github. For example:

Criteo Dataset

Criteo released a medium-sized (12 GB) dataset with a single target (click, no click) with a very sparse set of features. This is representative of many click prediction tasks in industry.
SystemNodes/CoresnpassesAUCTimeCostEnergy(KJ)
Spark8/32100.62964s$0.641500
Spark32/128100.62400s$1.002500
BIDMach110.6681s$0.016
BIDMach1100.72805s$0.1360

No comments:

Post a Comment