Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-4708

Make k-mean runs two/three times faster with dense/sparse sample

    XMLWordPrintableJSON

    Details

    • Type: Improvement
    • Status: Resolved
    • Priority: Major
    • Resolution: Implemented
    • Affects Version/s: None
    • Fix Version/s: 1.2.0
    • Component/s: MLlib
    • Labels:
      None
    • Target Version/s:

      Description

      Note that the usage of `breezeSquaredDistance` in `org.apache.spark.mllib.util.MLUtils.fastSquaredDistance` is in the critical path, and breezeSquaredDistance is slow. We should replace it with our own implementation.

      Here is the benchmark against mnist8m dataset.

      Before
      DenseVector: 70.04secs
      SparseVector: 59.05secs

      With this PR
      DenseVector: 30.58secs
      SparseVector: 21.14secs

        Attachments

          Activity

            People

            • Assignee:
              dbtsai DB Tsai
              Reporter:
              dbtsai DB Tsai
            • Votes:
              0 Vote for this issue
              Watchers:
              2 Start watching this issue

              Dates

              • Created:
                Updated:
                Resolved: