CoMind: Towards Community-Driven Agents for Machine Learning Engineering

Sijie Li, Weiwei Sun, Shanda Li, Ameet Talwalkar, Yiming Yang
6/25/2025
cs.AIcs.LG

No official code found. Best implementation: comind-ml/CoMind (4 stars, MIT, last commit 2026-02-24).

Abstract

Large language model (LLM) agents show promise in automating machine learning (ML) engineering. However, existing agents typically operate in isolation on a given research problem, without engaging with the broader research community, where human researchers often gain insights and contribute by sharing knowledge. To bridge this gap, we introduce MLE-Live, a live evaluation framework designed to assess an agent's ability to communicate with and leverage collective knowledge from a simulated Kaggle research community. Building on this framework, we propose CoMind, an multi-agent system designed to actively integrate external knowledge. CoMind employs an iterative parallel exploration mechanism, developing multiple solutions simultaneously to balance exploratory breadth with implementation depth. On 75 past Kaggle competitions within our MLE-Live framework, CoMind achieves a 36% medal rate, establishing a new state of the art. Critically, when deployed in eight live, ongoing competitions, CoMind outperforms 92.6% of human competitors on average, placing in the top 5% on three official leaderboards and the top 1% on one.

View on arXivView PDF

Code Implementations(1)

comind-ml/CoMindCommunity100%
MIT

Command-line version of CoMind, a community-augmented machine learning engineering agent.

40PythonJun 16, 20257 months agoMIT
agentkagglelarge-language-modelsmachine-learning

Similar Papers

Cite this paper

@article{li2025comind,
  title  = {CoMind: Towards Community-Driven Agents for Machine Learning Engineering},
  author = {Sijie Li and Weiwei Sun and Shanda Li and Ameet Talwalkar and Yiming Yang},
  year   = {2025},
  eprint = {2506.20640},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2506.20640v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2506.20640.svg)](https://codeofpaper.com/paper/2506.20640)

Discussion