mlsanityData to 7 Oct 2026

Intelligent Spark Agents: A Modular LangGraph Framework for Scalable, Visualized, and Enhanced Big Data Machine Learning Workflows

arXiv 2412.01490 ↗First posted 2 Dec 2024cs.AIcited by 4

On the map: LLM agents and tool use → Web, search, and research agents

Abstract (arXiv)

This paper presents a Spark-based modular LangGraph framework, designed to enhance machine learning workflows through scalability, visualization, and intelligent process optimization. At its core, the framework introduces Agent AI, a pivotal innovation that leverages Spark's distributed computing capabilities and integrates with LangGraph for workflow orchestration. Agent AI facilitates the automation of data preprocessing, feature engineering, and model evaluation while dynamically interacting with data through Spark SQL and DataFrame agents. Through LangGraph's graph-structured workflows, the agents execute complex tasks, adapt to new inputs, and provide real-time feedback, ensuring seamless decision-making and execution in distributed environments. This system simplifies machine learning processes by allowing users to visually design workflows, which are then converted into Spark-compatible code for high-performance execution. The framework also incorporates large language models through the LangChain ecosystem, enhancing interaction with unstructured data and enabling advanced data analysis. Experimental evaluations demonstrate significant improvements in process efficiency and scalability, as well as accurate data-driven decision-making in diverse application scenarios. This paper emphasizes the integration of Spark with intelligent agents and graph-based workflows to redefine the development and execution of machine learning tasks in big data environments, paving the way for scalable and user-friendly AI solutions.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Web, search, and research agents, one of 16 directions in LLM agents and tool use. Builds on 1 of its 9 references that are in the map. First posted on arXiv 2 Dec 2024.

Builds on

1 of 9 references.

References · 1

  1. 2015MLlib: Machine Learning in Apache Sparkcited by 55

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 3; 2026: 1.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024Research on the Application of Spark Streaming Real-Time Data Analysis System and large language model Intelligent Agents
  2. 2025Data Agent: A Holistic Architecture for Orchestrating Data+AI Ecosystems
  3. 2026AgenticDataBench: A Comprehensive Benchmark for Data Agents
  4. 2024Data Interpreter: An LLM Agent For Data Science
  5. 2025R&D-Agent: An LLM-Agent Framework Towards Autonomous Data Science
  6. 2025ML-Tool-Bench: Tool-Augmented Planning for ML Tasks
  7. 2025Large Language Model-based Data Science Agent: A Survey
  8. 2025LLM-Based Data Science Agents: A Survey of Capabilities, Challenges, and Future Directions