What Does It Take to Be a Good AI Research Agent? Studying the Role of Ideation Diversity
Authors: name: Alexis Audran-Reiss, name: Jordi Armengol Estapé, name: Karen Hambardzumyan, name: Amar Budhiraja, name: Martin Josifoski, name: Edan Toledo, name: Rishi Hazra, name: Despoina Magka, name: Michael Shvartsman, name: Parth Pathak, name: Justine T Kao, name: Lucia Cipolina-Kun, name: Bhavul Gauri, name: Jean-Christophe Gagnon-Audet, name: Emanuel Tewolde, name: Jenny Zhang, name: Taco Cohen, name: Yossi Adi, name: Tatiana Shavrina, name: Yoram Bachrach
Organization: AI at Meta

AI research agents offer the promise to accelerate scientific progress by automating the design, implementation, and training of machine learning models. However, the field is still in its infancy, and the key factors driving the success or failure of agent trajectories are not fully understood. We examine the role that ideation diversity plays in agent performance. First, we analyse agent trajectories on MLE-bench, a well-known benchmark to evaluate AI research agents, across different models and agent scaffolds. Our analysis reveals that different models and agent scaffolds yield varying degrees of ideation diversity, and that higher-performing agents tend to have increased ideation diversity. Further, we run a controlled experiment where we modify the degree of ideation diversity, demonstrating that higher ideation diversity results in stronger performance. Finally, we strengthen our results by examining additional evaluation metrics beyond the standard medal-based scoring of MLE-bench, showing that our findings still hold across other agent performance metrics.