English

CoMind: Towards Community-Driven Agents for Machine Learning Engineering

Artificial Intelligence 2026-03-02 v3 Machine Learning

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, a multi-agent system designed to systematically leverage 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.

Keywords

Cite

@article{arxiv.2506.20640,
  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},
  journal= {arXiv preprint arXiv:2506.20640},
  year   = {2026}
}

Comments

ICLR 2026. Code available at https://github.com/comind-ml/CoMind

R2 v1 2026-07-01T03:33:24.432Z