Skip to content
/ sona Public

Spark On Angel, arming Spark with a powerful Parameter Server, which enable Spark to train very big models

License

Notifications You must be signed in to change notification settings

Angel-ML/sona

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SONA Overview

Spark On Angel (SONA), arming Spark with a powerful Parameter Server, which enable Spark to train very big models

Similar to Spark MLlib, Spark on Angel is a standalone machine learning library built on Spark (yet it does not rely on Spark MLlib, Figure 1). SONA was based on RDD APIs and only included model training step in previous versions. In Angel 3.0, we introduce various new features to SONA:

  • Integration of feature engineering into SONA. Instead of simply borrowing Spark’s feature engineering operators, we add support for long index vector to all the operators to enable training of high dimensional sparse models.
  • Seamless connection with automatic hyperparameter tuning.
  • Spark-fashion APIs that introduce no cost for Spark users to switch to Angel.
  • Support for two new data formats: LibFFM and Dummy.
sona_fig00
Figure 1: SONA is a another machine learning & graph library on Spark Core

Figure 2 demonstrate the run time architecture of SONA.

sona_fig01
Figure 2: Architecture of SONA
  • There is a AngelClient on Spark driver. AngelClient is used to start Angel parameter server, create, load, initial and save matrix of the model.
  • There is a PSClient/PSAgent on Spark executor. Algorithms can pull parameter and push gradient through PSAgent
  • The Angel MLcore is running in each Task

Compared to previous version, a variety of new algorithms were added on SONA, such as Deep & Cross Network (DCN) and Attention Factorization Machines (AFM). As can be seen from Figure 2, there are significant differences between algorithms on SONA and those on Spark: algorithms on SONA are mainly designated for recommendations and graph embedding, while algorithms on Spark tend to be more general-purpose.

sona_fig02
Figure 3: Algorithms comparison of Spark and Angel

As a result, SONA can serve as a supplement of Spark

sparkonangel
Figure 4: Programming Example of SONA

Figure 4 provides an example of running distributed machine learning algorithms on SONA, including following steps:

  • Start parameter server at the beginning and stop it in the end.
  • Load training and test data as Spark DataFrame.
  • Define an Angel model and set parameters in Spark fashion. In this example, the algorithm is defined as a computing graph via JSON.
  • Use “fit” method to train the model.
  • Use “evaluate” method to evaluate the trained model.

Quick Start

SONA supports three types of runtime models: YARN, K8s and Local. The local mode enable it easy to debug. sona quick start

Algorithms

Deployment

Support

  • QQ account: 20171688

References

Other Resources

About

Spark On Angel, arming Spark with a powerful Parameter Server, which enable Spark to train very big models

Resources

License

Stars

Watchers

Forks

Packages

No packages published