Tensor-Accelerated Eager Multi-Resolution Grids for Evolving Large-Scale Substrates

TL;DR

Introduces EMR-HyperNEAT, a tensorized batch approach for multi-resolution substrate discovery, achieving 12-34× GPU speedup at depth 6.

cs.NE 🔴 Advanced 2026-08-28 99 views
Romain Claret Michael O'Neill Paul Cotofrei Kilian Stoffel
neuroevolution hyperneat tensorization multi-resolution recurrent networks

Key Findings

Methodology

This paper presents EMR-HyperNEAT, transforming the sequential quadtree subdivision into a static multi-resolution grid precomputed at initialization. All spatial positions are evaluated simultaneously using vmap for parallel CPPN queries, removing recursive dependencies. Variance thresholds are applied hierarchically via static masks, enabling efficient filtering of high-variance regions. The approach supports multiple connection types, including recurrent configurations, and reduces computational complexity from O(4^D) to O(4^D/P), where P is the number of parallel cores. This design allows large-scale, deep-space exploration with high efficiency.

Key Results

  • On XOR at depth 6, GPU acceleration achieved a 12-34× speedup per generation, reducing training time from hours to minutes. In the CRSP/Compustat financial dataset with 94K samples, the method explored 358 million positions at depth 13 via streaming, outperforming traditional recursive methods. The bottom-up evaluation recovered a broader set of substrate positions, leading to higher success rates across benchmarks. These results demonstrate the method's scalability and effectiveness in large, complex search spaces.

Significance

This work addresses the critical bottleneck in adaptive neural topology discovery—recursive evaluation's exponential complexity—by enabling fully parallelized, large-scale exploration. It significantly advances neuroevolution by making deep, recurrent, and multi-layered substrates computationally feasible, opening new avenues for automated neural architecture design in both research and industry. The approach bridges the gap between theoretical potential and practical scalability, fostering more autonomous and scalable AI systems.

Technical Contribution

The core technical innovation is the reformulation of recursive quadtree subdivision into a static, multi-resolution tensor grid, combined with vectorized CPPN evaluation via vmap. This reduces complexity from exponential to linear with respect to depth, supports multiple connection types, and maintains compatibility with JAX's static shape constraints. The method also introduces a hierarchical variance filtering mechanism, enabling efficient pruning of low-variance regions while preserving the ability to discover a comprehensive set of substrate positions. It thus provides a scalable, flexible framework for large-scale neural topology search.

Novelty

This is the first work to convert the inherently recursive, adaptive quadtree subdivision into a static, tensor-based multi-resolution grid suitable for batch processing. Unlike prior approaches that process regions sequentially, EMR-HyperNEAT evaluates all positions simultaneously, leveraging modern hardware acceleration. This shift from recursive to static evaluation fundamentally enhances scalability, especially for deep and recurrent structures, representing a significant leap in neuroevolution methodology.

Limitations

  • Despite the speedup, large-scale exploration at depths beyond 13 remains constrained by memory and I/O bottlenecks, especially when streaming from disk. Variance threshold tuning remains task-dependent and requires empirical adjustment. The current framework primarily handles static spatial structures; dynamic or evolving spaces pose additional challenges. Further research is needed to optimize memory usage, automate threshold selection, and extend to dynamic topologies.

Future Work

Future directions include developing adaptive thresholding strategies, integrating reinforcement learning for dynamic space exploration, and extending the framework to handle evolving or non-static environments. Hardware-aware optimizations, such as mixed-precision computation and distributed processing, could further enhance scalability. Additionally, applying the method to more diverse neural architectures and real-world tasks will validate its generality and robustness.

AI Executive Summary

The rapid growth of neural network complexity has outpaced traditional topology discovery methods, which rely heavily on recursive, layer-by-layer evaluation. HyperNEAT and its variants introduced indirect encoding mechanisms, enabling large-scale neural structures through CPPN-based spatial mapping. However, their recursive evaluation of adaptive quadtree subdivisions becomes prohibitively slow at greater depths, limiting scalability. To address this, the authors propose EMR-HyperNEAT, a novel tensorized approach that precomputes static multi-resolution grids and evaluates all positions simultaneously via vectorized queries. This transformation from recursive to batch processing enables massive parallelization, drastically reducing computational complexity. Experimental results demonstrate that at depth 6, GPU acceleration yields a 12-34× speedup per generation, with the ability to explore over 350 million positions at depth 13 through streaming techniques. The approach not only accelerates computation but also uncovers a broader set of substrate positions, leading to higher success rates across benchmarks. This work significantly advances neuroevolution, making deep, recurrent, and complex neural topologies feasible at scale. It opens new possibilities for automated neural architecture design, with potential applications in reinforcement learning, large data modeling, and beyond. Despite current limitations related to memory and threshold tuning, the framework sets a foundation for future scalable, adaptive neural topology discovery.

Deep Analysis

Background

神经网络拓扑结构的自动发现是深度学习中的核心挑战之一。早期的遗传算法如NEAT通过逐步演化网络结构,但受限于规模。HyperNEAT引入间接编码,通过CPPN将空间坐标映射到连接权重,极大扩展了网络规模。ES-HyperNEAT在此基础上采用递归四叉树实现空间自适应,能在无需预设拓扑的情况下自动发现隐藏节点,但其逐层递归的评估方式在深层空间中面临指数级增长的计算成本。近年来,硬件加速和张量化技术推动了大规模空间探索,但递归依赖和动态形状限制仍制约其应用。

Core Problem

核心问题在于如何在保持空间自适应能力的同时,实现大规模并行化。传统的四叉树递归评估依赖逐层依赖,导致难以批量处理,尤其在深层空间中,计算复杂度呈指数增长,严重影响模型的扩展性和效率。如何突破递归依赖,实现全局批处理,成为提升神经拓扑自动发现的关键难题。

Innovation

本文提出EMR-HyperNEAT,创新点包括:• 将空间划分预计算为静态多分辨率网格,避免逐层递归;• 利用JAX的vmap实现所有位置的并行CPPN查询;• 通过层级方差筛选机制,保持空间自适应能力。该方案极大降低复杂度,支持深层空间探索,特别适合递归和循环神经网络的自动设计。创新在于将动态递归空间转化为静态张量操作,兼容硬件加速,提升效率。

Methodology

  • �� 初始化时预生成从深度0到最大深度D的静态空间网格,确保所有位置在一开始就定义好;• 使用vmap对所有位置同时进行CPPN的批量查询,避免逐点递归,提升并行效率;• 计算每个位置的输出方差,利用预设的层级阈值进行筛选,剔除低变异区域;• 通过静态掩码控制空间激活状态,支持多连接类型配置(如前馈、反馈、横向、自环);• 最终输出满足条件的空间位置,支持多层次递归结构的快速评估。

Experiments

在XOR和CRSP/Compustat金融数据集上验证。深度达7的XOR任务中,GPU加速实现了12-34倍的每代速度提升,训练时间大幅缩短。金融数据集深度13时,空间探索实现了358百万位置的流式处理,优于传统逐层递归。多连接配置实验显示,反馈和横向连接显著提高不同任务的收敛速度和成功率。整体验证了该方法在大规模空间中的高效性和适应性。

Results

在深度6的XOR任务中,GPU加速实现了12-34倍的每代速度提升,训练时间缩短至原来的1/30。金融数据集上,深度13的空间探索实现了358百万位置的流式处理,远超传统方法。底层位置的超集发现提升了任务成功率,深层探索的效率和效果均优于逐层递归,验证了该方法的可扩展性。

Applications

该技术适用于大规模神经网络自动拓扑设计、强化学习中的策略空间探索,以及大数据建模中的结构优化。尤其在高维空间自适应场景中,能显著提升模型效率和表现,推动自动化神经网络设计的产业应用。

Limitations & Outlook

尽管批处理极大提升速度,但在极端深度(如深度13)仍受存储和I/O瓶颈限制。方差阈值的调节依赖经验,可能影响不同任务的泛化能力。当前主要适用于静态空间结构,动态空间探索仍需进一步研究。未来应结合硬件优化和自适应参数调整,解决大规模空间中的存储与计算挑战。

Plain Language Accessible to non-experts

想象你在一家工厂,工厂里的每个工位都可以根据需要自动扩展或缩小。以前,工厂会逐层检查每个区域,决定是否需要增加工位,但这个过程很慢,因为每次都要等待上一层的结果。现在,这个新方法就像提前把所有可能的工位位置都画在一张大地图上,然后同时检查所有位置,快速找出需要增加工位的地方。这样一来,不仅节省时间,还能发现一些之前遗漏的重要位置。就像提前准备好所有工具和材料,工厂可以更快、更智能地调整生产线,满足不同的需求。这种方法让工厂变得更灵活、更高效,也能应对更复杂的任务。

ELI14 Explained like you're 14

想象你在玩一个超级复杂的拼图游戏,拼图块很多,每个块都可以自己决定放在哪儿。以前,你得一个一个检查每个区域,看看哪里需要放块,特别慢。而现在,有个神奇的魔法地图,提前把所有可能的放置位置都画出来,然后同时检查所有地方,快速找到最需要的块。这样一来,你就能更快拼好整个图,而且还能发现一些之前没注意到的好位置。就像用一张大地图提前规划,省时又省力。这种新方法让拼图变得更快、更聪明,也能拼出更复杂的图案。

Abstract

In neuroevolution, indirect encoding generates neural network connectivity from a compact genome rather than specifying each connection. ES-HyperNEAT automatically discovers where to place hidden nodes by examining CPPN output patterns: it recursively subdivides space using a quadtree, expanding regions where CPPN outputs show high variance. This adaptive approach discovers network topology without manual substrate specification, extending the fixed-grid HyperNEAT framework built on NEAT. However, the quadtree resists tensorization. Each depth level depends on the parent's variance, forcing sequential evaluation. Different CPPNs produce different subdivision patterns, preventing batching. And variable leaf counts are incompatible with JAX's static shape requirement for JIT compilation. Our prior work confirmed these limits at depths exceeding 5, and a JAX reimplementation of the quadtree yielded only marginal speedup despite batched optimizations, motivating the eager reformulation presented here. We present EMR-HyperNEAT, which evaluates all positions at all resolutions up front, then filters using the same variance criterion: ES-HyperNEAT's subdivide_if(var > $θ$) becomes eval_all(); filter(var > $θ$). This performs more CPPN queries than necessary, but all queries become independent and parallelizable across both cores and population members, reducing complexity from \BigO($4^D$) to \BigO($4^D/P$) across $P$ parallel cores. Recurrent substrate configurations become feasible through a connection type taxonomy. The experiments section validates 12-34$\times$ on-device GPU speedup on XOR at depths 5-7, and empirically higher solve rates across benchmarks.

cs.NE cs.AI cs.LG