Upgrade to Pro — share decks privately, control downloads, hide ads and more …

Scott Le Grand - DSSTNE - LA Data Science Meetu...

Data Science LA
October 28, 2016
350

Scott Le Grand - DSSTNE - LA Data Science Meetup - Oct 2016

Data Science LA

October 28, 2016
Tweet

More Decks by Data Science LA

Transcript

  1. DSSTNE: A New Deep Learning Framework For Large Sparse Datasets

    https://github.com/amznlabs/amazon-dsstne Scott Le Grand Senior Scientist Teza Technologies [email protected]
  2. Outline • What's Deep Learning? • Why GPUs? • What's

    Unique about Deep Learning at Amazon • DSSTNE • Benchmarks • Demo • DSSTNE at scale • Fun with CNNs • How to buy Deep Learning Hardware
  3. Neural Networks • World’s most lucrative application of the chain

    rule from calculus • x is the input data • A1 and A2 are linear transformations • f1 and f2 are some sort of nonlinear function
  4. Deep Learning/Neural Networks in One Slide* X L+1 = X

    L * W L→L+1 d L = d L+1 * W L→L+1 DW L→L+1 = XT L * d L+1 *The definitive answer to whether you should take Calculus, Statistics and Linear Algebra in college
  5. What's a GPU “A Graphics Processing Unit (GPU) is a

    specialized electronic circuit designed to rapidly manipulate and alter memory to accelerate the creation of images in a frame buffer intended for output to a display.”
  6. Pretty Pictures Require Lots of Arithmetic • $1,723 for the

    Intel Core i7-6950x: 10 cores, 1.12 TFLOPS • $1,200 for the NVIDIA GTX Titan XP, 56 cores, 10.8 TFLOPS • $600 for the NVIDIA GTX 1080, 40 cores, 8.9 TFLOPS • $500 for the AMD R9 Fury X, 64 cores, 8.6 TFLOPS
  7. Product Recommendations Also Require Lots of Arithmetic (2014) What are

    people who bought items A, B, C...Z most likely to purchase next? Traditionally addressed with variants of Matrix Factorization, Logistic Regression, Naive Bayes, etc...
  8. Large Output Layers, Small Hidden Layers Output (10K-10M) Input (10K-10M)

    Hidden (100-1K) Existing frameworks were not designed to handle neural networks with input (purchase history) and output (recommendations) layers 10K to 10M units wide because…
  9. This Is A Huge Sparse Data Problem • Uncompressed sparse

    data either eats a lot of memory or it eats a lot of bandwidth uploading it to the GPU • Naively running networks with uncompressed sparse data leads to lots of multiplications of zero and/or by zero. This wastes memory, power, and time • Product Recommendation Networks can have billions of parameters that cannot fit in a single GPU so summarizing...
  10. Framework Requirements (2014) • Efficient support for large input and

    output layers • Efficient handling of sparse data (i.e. don't store zero) • Automagic multi-GPU support for large networks and scaling • Avoids multiplying zero and/or by zero • <24 hours training and recommendations cycle • Human-readable descriptions of networks (API)
  11. DSSTNE: Deep Sparse Scalable Tensor Network Engine* • A Neural

    Network framework released into OSS by Amazon in May of 2016 • Optimized for large sparse data problems and fully connected layers • Extremely efficient model-parallel multi-GPU support • ~6x faster than TensorFlow on such datasets (and that's just on one GTX Titan X (Maxwell), ~15x faster using 4 of them) • 100% Deterministic Execution #reproducibilitymatters #noASGD • Full SM 3.x, 5.x, and 6.x support (Kepler or better GPUs) • Distributed training support OOTB (~20 lines of MPI *”Destiny”
  12. Key Features • Stores networks and data sets in NetCDF

    format with optional HDF5 support • Multi-GPU handled with MPI and Interprocess P2P copies • Unapologetic emphasis on fully-connected networks • Dependencies are C++11, CUDA 7.x+, netcdf, a C++11-aware MPI library, and libjsoncpp (coming soon: cuDNN) • There are no computational shortcuts here, all we're doing is avoiding multiplying by zero and storing zeroes
  13. Describes Neural Networks As JSON Objects { "Version" : 0.7,

    "Name" : "AE", "Kind" : "FeedForward", "SparsenessPenalty" : { "p" : 0.5, "beta" : 2.0 }, "ShuffleIndices" : false, "Denoising" : { "p" : 0.2 }, "ScaledMarginalCrossEntropy" : { "oneTarget" : 1.0, "zeroTarget" : 0.0, "oneScale" : 1.0, "zeroScale" : 1.0 }, "Layers" : [ { "Name" : "Input", "Kind" : "Input", "N" : "auto", "DataSet" : "input", "Sparse" : true }, { "Name" : "Hidden", "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 128, "Activation" : "Sigmoid", "Sparse" : true }, { "Name" : "Output", "Kind" : "Output", "Type" : "FullyConnected", "DataSet" : "output", "N" : "auto", "Activation" : "Sigmoid", "Sparse" : true } ], "ErrorFunction" : "ScaledMarginalCrossEntropy" }
  14. AlexNet As A JSON Object* *Accidentally similar to Andrej Karpathy's

    ConvnetJS framework { "Version" : 0.81, "Name" : "AlexNet", "Kind" : "FeedForward", "LocalResponseNormalization" : { "k" : 2, "n" : 5, "alpha" : 0.0001, "beta" : 0.75 }, "Layers" : [ { "Kind" : "Input", "Type" : "Convolutional", "N" : "auto", "DataSet" : "input"}, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 96, "Kernel" : [11, 11], "KernelStride" : [4, 4], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "LRN" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "Max", "Kernel" : [3, 3], "KernelStride" : [2, 2]}, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 256, "Kernel" : [5, 5], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "LRN" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "Max", "Kernel" : [3, 3], "KernelStride" : [2, 2] }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 384, "Kernel" : [3, 3], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 384, "Kernel" : [3, 3], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 256, "Kernel" : [3, 3], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "Max", "Kernel" : [3, 3], "KernelStride" : [2, 2] }, { "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 4096, "Activation" : "Relu", "pDropout" : 0.5 }, { "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 4096, "Activation" : "Relu", "pDropout" : 0.5 }, { "Kind" : "Output", "Type" : "FullyConnected", "N" : "auto", "DataSet" : "output", "Activation" : "SoftMax" } ], "ErrorFunction" : "CrossEntropy" }
  15. Human-Readable Doesn't Suck.. • name: "AlexNet" • layer { •

    name: "data" • type: "Input" • top: "data" • input_param { shape: { dim: 10 dim: 3 dim: 227 dim: 227 } } • } • layer { • name: "conv1" TLDR: 278 Lines of Code for AlexNet in Caffe...
  16. JSON API Is Just An Interface to DSSTNE struct NNNetworkDescriptor

    { string _name; // Optional name for neural network NNNetwork::Kind _kind; // Either AutoEncoder or FeedForward (default) ErrorFunction _errorFunction; // Error function for training vector<NNLayerDescriptor> _vLayerDescriptor; // Vector containing neural network layers DSSTNE's Engine is API-Agnostic
  17. Amazon's Definition Of Sparsity • 0.01% to 0.1% Density, far

    lower than the optimal sparsity for the cuSparse library (too cuSlow) • Sparse data stored in CSR format (Index, value) or just indices • Sparse SGEMM is 5-20x faster than a full SGEMM depending on density (ultimately memory-limited) • Sparse input layers are nearly “free”
  18. Sparse Neural Network Training* X L+1 = X L *

    W L→L+1 DW = XT L * d L+1 *Sparse output layers are easy (exercise for the listener)
  19. Sparse DW = XT L * d L+1 • Need

    to transpose X L matrix in parallel • This is easy to do with atomic ops • But the transpose ordering is not deterministic, floating point math is not associative (A + B + C) != (C + A + B) • Solution: use 64-bit fixed point summation because fixed point accumulation is associative (A + B + C) == (C + A + B) • 64-bit fixed point adds with are 32-bit instructions
  20. Model Parallel vs Data Parallel • Amazon Product Categories range

    from 10K to 10M items • Amazon Catalog is billions of items • GPUs have up to 12 (2015) $ I mean 24 (2016) $$ oops I mean 32 GB (2016) $$$ of memory • All the interesting problems need >12 GB of memory • Data Parallel Implementation unacceptably slow (GBs of weight gradients)
  21. “Automagic” Model Parallel • Uses the same JSON Object •

    1 GPU/process because reasons(tm) and simplicity • DSSTNE Engine automatically distributes the neural network based on the number of processes running ./trainer (serial job, 1 GPU) mpirun -np 3 ./trainer (model parallel, 3 GPUs) mpirun -np n ./predictor (model parallel, n GPUs)
  22. One Weird Trick For Model Parallelism To parallelize an SGEMM

    operation, one shards the input data across all N GPUs (N = 4 here) W1 X L X L3 X L2 X L1 X L4
  23. Two Ways To Shard The Weights W1 W W 3

    W 2 W 1 W 4 W W 1 W 4 W 3 W 2
  24. Output Layer Larger Than Input Layer? allGather* Input Layer Data

    Then SGEMM W1 W 1 W1 X L3 X L2 X L1 X L4 * W1 X L+1 1 = X L X L+1 *Using custom 2d allGather code, not NCCL/MPI
  25. Input Layer Larger Than Output Layer? SGEMM Then Reduce Outputs*

    X L X L+1 W1 X L1 X L+1 1 W 1 = * *Using custom 2D partial reduction code which is also O(n)
  26. Yes Yes But How Good Are The Recommendations? • This

    is a strange question IMO • DSSTNE runs the same mathematics as everyone else • Amazon OSSed the framework, not the actual networks, and definitely not how they prepare customer purchase histories • So for a surrogate, let's use the binary prediction of a random 80/20 split of the MovieLens 20M dataset • Competing numbers provided by Saul Vargas
  27. MovieLens 20M DSSTNE { "Version" : 0.8, "Name" : "AIV

    NNC", "Kind" : "FeedForward", "ShuffleIndices" : false, "ScaledMarginalCrossEntropy" : { "oneTarget" : 1.0, "zeroTarget" : 0.0, "oneScale" : 1.0, "zeroScale" : 1.0 }, "Layers" : [ { "Name" : "Input", "Kind" : "Input", "N" : "auto", "DataSet" : "input", "Sparse" : true }, { "Name" : "Hidden1", "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 1536, "Activation" : "Relu", "Sparse" : false, "pDropout" : 0.37, "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.01 } }, { "Name" : "Hidden2", "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 1536, "Activation" : "Relu", "Sparse" : false, "pDropout" : 0.37, "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.01 } }, { "Name" : "Hidden3", "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 1536, "Activation" : "Relu", "Sparse" : false, "pDropout" : 0.37, "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.01 } }, { "Name" : "Output", "Kind" : "Output", "Type" : "FullyConnected", "DataSet" : "output", "N" : "auto", "Activation" : "Sigmoid", "Sparse" : true , "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.01, "Bias" : -10.2 }} ], "ErrorFunction" : "ScaledMarginalCrossEntropy" }
  28. Raw Performance 0 2 4 6 8 10 12 0

    0.1 0.2 0.3 0.4 0.5 0.6 P@K K
  29. AWS Recommendations at Scale • AWS released a blog by

    Kiuk Chung on how to perform product recommendations with DSSTNE (and other frameworks BTW) and Spark to experiment with deep learning for product recommendations at scale • This is the Amazon Deep Learning Recommendations System minus the secret sauce networks, hyperparameters, and private customer data
  30. AlexNet in DSSTNE { "Version" : 0.81, "Name" : "AlexNet",

    "Kind" : "FeedForward", "LocalResponseNormalization" : { "k" : 2, "n" : 5, "alpha" : 0.0001, "beta" : 0.75 }, "Layers" : [ { "Kind" : "Input", "Type" : "Convolutional", "N" : "auto", "DataSet" : "input"}, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 96, "Kernel" : [11, 11], "KernelStride" : [4, 4], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "LRN" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "Max", "Kernel" : [3, 3], "KernelStride" : [2, 2]}, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 256, "Kernel" : [5, 5], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "LRN" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "Max", "Kernel" : [3, 3], "KernelStride" : [2, 2] }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 384, "Kernel" : [3, 3], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 384, "Kernel" : [3, 3], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 256, "Kernel" : [3, 3], "Activation" : "Relu" }, { "Kind" : "Hidden", "Type" : "Pooling", "Function" : "Max", "Kernel" : [3, 3], "KernelStride" : [2, 2] }, { "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 4096, "Activation" : "Relu", "pDropout" : 0.5 }, { "Kind" : "Hidden", "Type" : "FullyConnected", "N" : 4096, "Activation" : "Relu", "pDropout" : 0.5 }, { "Kind" : "Output", "Type" : "FullyConnected", "N" : "auto", "DataSet" : "output", "Activation" : "SoftMax" } ], "ErrorFunction" : "CrossEntropy" }
  31. Unfortunately... • ImageNet is ~200 GB of Data • GPUs

    have about 12 GB of data • Titan XP was supposed to have solved this with unified memory • But that turns out to be a disappointing work in progress for now • So we'll need something a lot smaller • And then we'll need to stream data old-school style
  32. CIFAR-100 • 50,000 32x32x3 images classified into 100 categories •

    About 150 MB of data, or *perfect* • State of the art prediction accuracy is a P@1of 70-80%
  33. Strawman CNN { "Version" : 0.81, "Name" : "CIFAR-100", "Kind"

    : "FeedForward", "LocalResponseNormalization" : "Layers" : [ { "Kind" : "Input", "Type" : "Convolutional", "N" : "auto", "DataSet" : "input", "Name" : "Input"}, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 96, "Kernel" : [3, 3], "KernelStride" : [1, 1], "Activation" : "Relu", "pDropout" : 0.2, "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.04 , "Bias" : 0.0 } }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 96, "Kernel" : [3, 3], "KernelStride" : [1, 1], "Activation" : "Relu", "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.04 , "Bias" : 0.0 } }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 96, "Kernel" : [3, 3], "KernelStride" : [2, 2], "Activation" : "Relu", "pDropout" : 0.5, "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.04 , "Bias" : 0.1 } }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 192, "Kernel" : [3, 3], "KernelStride" : [1, 1], "Activation" : "Relu", "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.04 , "Bias" : 0.0 } }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 192, "Kernel" : [3, 3], "KernelStride" : [1, 1], "Activation" : "Relu", "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.04 , "Bias" : 0.0 } }, { "Kind" : "Hidden", "Type" : "Convolutional", "N" : 192, "Kernel" : [3, 3], "KernelStride" : [2, 2], "Activation" : "Relu", "pDropout" : 0.5, "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.04 , "Bias" : 0.1 } }, { "Kind" : "Output", "Type" : "FullyConnected", "N" : "auto", "DataSet" : "output", "Name" : "Output", "Activation" : "SoftMax", "WeightInit" : { "Scheme" : "Gaussian", "Scale" : 0.01 , "Bias" : 0.0 } } ], "ErrorFunction" : "CrossEntropy" }
  34. Results • First attempt with buggy code hit 18% P@1

    • Second attempt with running code hit 38% P@1 • Third attempt with a tweaked network hit 46% P@1 • And that was Tuesday night
  35. Deep Learning For The 99% “You have to spend at

    least $24,000(US) to be able to run experiments on Atari or ImageNet competitively.” - Nando de Freitas
  36. P13N at Amazon • A $3000 GTX 880M Laptop •

    A $3000 GTX 980M Laptop • A $2000 Desktop with two $1000 GTX TitanX GPUs • A bunch of AWS time on GPUs from 2012 Developed DSSTNE and changed the way Amazon does recommendations with:
  37. What Do you REALLY Need? C P U 8 7

    4 7 P C I E S w it c h 8 7 4 7 P C I E S w it c h G P U 0 G P U 1 G P U 2 G P U 3
  38. Total Cost: $7,000 or less  Asus P9X79-E WS MB

    ($500) plus Intel Core-i7 4820 (Ivybridge) CPU ($320)  Asus X99-E WS MB ($520) plus Intel Core-i7 5930K (Haswell) CPU ($560)  4 Titan XP or (when they come) GTX 1080 Ti GPUs  44 TFLOPs for $7,000! (<<$24,000)  Was sold by NVIDIA as the Digits Dev Box for $15,000
  39. What if $24K is Burning A Hole in Your Pocket?

    C P U 8 7 9 6 P C I E S w it c h G P U 0 16x 16x 16x G P U 0 G P U 2 G P U 3 G P U 1 16x 16x G P U 0 16x 16x G P U 0 G P U 2 G P U 3 G P U 1 16x 16x 8 7 9 6 P C I E S w it c h 16x
  40. Or Do You Need A $149K DGX-1? • 85 TFLOPS

    FP32 (~10.6 TFLOPS per GPU) no FP16 for now • ~64 GB/s connected in a cube (N == 8) Significant reduction in communication costs, but is AlexNet communication-limited? Reduction: Gather: AllReduce: D * (N – 1) / N D * (N – 1) / N D * 2 * (N – 1) / N
  41. Are you data-parallel? • AlexNet has ~61M parameters • We'll

    assume a batch size of 128 and Soumith Chintala's training perf numbers for TitanX scaled up by ~1.6 to arrive at 2,884 images/s FP32 • 16 images/s/GPU at 2,884 images/s is ~5.5 ms • AllReducing 61M (244 MB) parameters at ~64 GB/s is ~6.7 ms (buried 5.5 ms of backprop for overlapping copy and compute) for a final result of 1.2 ms. • Using 12.5 GB/s P2P, this would take ~34 ms, $129K^H^H^H^H149K is a bargain! Such value! Very Deeply! So much Learnings!
  42. Alex Krizhevsky to the Rescue! (or are you model-parallel?) •

    AlexNet has ~61M parameters. ~4.3M of which are convolutional (data-parallel) and ~56.7M of which are fully-connected (model- parallel) • Fully connected layers at a batch size of 128 is ~1.7M neurons • P2P allReduce of 4.3M parameters takes ~2.4 ms • P2P gather/reduction of 1.7M neurons is ~0.5 ms • 2.9 ms is << 5.5 ms so once again it's free(tm) • It's also faster than NVLINK data-parallel… • NVLINK model-parallel would of course win here but it doesn't exist...
  43. DSSTNE RoadMap • Finish CNN/Pooling/LSTM support • Compile DSSTNE under

    Radeon Open Compute (ROC) and truly fulfill the promise of “CUDA Everywhere(tm)”* • Don't port your code to OpenCL, port your processor to CUDA and get instant access to all existing OSS CUDA applications • Provide a Python API through Python extensions (section 2.7.12) • Caffe/TensorFlow import • Automagic streaming of models and data on SM 6.x and up *https://github.com/RadeonOpenCompute
  44. Summary • DSSTNE's automagic model-parallel training is a big win

    • DSSTNE's efficient sparse data processing is a big win • Both of these required bespoke GPU code as will filling in the rest of the road map • AWS has released all the tools for using DSSTNE for recommendations at scale • Torch and Caffe have recently improved their sparse data support • CNN/RNN support is a work-in-progress