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

TL;DR

提出EMR-HyperNEAT,通过批处理实现多分辨率空间自适应拓扑发现,显著提升速度。

cs.NE 🔴 高级 2026-08-28 98 次浏览
Romain Claret Michael O'Neill Paul Cotofrei Kilian Stoffel
神经进化 超网格 张量化 递归网络 多分辨率

核心发现

方法论

本文提出EMR-HyperNEAT,将原有的逐层四叉树递归划分改为预计算多分辨率网格,通过批量向量化查询CPPN,利用JAX的静态形状支持实现全局并行。核心机制包括:• 预生成从深度0到最大深度D的静态空间网格;• 使用vmap进行所有位置的并行CPPN查询;• 利用方差阈值进行层级筛选,避免逐层递归。该方法支持多种连接类型配置,极大减少计算复杂度。

关键结果

  • 在深度6的XOR任务中,GPU加速实现12-34倍的每代速度提升,整体训练时间缩短至原来的1/30;在金融数据集CRSP/Compustat(94K样本)上,深度13的空间探索实现了358百万位置的流式处理,显著优于传统逐层方法。实验还显示,底层位置的超集发现提高了任务成功率,深度越大优势越明显。

研究意义

该研究突破了传统四叉树递归评估的瓶颈,首次实现大规模自适应神经网络拓扑的全局并行化,极大推动神经结构搜索的效率和规模。其技术可广泛应用于复杂神经网络设计、强化学习和大数据建模,解决了高维空间中拓扑自动发现的关键难题,为深度学习的自动化发展提供新路径。

技术贡献

本文提出的张量化重构方案,将逐层递归的空间划分转变为静态多分辨率网格,结合vmap实现全局批处理,显著降低复杂度,从O(4^D)到O(4^D/P),支持多连接类型配置,特别适合递归和循环结构。该方法兼容JAX的静态形状限制,提供了可扩展的多层次空间探索框架,为未来大规模神经网络自动设计奠定基础。

新颖性

首次将适应性空间划分的四叉树方法转化为全局预计算的多分辨率网格,结合批量向量化实现大规模并行,解决了递归依赖和动态形状带来的瓶颈。与以往逐层递归不同,EMR-HyperNEAT实现了在深层次空间中高效探索,拓扑发现的同时大幅提升速度。

局限性

  • 尽管批处理大幅提升速度,但在极端深度(如深度13)仍受限于存储和I/O瓶颈,尤其在超大空间中流式处理成本较高;
  • 方差阈值的调整依赖经验,可能影响不同任务的泛化能力;
  • 当前主要适用于静态空间结构,动态变化的空间探索仍需进一步研究。

未来方向

未来将探索自适应阈值策略,结合强化学习优化空间划分参数;同时扩展到动态图结构和多模态数据,提升模型的泛化能力和适应性。此外,将结合硬件加速技术,进一步降低大规模空间探索的成本。

AI 总览摘要

随着神经网络规模不断扩大,传统的空间拓扑自动发现方法面临计算瓶颈。HyperNEAT等方法通过递归四叉树实现空间自适应,但在深层次探索中受制于逐层依赖和动态形状,难以实现大规模并行。本文提出EMR-HyperNEAT,将空间划分预计算为静态多分辨率网格,通过批量向量化实现所有位置的同时评估,极大提升了深度探索的效率。在深度6的XOR任务中,GPU加速实现了12-34倍的速度提升,显著缩短了训练时间。金融数据集上的实验也验证了其在大规模空间中的优越性能。该方法不仅突破了递归空间划分的瓶颈,还支持多种连接配置,特别适合复杂递归网络的自动设计。未来,结合自适应阈值和硬件优化,将进一步推动神经结构搜索的自动化和规模化发展。

深度分析

研究背景

神经网络拓扑结构的自动发现一直是深度学习中的关键难题。早期方法如NEAT通过遗传算法优化结构,但规模有限。HyperNEAT引入间接编码,通过CPPN映射空间坐标,显著扩展网络规模。ES-HyperNEAT在此基础上采用递归四叉树实现空间自适应,能在无需预设拓扑的情况下自动发现隐藏节点,但其逐层递归特性限制了大规模并行化,尤其在深层空间中计算成本呈指数增长。近年来,硬件加速和张量化技术推动了大规模空间探索的可能性,但仍受限于动态形状和依赖递归的计算瓶颈。

核心问题

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

核心创新

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

方法详解

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

实验设计

在XOR任务和CRSP/Compustat金融数据集上验证。XOR深度达7,GPU加速实现12-34倍速度提升,显著缩短训练时间。金融数据集深度13时,空间探索实现358百万位置的流式处理。对比传统逐层方法,EMR-HyperNEAT在保持或提升解题成功率的同时,大幅降低计算成本。多连接配置的实验显示,反馈和横向连接显著提高不同任务的收敛速度和成功率。

结果分析

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

应用场景

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

局限与展望

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

通俗解读 非专业人士也能看懂

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

简单解释 像给14岁少年讲一样

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

原文摘要

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