Paper

Building AI Research Teams for Autonomous Discovery of Linear Attention Architectures

A heterogeneous AI research team searches linear attention under fixed budget, using private memory plus adaptive researcher selection instead of flat agent pools.

arXiv ↗Oct 20, 2026

  • llm
  • architecture search
  • linear attention
  • multi-agent systems
Detailed pipeline diagram showing research cycle, researchers, and organizational factors.
Detailed pipeline diagram showing research cycle, researchers, and organizational factors.

What it is

Builds a team of persistent AI research agents with different scientific backgrounds, private memory, and adaptive scheduling to autonomously discover better linear attention designs. Instead of a single agent or uniform multi-agent search, it uses discipline-specialized researchers with private histories, shared evidence, and UCB-style allocation to drive orthogonal exploration and iterative refinement of architectures.

Key results

  • At 1.3B, CFDN outperforms GDN2 on LM, CSR, synthetic retrieval, and RWR
  • Search evaluated 63 candidates within an 80-iteration research budget
  • Removing backgrounds, private history, or adaptive allocation gave weaker discoveries

Autonomous discovery improves when agents differ in expertise, retain their own research trajectories, and are allocated experiments adaptively.

One useful ML email a week.

Worth your time, not all of your time.

We track new ML work across papers, code, and data, then keep only the pieces worth a closer look. Selected, not collected.

Papers.Data.Code

The Weekly ML Signal

Selected, not collected.

One useful ML email a week: the papers, repos and datasets worth your attention.

Worth your time, not all of your time.