mllib: scalable machine learning on sparkrezab/sparkworkshop/slides/...spark sql + mllib // data can...
TRANSCRIPT
![Page 1: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/1.jpg)
MLlib: Scalable Machine Learning on Spark
Xiangrui Meng
1
Collaborators: Ameet Talwalkar, Evan Sparks, Virginia Smith, Xinghao Pan, Shivaram Venkataraman, Matei Zaharia, Rean Griffith, John Duchi, Joseph Gonzalez, Michael Franklin, Michael I. Jordan, Tim Kraska, etc.
![Page 2: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/2.jpg)
What is MLlib?
2
![Page 3: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/3.jpg)
What is MLlib?
MLlib is a Spark subproject providing machine learning primitives:
• initial contribution from AMPLab, UC Berkeley
• shipped with Spark since version 0.8
• 33 contributors
3
![Page 4: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/4.jpg)
What is MLlib?Algorithms:!
• classification: logistic regression, linear support vector machine (SVM), naive Bayes
• regression: generalized linear regression (GLM)
• collaborative filtering: alternating least squares (ALS)
• clustering: k-means
• decomposition: singular value decomposition (SVD), principal component analysis (PCA)
4
![Page 5: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/5.jpg)
Why MLlib?
5
![Page 6: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/6.jpg)
scikit-learn?
Algorithms:!
• classification: SVM, nearest neighbors, random forest, …
• regression: support vector regression (SVR), ridge regression, Lasso, logistic regression, …!
• clustering: k-means, spectral clustering, …
• decomposition: PCA, non-negative matrix factorization (NMF), independent component analysis (ICA), …
6
![Page 7: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/7.jpg)
Mahout?
Algorithms:!
• classification: logistic regression, naive Bayes, random forest, …
• collaborative filtering: ALS, …
• clustering: k-means, fuzzy k-means, …
• decomposition: SVD, randomized SVD, …
7
![Page 8: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/8.jpg)
Vowpal Wabbit?H2O?
R?MATLAB?
Mahout?
Weka?scikit-learn?
LIBLINEAR?
8
![Page 9: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/9.jpg)
Why MLlib?
9
![Page 10: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/10.jpg)
• It is built on Apache Spark, a fast and general engine for large-scale data processing.
• Run programs up to 100x faster than Hadoop MapReduce in memory, or 10x faster on disk.
• Write applications quickly in Java, Scala, or Python.
Why MLlib?
10
![Page 11: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/11.jpg)
Gradient descent
val points = spark.textFile(...).map(parsePoint).cache() var w = Vector.zeros(d) for (i <- 1 to numIterations) { val gradient = points.map { p => (1 / (1 + exp(-p.y * w.dot(p.x)) - 1) * p.y * p.x ).reduce(_ + _) w -= alpha * gradient }
w w � ↵ ·nX
i=1
g(w;xi, yi)
11
![Page 12: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/12.jpg)
// Load and parse the data. val data = sc.textFile("kmeans_data.txt") val parsedData = data.map(_.split(‘ ').map(_.toDouble)).cache() "// Cluster the data into two classes using KMeans. val clusters = KMeans.train(parsedData, 2, numIterations = 20) "// Compute the sum of squared errors. val cost = clusters.computeCost(parsedData) println("Sum of squared errors = " + cost)
k-means (scala)
12
![Page 13: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/13.jpg)
k-means (python)# Load and parse the data data = sc.textFile("kmeans_data.txt") parsedData = data.map(lambda line: array([float(x) for x in line.split(' ‘)])).cache() "# Build the model (cluster the data) clusters = KMeans.train(parsedData, 2, maxIterations = 10, runs = 1, initialization_mode = "kmeans||") "# Evaluate clustering by computing the sum of squared errors def error(point): center = clusters.centers[clusters.predict(point)] return sqrt(sum([x**2 for x in (point - center)])) "cost = parsedData.map(lambda point: error(point)) .reduce(lambda x, y: x + y) print("Sum of squared error = " + str(cost))
13
![Page 14: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/14.jpg)
Dimension reduction+ k-means
// compute principal components val points: RDD[Vector] = ... val mat = RowRDDMatrix(points) val pc = mat.computePrincipalComponents(20) "// project points to a low-dimensional space val projected = mat.multiply(pc).rows "// train a k-means model on the projected data val model = KMeans.train(projected, 10)
![Page 15: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/15.jpg)
Collaborative filtering// Load and parse the data val data = sc.textFile("mllib/data/als/test.data") val ratings = data.map(_.split(',') match { case Array(user, item, rate) => Rating(user.toInt, item.toInt, rate.toDouble) }) "// Build the recommendation model using ALS val model = ALS.train(ratings, 1, 20, 0.01) "// Evaluate the model on rating data val usersProducts = ratings.map { case Rating(user, product, rate) => (user, product) } val predictions = model.predict(usersProducts)
15
![Page 16: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/16.jpg)
Why MLlib?• It ships with Spark as
a standard component.
16
![Page 17: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/17.jpg)
Out for dinner?!
• Search for a restaurant and make a reservation.
• Start navigation.
• Food looks good? Take a photo and share.
17
![Page 18: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/18.jpg)
Why smartphone?
Out for dinner?!
• Search for a restaurant and make a reservation. (Yellow Pages?)
• Start navigation. (GPS?)
• Food looks good? Take a photo and share. (Camera?)
18
![Page 19: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/19.jpg)
Why MLlib?
A special-purpose device may be better at one aspect than a general-purpose device. But the cost of context switching is high: • different languages or APIs
• different data formats
• different tuning tricks
19
![Page 20: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/20.jpg)
Spark SQL + MLlib// Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable = sql(""" SELECT e.action, u.age, u.latitude, u.longitude FROM Users u JOIN Events e ON u.userId = e.userId""") "// Since `sql` returns an RDD, the results of the above // query can be easily used in MLlib. val training = trainingTable.map { row => val features = Vectors.dense(row(1), row(2), row(3)) LabeledPoint(row(0), features) } "val model = SVMWithSGD.train(training)
![Page 21: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/21.jpg)
Streaming + MLlib
// collect tweets using streaming "// train a k-means model val model: KMmeansModel = ... "// apply model to filter tweets val tweets = TwitterUtils.createStream(ssc, Some(authorizations(0))) val statuses = tweets.map(_.getText) val filteredTweets = statuses.filter(t => model.predict(featurize(t)) == clusterNumber) "// print tweets within this particular cluster filteredTweets.print()
![Page 22: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/22.jpg)
GraphX + MLlib// assemble link graph val graph = Graph(pages, links) val pageRank: RDD[(Long, Double)] = graph.staticPageRank(10).vertices "// load page labels (spam or not) and content features val labelAndFeatures: RDD[(Long, (Double, Seq((Int, Double)))] = ... val training: RDD[LabeledPoint] = labelAndFeatures.join(pageRank).map { case (id, ((label, features), pageRank)) => LabeledPoint(label, Vectors.sparse(features ++ (1000, pageRank)) } "// train a spam detector using logistic regression val model = LogisticRegressionWithSGD.train(training)
![Page 23: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/23.jpg)
Why MLlib?• Spark is a general-purpose big data platform.
• Runs in standalone mode, on YARN, EC2, and Mesos, also on Hadoop v1 with SIMR.
• Reads from HDFS, S3, HBase, and any Hadoop data source.
• MLlib is a standard component of Spark providing machine learning primitives on top of Spark.
• MLlib is also comparable to or even better than other libraries specialized in large-scale machine learning.
23
![Page 24: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/24.jpg)
Why MLlib?• Spark is a general-purpose big data platform.
• Runs in standalone mode, on YARN, EC2, and Mesos, also on Hadoop v1 with SIMR.
• Reads from HDFS, S3, HBase, and any Hadoop data source.
• MLlib is a standard component of Spark providing machine learning primitives on top of Spark.
• MLlib is also comparable to or even better than other libraries specialized in large-scale machine learning.
24
![Page 25: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/25.jpg)
Why MLlib?
• Scalability
• Performance
• User-friendly APIs
• Integration with Spark and its other components
25
![Page 26: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/26.jpg)
Logistic regression
26
![Page 27: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/27.jpg)
Logistic regression - weak scaling
• Full dataset: 200K images, 160K dense features. • Similar weak scaling. • MLlib within a factor of 2 of VW’s wall-clock time.
MLbase VW Matlab0
1000
2000
3000
4000
wal
ltim
e (s
)
n=6K, d=160Kn=12.5K, d=160Kn=25K, d=160Kn=50K, d=160Kn=100K, d=160Kn=200K, d=160K
MLlib
MLbase VW Matlab0
1000
2000
3000
4000
wal
ltim
e (s
)
n=12K, d=160Kn=25K, d=160Kn=50K, d=160Kn=100K, d=160Kn=200K, d=160K
Fig. 5: Walltime for weak scaling for logistic regression.
0 5 10 15 20 25 300
2
4
6
8
10
rela
tive
wal
ltim
e
# machines
MLbaseVWIdeal
Fig. 6: Weak scaling for logistic regression
MLbase VW Matlab0
200
400
600
800
1000
1200
1400
wal
ltim
e (s
)
1 Machine2 Machines4 Machines8 Machines16 Machines32 Machines
Fig. 7: Walltime for strong scaling for logistic regression.
0 5 10 15 20 25 300
5
10
15
20
25
30
35
# machines
spee
dup
MLbaseVWIdeal
Fig. 8: Strong scaling for logistic regression
with respect to computation. In practice, we see comparablescaling results as more machines are added.
In MATLAB, we implement gradient descent instead ofSGD, as gradient descent requires roughly the same numberof numeric operations as SGD but does not require an innerloop to pass over the data. It can thus be implemented in a’vectorized’ fashion, which leads to a significantly more favor-able runtime. Moreover, while we are not adding additionalprocessing units to MATLAB as we scale the dataset size, weshow MATLAB’s performance here as a reference for traininga model on a similarly sized dataset on a single multicoremachine.
Results: In our weak scaling experiments (Figures 5 and6), we can see that our clustered system begins to outperformMATLAB at even moderate levels of data, and while MATLABruns out of memory and cannot complete the experiment onthe 200K point dataset, our system finishes in less than 10minutes. Moreover, the highly specialized VW is on average35% faster than our system, and never twice as fast. Thesetimes do not include time spent preparing data for input inputfor VW, which was significant, but expect that they’d be aone-time cost in a fully deployed environment.
From the perspective of strong scaling (Figures 7 and 8),our solution actually outperforms VW in raw time to train amodel on a fixed dataset size when using 16 and 32 machines,and exhibits stronger scaling properties, much closer to thegold standard of linear scaling for these algorithms. We areunsure whether this is due to our simpler (broadcast/gather)communication paradigm, or some other property of the sys-tem.
System Lines of CodeMLbase 32
GraphLab 383Mahout 865
MATLAB-Mex 124MATLAB 20
TABLE II: Lines of code for various implementations of ALS
B. Collaborative Filtering: Alternating Least Squares
Matrix factorization is a technique used in recommendersystems to predict user-product associations. Let M 2 Rm⇥n
be some underlying matrix and suppose that only a smallsubset, ⌦(M), of its entries are revealed. The goal of matrixfactorization is to find low-rank matrices U 2 Rm⇥k andV 2 Rn⇥k, where k ⌧ n,m, such that M ⇡ UV
T .Commonly, U and V are estimated using the following bi-convex objective:
min
U,V
X
(i,j)2⌦(M)
(Mij � U
Ti Vj)
2+ �(||U ||2F + ||V ||2F ) . (2)
Alternating least squares (ALS) is a widely used method formatrix factorization that solves (2) by alternating betweenoptimizing U with V fixed, and V with U fixed. ALS iswell-suited for parallelism, as each row of U can be solvedindependently with V fixed, and vice-versa. With V fixed, theminimization problem for each row ui is solved with the closedform solution. where u
⇤i 2 Rk is the optimal solution for the
i
th row vector of U , V⌦i is a sub-matrix of rows vj such thatj 2 ⌦i, and Mi⌦i is a sub-vector of observed entries in the
MLlib
27
![Page 28: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/28.jpg)
Logistic regression - strong scaling
• Fixed Dataset: 50K images, 160K dense features. • MLlib exhibits better scaling properties. • MLlib is faster than VW with 16 and 32 machines.
MLbase VW Matlab0
1000
2000
3000
4000
wal
ltim
e (s
)
n=12K, d=160Kn=25K, d=160Kn=50K, d=160Kn=100K, d=160Kn=200K, d=160K
Fig. 5: Walltime for weak scaling for logistic regression.
0 5 10 15 20 25 300
2
4
6
8
10
rela
tive
wal
ltim
e
# machines
MLbaseVWIdeal
Fig. 6: Weak scaling for logistic regression
MLbase VW Matlab0
200
400
600
800
1000
1200
1400
wal
ltim
e (s
)
1 Machine2 Machines4 Machines8 Machines16 Machines32 Machines
Fig. 7: Walltime for strong scaling for logistic regression.
0 5 10 15 20 25 300
5
10
15
20
25
30
35
# machines
spee
dup
MLbaseVWIdeal
Fig. 8: Strong scaling for logistic regression
with respect to computation. In practice, we see comparablescaling results as more machines are added.
In MATLAB, we implement gradient descent instead ofSGD, as gradient descent requires roughly the same numberof numeric operations as SGD but does not require an innerloop to pass over the data. It can thus be implemented in a’vectorized’ fashion, which leads to a significantly more favor-able runtime. Moreover, while we are not adding additionalprocessing units to MATLAB as we scale the dataset size, weshow MATLAB’s performance here as a reference for traininga model on a similarly sized dataset on a single multicoremachine.
Results: In our weak scaling experiments (Figures 5 and6), we can see that our clustered system begins to outperformMATLAB at even moderate levels of data, and while MATLABruns out of memory and cannot complete the experiment onthe 200K point dataset, our system finishes in less than 10minutes. Moreover, the highly specialized VW is on average35% faster than our system, and never twice as fast. Thesetimes do not include time spent preparing data for input inputfor VW, which was significant, but expect that they’d be aone-time cost in a fully deployed environment.
From the perspective of strong scaling (Figures 7 and 8),our solution actually outperforms VW in raw time to train amodel on a fixed dataset size when using 16 and 32 machines,and exhibits stronger scaling properties, much closer to thegold standard of linear scaling for these algorithms. We areunsure whether this is due to our simpler (broadcast/gather)communication paradigm, or some other property of the sys-tem.
System Lines of CodeMLbase 32
GraphLab 383Mahout 865
MATLAB-Mex 124MATLAB 20
TABLE II: Lines of code for various implementations of ALS
B. Collaborative Filtering: Alternating Least Squares
Matrix factorization is a technique used in recommendersystems to predict user-product associations. Let M 2 Rm⇥n
be some underlying matrix and suppose that only a smallsubset, ⌦(M), of its entries are revealed. The goal of matrixfactorization is to find low-rank matrices U 2 Rm⇥k andV 2 Rn⇥k, where k ⌧ n,m, such that M ⇡ UV
T .Commonly, U and V are estimated using the following bi-convex objective:
min
U,V
X
(i,j)2⌦(M)
(Mij � U
Ti Vj)
2+ �(||U ||2F + ||V ||2F ) . (2)
Alternating least squares (ALS) is a widely used method formatrix factorization that solves (2) by alternating betweenoptimizing U with V fixed, and V with U fixed. ALS iswell-suited for parallelism, as each row of U can be solvedindependently with V fixed, and vice-versa. With V fixed, theminimization problem for each row ui is solved with the closedform solution. where u
⇤i 2 Rk is the optimal solution for the
i
th row vector of U , V⌦i is a sub-matrix of rows vj such thatj 2 ⌦i, and Mi⌦i is a sub-vector of observed entries in the
MLlib
MLbase VW Matlab0
1000
2000
3000
4000
wal
ltim
e (s
)
n=12K, d=160Kn=25K, d=160Kn=50K, d=160Kn=100K, d=160Kn=200K, d=160K
Fig. 5: Walltime for weak scaling for logistic regression.
0 5 10 15 20 25 300
2
4
6
8
10
rela
tive
wal
ltim
e# machines
MLbaseVWIdeal
Fig. 6: Weak scaling for logistic regression
MLbase VW Matlab0
200
400
600
800
1000
1200
1400
wal
ltim
e (s
)
1 Machine2 Machines4 Machines8 Machines16 Machines32 Machines
Fig. 7: Walltime for strong scaling for logistic regression.
0 5 10 15 20 25 300
5
10
15
20
25
30
35
# machines
spee
dup
MLbaseVWIdeal
Fig. 8: Strong scaling for logistic regression
with respect to computation. In practice, we see comparablescaling results as more machines are added.
In MATLAB, we implement gradient descent instead ofSGD, as gradient descent requires roughly the same numberof numeric operations as SGD but does not require an innerloop to pass over the data. It can thus be implemented in a’vectorized’ fashion, which leads to a significantly more favor-able runtime. Moreover, while we are not adding additionalprocessing units to MATLAB as we scale the dataset size, weshow MATLAB’s performance here as a reference for traininga model on a similarly sized dataset on a single multicoremachine.
Results: In our weak scaling experiments (Figures 5 and6), we can see that our clustered system begins to outperformMATLAB at even moderate levels of data, and while MATLABruns out of memory and cannot complete the experiment onthe 200K point dataset, our system finishes in less than 10minutes. Moreover, the highly specialized VW is on average35% faster than our system, and never twice as fast. Thesetimes do not include time spent preparing data for input inputfor VW, which was significant, but expect that they’d be aone-time cost in a fully deployed environment.
From the perspective of strong scaling (Figures 7 and 8),our solution actually outperforms VW in raw time to train amodel on a fixed dataset size when using 16 and 32 machines,and exhibits stronger scaling properties, much closer to thegold standard of linear scaling for these algorithms. We areunsure whether this is due to our simpler (broadcast/gather)communication paradigm, or some other property of the sys-tem.
System Lines of CodeMLbase 32
GraphLab 383Mahout 865
MATLAB-Mex 124MATLAB 20
TABLE II: Lines of code for various implementations of ALS
B. Collaborative Filtering: Alternating Least Squares
Matrix factorization is a technique used in recommendersystems to predict user-product associations. Let M 2 Rm⇥n
be some underlying matrix and suppose that only a smallsubset, ⌦(M), of its entries are revealed. The goal of matrixfactorization is to find low-rank matrices U 2 Rm⇥k andV 2 Rn⇥k, where k ⌧ n,m, such that M ⇡ UV
T .Commonly, U and V are estimated using the following bi-convex objective:
min
U,V
X
(i,j)2⌦(M)
(Mij � U
Ti Vj)
2+ �(||U ||2F + ||V ||2F ) . (2)
Alternating least squares (ALS) is a widely used method formatrix factorization that solves (2) by alternating betweenoptimizing U with V fixed, and V with U fixed. ALS iswell-suited for parallelism, as each row of U can be solvedindependently with V fixed, and vice-versa. With V fixed, theminimization problem for each row ui is solved with the closedform solution. where u
⇤i 2 Rk is the optimal solution for the
i
th row vector of U , V⌦i is a sub-matrix of rows vj such thatj 2 ⌦i, and Mi⌦i is a sub-vector of observed entries in the
MLlib
28
![Page 29: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/29.jpg)
Collaborative filtering
29
![Page 30: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/30.jpg)
Collaborative filtering
• Recover a ra-ng matrix from a subset of its entries. ?
?
?
?
?
30
![Page 31: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/31.jpg)
ALS - wall-clock time
• Dataset: scaled version of Netflix data (9X in size). • Cluster: 9 machines. • MLlib is an order of magnitude faster than Mahout. • MLlib is within factor of 2 of GraphLab.
System Wall-‐clock /me (seconds)
MATLAB 15443
Mahout 4206
GraphLab 291
MLlib 481
31
![Page 32: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/32.jpg)
Implementation of k-means
Initialization:
• random
• k-means++
• k-means||
![Page 33: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/33.jpg)
Iterations:
• For each point, find its closest center.
• Update cluster centers.
Implementation of k-means
li = argminj
kxi � cjk22
cj =
Pi,li=j xjPi,li=j 1
![Page 34: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/34.jpg)
Implementation of k-meansThe points are usually sparse, but the centers are most likely to be dense. Computing the distance takes O(d) time. So the time complexity is O(n d k) per iteration. We don’t take any advantage of sparsity on the running time. However, we have
kx� ck22 = kxk22 + kck22 � 2hx, ci
Computing the inner product only needs non-zero elements. So we can cache the norms of the points and of the centers, and then only need the inner products to obtain the distances. This reduce the running time to O(nnz k + d k) per iteration. "However, is it accurate?
![Page 35: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/35.jpg)
Implementation of ALS
• broadcast everything • data parallel • fully parallel
35
![Page 36: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/36.jpg)
Alternating least squares (ALS)
36
![Page 37: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/37.jpg)
Broadcast everything• Master loads (small)
data file and initializes models.
• Master broadcasts data and initial models.
• At each iteration, updated models are broadcast again.
• Works OK for small data.
• Lots of communication overhead - doesn’t scale well. Master
Workers
Ratings
Movie!Factors
User!Factors
![Page 38: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/38.jpg)
Data parallel
• Workers load data
• Master broadcasts initial models
• At each iteration, updated models are broadcast again
• Much better scaling
• Works on large datasets
• Works well for smaller models. (low K)
MasterWorkers
Ratings
Movie!Factors
User!Factors
Ratings
Ratings
Ratings
38
![Page 39: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/39.jpg)
Fully parallel• Workers load data
• Models are instantiated at workers.
• At each iteration, models are shared via join between workers.
• Much better scalability.
• Works on large datasets
MasterWorkers
RatingsMovie!FactorsUser!Factors
RatingsMovie!FactorsUser!Factors
RatingsMovie!FactorsUser!Factors
RatingsMovie!FactorsUser!Factors
39
![Page 40: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/40.jpg)
Implementation of ALS
• broadcast everything • data parallel • fully parallel • block-wise parallel
• Users/products are partitioned into blocks and join is based on blocks instead of individual user/product.
40
![Page 41: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/41.jpg)
New features for v1.x• Sparse data
• Classification and regression tree (CART)
• SVD and PCA
• L-BFGS
• Model evaluation
• Discretization
41
![Page 42: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/42.jpg)
Contributors
Ameet Talwalkar, Andrew Tulloch, Chen Chao, Nan Zhu, DB Tsai, Evan Sparks, Frank Dai, Ginger Smith, Henry Saputra, Holden Karau, Hossein Falaki, Jey Kottalam, Cheng Lian, Marek Kolodziej, Mark Hamstra, Martin Jaggi, Martin Weindel, Matei Zaharia, Nick Pentreath, Patrick Wendell, Prashant Sharma, Reynold Xin, Reza Zadeh, Sandy Ryza, Sean Owen, Shivaram Venkataraman, Tor Myklebust, Xiangrui Meng, Xinghao Pan, Xusen Yin, Jerry Shao, Ryan LeCompte
42
![Page 43: MLlib: Scalable Machine Learning on Sparkrezab/sparkworkshop/slides/...Spark SQL + MLlib // Data can easily be extracted from existing sources, // such as Apache Hive. val trainingTable](https://reader034.vdocuments.mx/reader034/viewer/2022043000/5f76cb9cd7360f467144ca20/html5/thumbnails/43.jpg)
Interested?• Website: http://spark.apache.org
• Tutorials: http://ampcamp.berkeley.edu
• Spark Summit: http://spark-summit.org
• Github: https://github.com/apache/spark
• Mailing lists: [email protected] [email protected]
43