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

Project Matrix: Linear Models revisit and refactor



    • New Feature
    • Status: Open
    • Major
    • Resolution: Unresolved
    • 3.1.0, 3.2.0
    • None
    • ML, PySpark
    • None


      We had been refactoring linear models for a long time, and there still are some works in the future. After some discussions among huaxingao srowen weichenxu123 mengxr podongfeng , we decide to gather related works under a sub-project Matrix, it includes:

      1. Blockification (vectorization of vectors)
        • vectors are stacked into matrices, so that high-level BLAS can be used for better performance. (about ~3x faster on sparse datasets, up to ~18x faster on dense datasets, see SPARK-31783 for details).
        • Since 3.1.1, LoR/SVC/LiR/AFT supports blockification, and we need to blockify KMeans in the future.
      2. Standardization (virutal centering)
        • Existing impl of standardization in linear models does NOT center the vectors by removing the means, for the purpose of keeping dataset sparsity. However, this will cause feature values with small var be scaled to large values, and underlying solver like LBFGS can not efficiently handle this case. see SPARK-34448 for details.
        • If internal vectors are centered (like famous GLMNET), the convergence ratio will be better. In the case in SPARK-34448, the number of iteration to convergence will be reduced from 93 to 6. Moreover, the final solution is much more close to the one in GLMNET.
        • Luckily, we find a new way to virtually center the vectors without densifying the dataset. Good results had been observed in LoR, we will take it into account in other linear models.
      3. Initialization (To be discussed)
        • Initializing model coef with a given model, should be beneficial to: 1, convergence ratio (should reduce number of iterations); 2, model stability (may obtain a new solution more close to the previous one);
      4. Early Stopping (To be discussed)
        • we can compute the test error in the procedure (like tree models), and stop the training procedure if test error begin to increase;


            If you want to add other features in these models, please comment in the ticket.




            podongfeng zhengruifeng
            podongfeng zhengruifeng
            0 Vote for this issue
            3 Start watching this issue