Intelligent Spark Agents: A Modular LangGraph Framework for Scalable, Visualized, and Enhanced Big Data Machine Learning Workflows
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
- 2015MLlib: Machine Learning in Apache Sparkcited by 55
Built on by
Citing papers per year: 2025: 3; 2026: 1.
4 papers.
- 2025A Comprehensive Survey on Multi-Agent Cooperative Decision-Making: Scenarios, Approaches, Challenges and Perspectivescited by 27
- 2025Agentic AI framework for End-to-End Medical Data Inferencecited by 8
- 2026Still Fresh? Evaluating Temporal Drift in Retrieval Benchmarkscited by 1
- 2025Match Chat: Real Time Generative AI and Generative Computing for Tenniscited by 0
- 2026Still Fresh? Evaluating Temporal Drift in Retrieval Benchmarks4 Mar 2026
- 2025Match Chat: Real Time Generative AI and Generative Computing for Tennis16 Sep 2025
- 2025Agentic AI framework for End-to-End Medical Data Inference24 Jul 2025
- 2025A Comprehensive Survey on Multi-Agent Cooperative Decision-Making: Scenarios, Approaches, Challenges and Perspectives17 Mar 2025
Similar, not cited
Close in wording, no citation either way
- 2024Research on the Application of Spark Streaming Real-Time Data Analysis System and large language model Intelligent Agents
- 2025Data Agent: A Holistic Architecture for Orchestrating Data+AI Ecosystems
- 2026AgenticDataBench: A Comprehensive Benchmark for Data Agents
- 2024Data Interpreter: An LLM Agent For Data Science
- 2025R&D-Agent: An LLM-Agent Framework Towards Autonomous Data Science
- 2025ML-Tool-Bench: Tool-Augmented Planning for ML Tasks
- 2025Large Language Model-based Data Science Agent: A Survey
- 2025LLM-Based Data Science Agents: A Survey of Capabilities, Challenges, and Future Directions