MLlib: Machine Learning in Apache Spark
Not on the map. Why?
Abstract (arXiv)
Apache Spark is a popular open-source platform for large-scale data processing that is well-suited for iterative machine learning tasks. In this paper we present MLlib, Spark's open-source distributed machine learning library. MLlib provides efficient functionality for a wide range of learning settings and includes several underlying statistical, optimization, and linear algebra primitives. Shipped with Spark, MLlib supports several languages and provides a high-level API that leverages Spark's rich ecosystem to simplify the development of end-to-end machine learning pipelines. MLlib has experienced a rapid growth due to its vibrant open-source community of over 140 contributors, and includes extensive documentation to support further growth and to let users quickly get up to speed.
Key facts. Cited by 55 papers in the map (data through 14 Oct 2026). First posted on arXiv 26 May 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 20; 2023: 10; 2024: 3; 2025: 5; 2026: 3.
55 papers.
Sort:
By area:
- 2022Data Debugging with Shapley Importance over End-to-End Machine Learning Pipelinescited by 17
- 2021Towards Demystifying Serverless Machine Learning Trainingcited by 16
- 2022End-to-end Optimization of Machine Learning Prediction Queriescited by 16
- 2022Query Processing on Tensor Computation Runtimescited by 12
- 2022Serving and Optimizing Machine Learning Workflows on Heterogeneous Infrastructurescited by 12
- 2021A Survey on Deep Reinforcement Learning for Data Processing and Analyticscited by 10
- 2022NCVX: A General-Purpose Optimization Solver for Constrained Machine and Deep Learningcited by 7
- 2021Functional Collection Programming with Semi-Ring Dictionariescited by 6
- 2021NCVX: A User-Friendly and Scalable Package for Nonconvex Optimization in Machine Learningcited by 5
- 2022Optimization for Classical Machine Learning Problems on the GPUcited by 5
- 2022An Empirical Evaluation of Flow Based Programming in the Machine Learning Deployment Contextcited by 5
- 2021Declarative Machine Learning Systemscited by 4
38 more
- 2022A State Transition Model for Mobile Notifications via Survival Analysiscited by 4
- 2023Test & Evaluation Best Practices for Machine Learning-Enabled Systemscited by 4
- 2023KAXAI: An Integrated Environment for Knowledge Analysis and Explainable AIcited by 4
- 2024Intelligent Spark Agents: A Modular LangGraph Framework for Scalable, Visualized, and Enhanced Big Data Machine Learning Workflowscited by 4
- 2025Topic Over Source: The Key to Effective Data Mixing for Language Models Pre-trainingcited by 4
- 2021Extending Isolation Forest for Anomaly Detection in Big Data via K-Meanscited by 3
- 2021Large-Scale News Classification using BERT Language Model: Spark NLP Approachcited by 3
- 2021Towards better data discovery and collection with flow-based programmingcited by 3
- 2022GCF: Generalized Causal Forest for Heterogeneous Treatment Effect Estimation in Online Marketplacecited by 3
- 2022Differential testing for machine learning: an analysis for classification algorithms beyond deep learningcited by 3
- 2022Streaming Encoding Algorithms for Scalable Hyperdimensional Computingcited by 3
- 2023What are the Machine Learning best practices reported by practitioners on Stack Exchange?cited by 3
- 2024Mew: Multiplexed Immunofluorescence Image Analysis through an Efficient Multiplex Networkcited by 3
- 2021PyKale: Knowledge-Aware Machine Learning from Multiple Sources in Pythoncited by 2
- 2021Decision-forest voting scheme for classification of rare classes in network intrusion detectioncited by 2
- 2022Stochastic Gradient Descent without Full Data Shufflecited by 2
- 2023Multi-Criteria Client Selection and Scheduling with Fairness Guarantee for Federated Learning Servicecited by 2
- 2021Evaluating Deep Learning in SystemML using Layer-wise Adaptive Rate Scaling(LARS) Optimizercited by 1
- 2022Fast Distributed k-Means with a Small Number of Roundscited by 1
- 2022A comprehensive survey on computational learning methods for analysis of gene expression datacited by 1
- 2022Improving the Learnability of Machine Learning APIs by Semi-Automated API Wrappingcited by 1
- 2023Safe Collaborative Filteringcited by 1
- 2023Towards General and Efficient Online Tuning for Sparkcited by 1
- 2023Efficient Hybrid Oversampling and Intelligent Undersampling for Imbalanced Big Data Classificationcited by 1
- 2024ASTM :Autonomous Smart Traffic Management System Using Artificial Intelligence CNN and LSTMcited by 1
- 2025Provable Model-Parallel Distributed Principal Component Analysis with Parallel Deflationcited by 1
- 2021Data Analytics and Machine Learning Methods, Techniques and Tool for Model-Driven Engineering of Smart IoT Servicescited by 0
- 2021Scalable Predictive Time-Series Analysis of COVID-19: Cases and Fatalitiescited by 0
- 2021A Survey on Proactive Customer Care: Enabling Science and Steps to Realize itcited by 0
- 2022Semantic of Cloud Computing services for Time Series workflowscited by 0
- 2022Distributed data analyticscited by 0
- 2022On some studies of Fraud Detection Pipeline and related issues from the scope of Ensemble Learning and Graph-based Learningcited by 0
- 2022NumS: Scalable Array Programming for the Cloudcited by 0
- 2022Benchmarking Apache Spark and Hadoop MapReduce on Big Data Classificationcited by 0
- 2023CMLCompiler: A Unified Compiler for Classical Machine Learningcited by 0
- 2023What Causes Exceptions in Machine Learning Applications? Mining Machine Learning-Related Stack Traces on Stack Overflowcited by 0
- 2023The Landscape of Modern Machine Learning: A Review of Machine, Distributed and Federated Learningcited by 0
- 2025Harmonia: End-to-End RAG Serving Optimizationcited by 0
50 most cited of 55 All 55 (JSON)
- 2026Spark Policy Toolkit: Semantic Contracts and Scalable Execution for Policy Learning in Spark27 Apr 2026
- 2026Vectorized Dynamic Histograms for Sparse Oblique Forests27 Feb 2026
- 2026Spark-LLM-Eval: A Distributed Framework for Statistically Rigorous Large Language Model Evaluation18 Jan 2026
- 2025High-Dimensional Data Processing: Benchmarking Machine Learning and Deep Learning Architectures in Local and Distributed Environments11 Dec 2025
- 2025Industrial brain: a human-like autonomous neuro-symbolic cognitive decision-making system30 Jun 2025
- 2025Harmonia: End-to-End RAG Serving Optimization1 May 2025
- 2025Provable Model-Parallel Distributed Principal Component Analysis with Parallel Deflation24 Feb 2025
- 2025Topic Over Source: The Key to Effective Data Mixing for Language Models Pre-training24 Feb 2025
- 2024Intelligent Spark Agents: A Modular LangGraph Framework for Scalable, Visualized, and Enhanced Big Data Machine Learning Workflows2 Dec 2024
- 2024ASTM :Autonomous Smart Traffic Management System Using Artificial Intelligence CNN and LSTM14 Oct 2024
- 2024Mew: Multiplexed Immunofluorescence Image Analysis through an Efficient Multiplex Network25 Jul 2024
- 2023KAXAI: An Integrated Environment for Knowledge Analysis and Explainable AI30 Dec 2023
38 more
- 2023Multi-Criteria Client Selection and Scheduling with Fairness Guarantee for Federated Learning Service5 Dec 2023
- 2023The Landscape of Modern Machine Learning: A Review of Machine, Distributed and Federated Learning5 Dec 2023
- 2023Test & Evaluation Best Practices for Machine Learning-Enabled Systems10 Oct 2023
- 2023Efficient Hybrid Oversampling and Intelligent Undersampling for Imbalanced Big Data Classification9 Oct 2023
- 2023Towards General and Efficient Online Tuning for Spark5 Sep 2023
- 2023Safe Collaborative Filtering8 Jun 2023
- 2023What Causes Exceptions in Machine Learning Applications? Mining Machine Learning-Related Stack Traces on Stack Overflow25 Apr 2023
- 2023CMLCompiler: A Unified Compiler for Classical Machine Learning31 Jan 2023
- 2023What are the Machine Learning best practices reported by practitioners on Stack Exchange?25 Jan 2023
- 2022NCVX: A General-Purpose Optimization Solver for Constrained Machine and Deep Learning3 Oct 2022
- 2022Benchmarking Apache Spark and Hadoop MapReduce on Big Data Classification21 Sep 2022
- 2022Streaming Encoding Algorithms for Scalable Hyperdimensional Computing20 Sep 2022
- 2022Differential testing for machine learning: an analysis for classification algorithms beyond deep learning25 Jul 2022
- 2022A State Transition Model for Mobile Notifications via Survival Analysis7 Jul 2022
- 2022NumS: Scalable Array Programming for the Cloud28 Jun 2022
- 2022Stochastic Gradient Descent without Full Data Shuffle12 Jun 2022
- 2022End-to-end Optimization of Machine Learning Prediction Queries31 May 2022
- 2022Serving and Optimizing Machine Learning Workflows on Heterogeneous Infrastructures10 May 2022
- 2022On some studies of Fraud Detection Pipeline and related issues from the scope of Ensemble Learning and Graph-based Learning10 May 2022
- 2022An Empirical Evaluation of Flow Based Programming in the Machine Learning Deployment Context27 Apr 2022
- 2022Data Debugging with Shapley Importance over End-to-End Machine Learning Pipelines23 Apr 2022
- 2022Optimization for Classical Machine Learning Problems on the GPU30 Mar 2022
- 2022Improving the Learnability of Machine Learning APIs by Semi-Automated API Wrapping29 Mar 2022
- 2022Distributed data analytics26 Mar 2022
- 2022GCF: Generalized Causal Forest for Heterogeneous Treatment Effect Estimation in Online Marketplace21 Mar 2022
- 2022Query Processing on Tensor Computation Runtimes3 Mar 2022
- 2022A comprehensive survey on computational learning methods for analysis of gene expression data7 Feb 2022
- 2022Semantic of Cloud Computing services for Time Series workflows1 Feb 2022
- 2022Fast Distributed k-Means with a Small Number of Rounds31 Jan 2022
- 2021NCVX: A User-Friendly and Scalable Package for Nonconvex Optimization in Machine Learning27 Nov 2021
- 2021A Survey on Proactive Customer Care: Enabling Science and Steps to Realize it11 Oct 2021
- 2021A Survey on Deep Reinforcement Learning for Data Processing and Analytics10 Aug 2021
- 2021Towards better data discovery and collection with flow-based programming9 Aug 2021
- 2021Decision-forest voting scheme for classification of rare classes in network intrusion detection25 Jul 2021
- 2021Declarative Machine Learning Systems16 Jul 2021
- 2021Large-Scale News Classification using BERT Language Model: Spark NLP Approach14 Jul 2021
- 2021PyKale: Knowledge-Aware Machine Learning from Multiple Sources in Python17 Jun 2021
- 2021Towards Demystifying Serverless Machine Learning Training17 May 2021
50 newest of 55 All 55 (JSON)
- 2025Harmonia: End-to-End RAG Serving Optimizationcited by 0
- 2025Harmonia: End-to-End RAG Serving Optimization1 May 2025
- 2022Data Debugging with Shapley Importance over End-to-End Machine Learning Pipelinescited by 17
- 2021Towards Demystifying Serverless Machine Learning Trainingcited by 16
- 2022End-to-end Optimization of Machine Learning Prediction Queriescited by 16
- 2022Query Processing on Tensor Computation Runtimescited by 12
- 2022Serving and Optimizing Machine Learning Workflows on Heterogeneous Infrastructurescited by 12
- 2021A Survey on Deep Reinforcement Learning for Data Processing and Analyticscited by 10
- 2022NCVX: A General-Purpose Optimization Solver for Constrained Machine and Deep Learningcited by 7
- 2021Functional Collection Programming with Semi-Ring Dictionariescited by 6
- 2021NCVX: A User-Friendly and Scalable Package for Nonconvex Optimization in Machine Learningcited by 5
- 2022Optimization for Classical Machine Learning Problems on the GPUcited by 5
- 2022An Empirical Evaluation of Flow Based Programming in the Machine Learning Deployment Contextcited by 5
- 2021Declarative Machine Learning Systemscited by 4
12 most cited of 53 All 53 (JSON)
- 2026Spark Policy Toolkit: Semantic Contracts and Scalable Execution for Policy Learning in Spark27 Apr 2026
- 2026Vectorized Dynamic Histograms for Sparse Oblique Forests27 Feb 2026
- 2026Spark-LLM-Eval: A Distributed Framework for Statistically Rigorous Large Language Model Evaluation18 Jan 2026
- 2025High-Dimensional Data Processing: Benchmarking Machine Learning and Deep Learning Architectures in Local and Distributed Environments11 Dec 2025
- 2025Industrial brain: a human-like autonomous neuro-symbolic cognitive decision-making system30 Jun 2025
- 2025Provable Model-Parallel Distributed Principal Component Analysis with Parallel Deflation24 Feb 2025
- 2025Topic Over Source: The Key to Effective Data Mixing for Language Models Pre-training24 Feb 2025
- 2024ASTM :Autonomous Smart Traffic Management System Using Artificial Intelligence CNN and LSTM14 Oct 2024
- 2024Mew: Multiplexed Immunofluorescence Image Analysis through an Efficient Multiplex Network25 Jul 2024
- 2023KAXAI: An Integrated Environment for Knowledge Analysis and Explainable AI30 Dec 2023
- 2023Multi-Criteria Client Selection and Scheduling with Fairness Guarantee for Federated Learning Service5 Dec 2023
- 2023The Landscape of Modern Machine Learning: A Review of Machine, Distributed and Federated Learning5 Dec 2023
12 newest of 53 All 53 (JSON)