HyperAIHyperAI

Command Palette

Search for a command to run...

Frontis-MA1: 機械学習エンジニアリングにおける再帰的自己改善に向けたAI4AIモデルの訓練

概要

再帰的自己改善(RSI)には、AIを構築するプロセスを改善するAIシステム(AI4AI)が必要である。機械学習エンジニアリング(MLE)は、この能力を研究するための具体的かつ実行可能なテストベッドを提供する。我々は、MLEにおけるRSI研究のためのオープンなフルスタックシステムOpenMLEを導入する。これは、実行フィードバックを伴う検証可能なタスク環境(OpenMLE-Gym)、オペレータ学習(OpenMLE-ERL)、および長期探索(OpenMLE-Evo)から構成される。このスタック上で、我々はFrontis-MA1-35BをMLEのためのメタ進化エージェントとして事後訓練する。事後訓練と推論を、4つの原子的なプログラム進化オペレータ(Draft、Improve、Debug、Crossover)を中心に整合させる。これらの同一のオペレータは、全ての評価ベンチマークに対して重複除去されたデータを用いた実行基盤型の教師あり微調整(SFT)と強化学習(RL)によって訓練され、その後、長期探索へと構成され、学習と進化を単一のループで結合する。RTX 4090 1基、12GBのVRAM制限、タスクあたり12時間の予算下でのMLE-Bench Liteにおいて、Frontis-MA1-35Bは、OpenMLE-Evoを用いることで、そのベースモデルからMedal Averageを39.39%から60.61%に改善し、OpenMLE-Evo-Max(ベンチマーク非依存の経験事前分布と非同期探索)を用いることで71.21%に達し、GPT-5.5 + Codexを上回り、GPT-5.6 Solや2.8TのKimi K3に迫る。ホールドアウトされたNatureBench Liteにおいては、両方のコンポーネントが転移する。フレームワークを固定し、訓練済みモデルに置き換えると、Match-SOTAが50%から70%に向上する。モデルを固定し、OpenMLE-Evoに置き換えると、20%から50%に向上する。我々は、RSIに向けた実行可能なAI4AIに関する再現可能な研究を可能にするため、モデルの重みと完全なOpenMLEスタックを公開する。

One-sentence Summary

Researchers from Horizon Research, Frontis.AI, and Tsinghua University introduce Frontis-MA1-35B, a meta-evolution agent that posttrains four atomic program-evolution operators via execution-grounded SFT and RL, composes them into long-horizon search, and boosts Medal Average on MLE-Bench Lite from 39.39% to 71.21% with OpenMLE-Evo-Max, surpassing GPT-5.5+Codex and approaching GPT-5.6 Sol and Kimi K3, while also transferring to NatureBench Lite.

Key Contributions

  • OpenMLE is an open full-stack system for meta-evolution research in machine learning engineering. It provides verifiable task environments with execution feedback, execution-grounded operator learning via supervised fine-tuning and reinforcement learning, and long-horizon evolutionary search guided by structured experience.
  • Frontis-MA1-35B is post-trained as a meta-evolution agent by aligning training and inference around four atomic program-evolution operators (Draft, Improve, Debug, Crossover) trained with execution-grounded SFT and RL on data deduplicated against evaluation benchmarks, then composed into long-horizon search, coupling learning and evolution.
  • Combining the trained model with OpenMLE-Evo improves Medal Average from 39.39% to 60.61% on MLE-Bench Lite, and with OpenMLE-Evo-Max reaches 71.21%. Both the model and the search framework transfer to NatureBench Lite, where the model raises Match-SOTA from 50% to 70% and the search framework raises it from 20% to 50%.

Introduction

The authors address the challenge of building AI systems that can autonomously improve other AI systems, a direction known as AI for AI (AI4AI) with the long-term goal of recursive self-improvement. Machine learning engineering (MLE) serves as a concrete testbed, where an agent must iteratively develop ML solutions under delayed, noisy execution feedback. Prior work advances MLE agents through evolutionary search, task environments, or execution-grounded post-training, but no existing public system jointly spans scalable task construction, agent training from execution feedback, and a long-horizon evolutionary harness in a reproducible workflow. The authors introduce OpenMLE, an open full-stack solution that unifies these components: OpenMLE-Gym provides quality-gated tasks and structured feedback, OpenMLE-ERL trains reusable program-transformation operators via supervised and reinforcement learning, and OpenMLE-Evo composes those operators into experience-guided search. This integrated stack enables training a meta-evolution agent that serves as the variation engine of its own evolutionary harness, demonstrating significant gains on MLE benchmarks and releasing all artifacts for broader study.

Dataset

The authors construct OpenMLE-Gym, a suite of 5,758 executable machine learning engineering tasks designed for post-training, search, and evaluation of LLM agents. Each task is a verifiable environment that couples a problem description, training data, hidden test labels, and a metric script into a standardized package. The dataset is built from three complementary sources:

  • Curated Anchors (156 tasks): Manually selected from existing papers and benchmarks, offering the highest design confidence but limited scale. Original assets are processed directly into executable packages.
  • Kaggle Datasets (3,362 tasks): Generated at scale using the MLE-Smith dataset-to-task pipeline, then filtered with package-level quality control. Objectives are automatically induced, so quality can vary.
  • Kaggle Competitions (2,240 tasks): Built through a custom crawling and construction pipeline that downloads competition files, extracts schemas and metadata from Meta Kaggle, and generates task descriptions, data splits, and evaluation scripts. Competitions overlapping with MLE-Bench are excluded to preserve evaluation integrity.

All tasks share a unified layout: raw assets stay in raw/, agent-visible training data, test inputs, and a sample submission reside in data/public/, hidden answers are isolated in data/private/, and a metric.py script validates predictions and returns a scalar score. For competition-derived tasks, the automated framework uses local evidence (schemas, data types, dimensions, sampled rows) together with Meta Kaggle records to produce prepare.py, which deterministically splits labeled data, separates public inputs from private answers, and creates a schema-compatible sample submission. The package is then validated by executing the preparation code and scoring the sample submission; unbuildable packages or those that fail to produce a valid scalar metric are discarded.

A subsequent LLM-based quality filter inspects each task package (description, raw files, processing script, outputs, data samples) and returns structured judgments on five dimensions: task validity, data sufficiency, raw-data usage, task complexity, and data quality. Only metric-valid tasks that receive the strictest recommendation are retained, removing degenerate targets, data leakage, annotation errors, and malformed processing. For the competition branch, this semantic gate is applied after leaderboard-length screening, MLE-Bench overlap removal, and licensing checks.

The final dataset spans tabular, text, time-series, image, and other modalities (11% are multimodal), with classification and regression accounting for 87% of tasks. The authors use these environments to provide agents with verifiable feedback: an agent reads the public training data, produces predictions on the test inputs, and receives a scalar reward from metric.py, enabling post-training, test-time search, and recursive self-improvement loops.

Method

The authors construct OpenMLE-Gym, a unified executable task suite, by mapping three source-specific paths into a shared executable task package. As shown in the figure below, the curation hierarchy balances quality and scale across Curated Anchors, Kaggle Competitions, and Kaggle Datasets. For the competition-derived branch, an automated framework converts a Kaggle competition slug into a standardized executable package, applying a strict semantic quality gate after executability validation. The resulting executable task format separates agent-visible public inputs from hidden private answers and includes executable utilities to ensure reproducible scoring.

To train the model for evolutionary inference, the authors separate local skills from the search algorithm by training a compact set of reusable program-transformation operators: Draft, Improve, Debug, and Crossover. The training process consists of an execution-grounded supervised warm start followed by reinforcement learning.

For the supervised warm start, the authors employ two sampling paths. The parallel path independently samples and executes complete Draft solutions, while the evolutionary path applies Improve, Debug, and Crossover over executed programs to retain useful steps from high-quality local trajectory segments. Collection stops based on a budget-adaptive rule, forming a large-scale SFT corpus.

Building on this warm start, the reinforcement learning phase focuses on making heterogeneous outcomes comparable and concentrating the learning signal. The authors first define a bounded base reward and then derive tighter adaptive bounds from the task historical on-policy score frontier to preserve resolution. To further amplify the reward gaps near the top of each rollout group, an entropic advantage is applied, directing substantially more learning signal to the best candidates. As illustrated in the figure below, combining entropic weighting with adaptive bounds yields a stronger smoothed Group Best Reward trajectory compared to previous reward constructions.

For test-time scaling, the authors introduce OpenMLE-Evo, an experience-driven search harness that converts execution outcomes into reusable evidence. The search process maintains a persistent representation of prior attempts through structured experience accumulation. Each evaluated node generates a node-level experience card capturing provenance and performance, which are aggregated into a task-global experience board.

Parent selection is guided by a multi-factor utility combining normalized validation score, positive improvement over the strongest parent, and method-family novelty. Once a parent is selected, the system defers rich natural-language memory synthesis until an operator call is made, retrieving relevant ancestors and siblings to construct a bounded, operator-conditioned context. The overall experience-guided search framework is depicted in the figure below, showing how experience-card metadata updates the global search state and informs memory injection for subsequent operations.

Experiment

The evaluation uses the 22-task MLE-Bench Lite split under a fixed 12-hour single-GPU budget, reporting Valid Rate, Medal Average, and Human Rank across three runs. Execution-grounded post-training substantially improves performance over base models, and combining it with a domain-specific evolutionary search harness that distills cross-task priors yields complementary gains that surpass strong external systems. Long-horizon search with structured memory and multi-factor selection converts additional compute into sustained improvement beyond the first executable solution, raising solution quality toward Gold medals rather than merely crossing the Bronze threshold. These benefits transfer across input modalities and to a scientific benchmark, where the post-trained model and adapted search framework together improve paper-relative scientific progress.

On MLE-Bench Lite, the Frontis-MA1-35B system with the OPENMLE-EVO framework substantially outperforms its base Qwen model, delivering a higher valid submission rate, medal average, and human rank. The OPENMLE-EVO-MAX variant further improves all metrics, achieving perfect validity and the highest medal average. The smaller Frontis-MA1-30B model trails the 35B configuration across the same measures. Frontis-MA1-35B with OPENMLE-EVO raises the medal average and valid rate markedly above the Qwen3.6-35B-A3B base model. Using OPENMLE-EVO-MAX, the system reaches perfect validity (22/22) and a medal average of 71.21%, the highest in the comparison. Human rank improves from 0.58 for the base model to 0.76 with EVO and to 0.81 with the MAX variant. The smaller Frontis-MA1-30B model scores lower on valid rate, medal average, and human rank than the 35B system.

The comparison evaluates Frontis-MA1-30B and GLM-5.2 on an MLE benchmark across different harnesses. OPENMLE-EVO-MAX yields perfect validity and the highest medal average for both models, while GLM-5.2 with this framework attains the top human rank. The OPENMLE-EVO framework shows lower validity but can still improve medal average over the Claude Code baseline for GLM-5.2. OPENMLE-EVO-MAX achieves a flawless valid rate of 22/22 and a medal average of 66.67% for both Frontis-MA1-30B and GLM-5.2. GLM-5.2 with OPENMLE-EVO-MAX records the highest human rank (0.8164), edging out Frontis-MA1-30B with the same framework (0.8053).

On the 10-task NatureBench Lite, the best reference agents achieve 70% Surpass-SOTA and perfect Match-SOTA. The post-trained Frontis-MA1-35B improves over its base model by 10 percentage points in Surpass-SOTA and 20 points in Match-SOTA, while the adapted search harness alone contributes gains of 10 and 30 points, respectively. Together, the model and harness match several strong reference systems, indicating that execution-grounded post-training transfers beyond competition-style machine learning engineering. Claude Opus 4.7 and GLM-5.2 lead the reference agents with 70% Surpass-SOTA and 100% Match-SOTA on the 10-task subset. Frontis-MA1-35B raises Match-SOTA by 20 points over its base model and, combined with the OpenMLE-Evo adapter, reaches performance on par with GPT-5.4 and GLM-5.1.

Experiments on MLE-Bench Lite show that the OPENMLE-EVO framework, especially its MAX variant, dramatically improves the Frontis-MA1-35B model's validity, medal average, and human rank over the base model, with perfect validity and the highest medal average; the smaller 30B model lags behind. A head-to-head comparison confirms that OPENMLE-EVO-MAX yields perfect validity and top medal averages for both Frontis-MA1-30B and GLM-5.2, with GLM-5.2 leading in human rank. On NatureBench Lite, post-trained Frontis-MA1-35B combined with the harness matches top reference agents, validating the transfer of execution-grounded post-training beyond competition tasks.


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています