摘要
arXiv:2501.17168v3 通知类型: replace-cross
摘要:基于树的遗传编程(TGP)是一种广泛用于符号回归、特征工程和科学建模的关键进化算法。其高计算需求使得GPU加速对于可扩展和高性能的进化计算至关重要。然而,TGP的GPU加速面临着三个关键挑战:不高效的树编码、高度异构的遗传操作以及评估适应度时有限的并行性。为了解决这些问题,我们引入了EvoGP,这是一个全面的GPU加速TGP框架。首先,我们设计了一种张量编码方案,以相同形状的张量来表示不同结构的树,优化内存访问并支持高效并行执行。其次,我们通过利用共享计算原语并为可扩展性能实施专门的CUDA内核,提出了一种统一的遗传操作并行框架。第三,我们提出了一种完全并行的适应度评估策略,利用种群级和数据级并行性来最大限度地利用GPU。此外,我们实现了一个全面的库,提供丰富的算法操作符和基准问题。EvoGP已在各种任务上进行了广泛测试,包括符号回归、分类和机器人控制,展示了其在不同类型应用场景中的 versatility 和有效性。实验结果显示,EvoGP在与最先进的GPU加速TGP实现相比,速度提高了多达140.89倍,同时保持或超过了基线方法的准确性。EvoGP是开源的,并可在以下链接访问:https://github.com/EMI-Group/evogp。