LLM2D
科扎克斯:JAX 中的灵活可扩展遗传编程
Kozax: Flexible and Scalable Genetic Programming in JAX
作者: Sigur de Vries, Sander W. Keemink, Marcel A. J. van Gerven
发布日期: 2/6/2025
arXiv ID: oai:arXiv.org:2502.03047v1

摘要

arXiv:2502.03047v1 交叉类型: 摘要:遗传编程是受自然选择启发的优化算法,它可以自动演化计算机程序的结构。生成的计算机程序与具有固定结构的黑盒模型相比,更为可解释且高效。遗传编程中的适应度评估面临着高计算要求的挑战,这限制了其在困难问题上的性能。为了减少运行时间,许多遗传编程的实现需要特定的数据格式,这使得其应用范围受限于特定的问题类。因此,目前没有一种高效且适用于广泛任务的遗传编程框架。为了解决这一问题,我们开发了Kozax,这是一个能够针对任意问题演化符号表达式的遗传编程框架。我们使用了JAX框架进行实施,JAX是一种高性能、可扩展的机器学习框架,这使得适应度评估能够高效地扩展到大量种群或数据集,特别是在GPU上。此外,Kozax提供了常数优化、自定义操作符定义以及同时演化多个树的能力。我们展示了Kozax在发现自然法则方程、恢复隐藏动态变量方程以及演化控制策略方面的成功应用。总体而言,Kozax提供了一个通用、快速且可扩展的库,用于在科学计算领域优化白盒解决方案。