2个稳定版本
使用旧的Rust 2015
2.0.0 | 2019年9月28日 |
---|---|
1.0.0 | 2019年5月18日 |
#2325 in 算法
140KB
3K SLoC
oxigen
Oxigen是一个在Rust中实现的并行遗传算法库。这个名字来自于将OXI化(Rust翻译成西班牙语)和GENetic合并。
Oxigen提供以下功能
- 快速且并行的遗传算法实现(它在N=255时几秒钟内解决N皇后问题)。有关基准测试,请查看此文件的基准测试部分。
- 可自定义的突变和选择率,根据代数内置的常量、线性二次函数(您可以通过MutationRate和SelectionRate特质实现自己的函数)。
- 可自定义个体的年龄不适应性,内置无不适应性、线性和不适应性二次函数,根据个体的代数设置阈值(您可以通过Age特质实现自己的年龄函数)。
- 内置的Roulette、Tournaments和Cup选择函数(您可以通过Selection特质实现自己的选择函数)。
- 内置的SingleCrossPoint、MultiCrossPoint和UniformCross交叉函数(您可以通过Crossover特质实现自己的交叉函数)。
- 许多内置的生存压力函数。您可以通过SurvivalPressure特质实现自己的生存压力函数。
- 内置的Niches PopulationRefitness函数。您可以通过PopulationRefitness特质实现自己的种群再适应函数。
- SolutionFound、Generation、Progress和更多内置的停止标准(您可以通过StopCriterion特质实现自己的停止标准)。
- Genotype特质用于定义遗传算法的基因型。以下结构可以实施Genotype特质,受以下限制
- 它有一个iter函数,该函数返回一个对其基因的std::slice::Iter迭代器。
- 它有一个into_iter函数,该函数消耗个体并返回一个对其基因的std::vec::IntoIter迭代器。
- 它有一个
from_iter
函数,用于从迭代器设置基因。 - 它实现了
Display
、Clone
、Send
和Sync
。 - 它具有生成随机个体、突变个体、获取个体的
fitness
以及判断个体是否为问题的is_solution
的函数。
- 个体的适应度被缓存起来,以避免不必要的重新计算(如果您的适应度函数是随机的,需要在每个代重新计算适应度,可以通过
.cache_fitness(false)
禁用)。 - 进度统计信息可以配置为每一定数量的代打印到文件中。
- 每一定数量的代,可以配置将具有适应度的种群个体打印到文件中。
- 可以在遗传算法执行过程中插入特定的初始个体。
- 可以使用上一代的种群作为初始种群来恢复遗传执行。
- 通过在适应度函数中执行小的遗传算法重新执行,可以实现协同进化。
2和1版本之间的差异
- Oxigen 2更灵活,因为任何包含
struct
的Vec
都可以实现Genotype
。在1版本中,这是不可能的,因为Genotype
必须实现FromIterator
。在2版本中,增加了一个from_iter
函数。 - Oxigen 2修复了问题#3(《Cuadratic’已被替换为内置枚举中的‘Quadratic’)。在1版本中,没有修复这个问题以避免破坏接口。
Genotype
中的fix
函数返回一个布尔值,以指定个体是否已更改以重新计算其适应度。- 现在,每一代得到的解决方案数量是使用
Genotype
的新distance
函数得到的不同解决方案的数量。 - 在
StopCriterion
、MutationRate
和SelectionRate
特质中,u16
类型已被替换为usize
。 - 添加了
PopulationRefitness
特质,以便可选地比较种群个体来重新拟合。还添加了内置的Niches
PopulationRefitness
函数。 - 重新定义了
SurvivalPressure
特质,现在它通过接收一个包含父母和子女对的列表来杀死个体,而不是返回要删除的索引。 - 添加了许多内置的
SurvivalPressure
函数,如Overpouplation
、CompetitiveOverpopulation
、DeterministicOverpopulation
、ChildrenFightParents
、ChildrenFightMostsimilar
等。 - 这两个新增功能允许在不同的搜索空间区域中搜索不同的解决方案,以避免局部次优解并找到不同的解决方案。
- 其他一些小的改进。
使用方法
在您的Cargo.toml
文件中添加oxigen
依赖项
[dependencies]
oxigen = "^2"
要使用 oxigen
,请使用 use oxigen::prelude::*
并在一个 GeneticExecution
实例上调用 run
方法,覆盖默认的超参数和函数,以满足您的需求
let n_queens: u8 = std::env::args()
.nth(1)
.expect("Enter a number between 4 and 255 as argument")
.parse()
.expect("Enter a number between 4 and 255 as argument");
let progress_log = File::create("progress.csv").expect("Error creating progress log file");
let population_log = File::create("population.txt").expect("Error creating population log file");
let log2 = (f64::from(n_queens) * 4_f64).log2().ceil();
let population_size = 2_i32.pow(log2 as u32) as usize;
let (solutions, generation, progress) = GeneticExecution::<u8, QueensBoard>::new()
.population_size(population_size)
.genotype_size(n_queens as u8)
.mutation_rate(Box::new(MutationRates::Linear(SlopeParams {
start: f64::from(n_queens) / (8_f64 + 2_f64 * log2) / 100_f64,
bound: 0.005,
coefficient: -0.0002,
})))
.selection_rate(Box::new(SelectionRates::Linear(SlopeParams {
start: log2 - 2_f64,
bound: log2 / 1.5,
coefficient: -0.0005,
})))
.select_function(Box::new(SelectionFunctions::Cup))
.age_function(Box::new(AgeFunctions::Quadratic(
AgeThreshold(50),
AgeSlope(1_f64),
)))
.progress_log(20, progress_log)
.population_log(2000, population_log)
.run();
有关完整示例,请访问 nqueens-oxigen 示例。
有关更多信息,请访问 文档。
恢复之前的执行
从版本 1.1.0 开始,遗传算法执行会返回上一代的人口,新的遗传执行接受一个初始人口。这允许恢复之前的执行,并允许协同进化,因为可以在适应度函数中启动少量遗传算法的重新执行。
以下示例中启动了一个10000代的执行,并在找到解决方案后继续执行,以不同的速率。
let n_queens: u8 = std::env::args()
.nth(1)
.expect("Enter a number between 4 and 255 as argument")
.parse()
.expect("Enter a number between 4 and 255 as argument");
let progress_log = File::create("progress.csv").expect("Error creating progress log file");
let population_log = File::create("population.txt").expect("Error creating population log file");
let log2 = (f64::from(n_queens) * 4_f64).log2().ceil();
let population_size = 2_i32.pow(log2 as u32) as usize;
let (_solutions, _generation, _progress, population) = GeneticExecution::<u8, QueensBoard>::new()
.population_size(population_size)
.genotype_size(n_queens as u8)
.mutation_rate(Box::new(MutationRates::Linear(SlopeParams {
start: f64::from(n_queens) / (8_f64 + 2_f64 * log2) / 100_f64,
bound: 0.005,
coefficient: -0.0002,
})))
.selection_rate(Box::new(SelectionRates::Linear(SlopeParams {
start: log2 - 2_f64,
bound: log2 / 1.5,
coefficient: -0.0005,
})))
.select_function(Box::new(SelectionFunctions::Cup))
.age_function(Box::new(AgeFunctions::Quadratic(
AgeThreshold(50),
AgeSlope(1_f64),
)))
.stop_criterion(Box::new(StopCriteria::Generation(10000)))
.run();
let (solutions, generation, progress, _population) = GeneticExecution::<u8, QueensBoard>::new()
.population_size(population_size)
.genotype_size(n_queens as u8)
.mutation_rate(Box::new(MutationRates::Linear(SlopeParams {
start: f64::from(n_queens) / (8_f64 + 4_f64 * log2) / 100_f64,
bound: 0.005,
coefficient: -0.0002,
})))
.selection_rate(Box::new(SelectionRates::Linear(SlopeParams {
start: log2 - 4_f64,
bound: log2 / 1.5,
coefficient: -0.0005,
})))
.select_function(Box::new(SelectionFunctions::Cup))
.age_function(Box::new(AgeFunctions::Quadratic(
AgeThreshold(50),
AgeSlope(1_f64),
)))
.population(population)
.progress_log(20, progress_log)
.population_log(2000, population_log)
.run();
构建
要构建 oxigen,请使用 cargo
,就像任何 Rust 项目一样
cargo build
以调试模式构建。cargo build --release
以优化方式构建。
要运行基准测试,您需要一个 nightly Rust 编译器。取消注释 // #![feature(test)]
和 // mod benchmarks;
从 lib.rs
,然后可以使用 cargo bench
运行基准测试。
基准测试
以下基准测试已创建,用于测量遗传算法函数的性能
running 25 tests
test benchmarks::bench_cross_multi_point_255inds ... bench: 348,371 ns/iter (+/- 10,506)
test benchmarks::bench_cross_single_point_255inds ... bench: 113,986 ns/iter (+/- 10,657)
test benchmarks::bench_cross_uniform_255inds ... bench: 88,426 ns/iter (+/- 2,302)
test benchmarks::bench_distance_255 ... bench: 20,715 ns/iter (+/- 1,648)
test benchmarks::bench_fitness_1024inds ... bench: 377,344 ns/iter (+/- 8,617)
test benchmarks::bench_fitness_age_1024inds ... bench: 31,360 ns/iter (+/- 1,204)
test benchmarks::bench_get_fitnesses_1024inds ... bench: 18,951 ns/iter (+/- 868)
test benchmarks::bench_get_solutions_1024inds ... bench: 30,133 ns/iter (+/- 1,612)
test benchmarks::bench_mutation_1024inds ... bench: 13 ns/iter (+/- 0)
test benchmarks::bench_not_cached_fitness_1024inds ... bench: 373,966 ns/iter (+/- 60,244)
test benchmarks::bench_not_cached_fitness_age_1024inds ... bench: 395,056 ns/iter (+/- 24,407)
test benchmarks::bench_selection_cup_255inds ... bench: 344,873 ns/iter (+/- 40,519)
test benchmarks::bench_selection_roulette_256inds ... bench: 140,994 ns/iter (+/- 1,294)
test benchmarks::bench_selection_tournaments_256inds ... bench: 420,272 ns/iter (+/- 49,178)
test benchmarks::bench_survival_pressure_children_fight_most_similar_255inds ... bench: 14,948,961 ns/iter (+/- 989,864)
test benchmarks::bench_survival_pressure_children_fight_parents_255inds ... bench: 108,691 ns/iter (+/- 157)
test benchmarks::bench_survival_pressure_children_replace_most_similar_255inds ... bench: 14,935,080 ns/iter (+/- 1,221,611)
test benchmarks::bench_survival_pressure_children_replace_parents_255inds ... bench: 167,392 ns/iter (+/- 15,839)
test benchmarks::bench_survival_pressure_children_replace_parents_and_the_rest_most_similar_255inds ... bench: 737,347,573 ns/iter (+/- 16,773,890)
test benchmarks::bench_survival_pressure_children_replace_parents_and_the_rest_random_most_similar_255inds ... bench: 7,757,258 ns/iter (+/- 612,047)
test benchmarks::bench_survival_pressure_competitive_overpopulation_255inds ... bench: 10,727,219 ns/iter (+/- 770,681)
test benchmarks::bench_survival_pressure_deterministic_overpopulation_255inds ... bench: 173,745 ns/iter (+/- 609)
test benchmarks::bench_survival_pressure_overpopulation_255inds ... bench: 10,710,336 ns/iter (+/- 657,369)
test benchmarks::bench_survival_pressure_worst_255inds ... bench: 20,318 ns/iter (+/- 298)
test benchmarks::bench_update_progress_1024inds ... bench: 7,424 ns/iter (+/- 273)
这些基准测试在一个 Intel Core i7 6700K、16 GB DDR4 和 512 GB 三星 950 Pro NVMe SSD(ext4 格式)的 Fedora 30 系统上,使用 Linux 内核 5.2.9 执行。
不同适应度基准测试之间的性能差异有以下解释
bench_fitness
测量缓存执行的性能,在每次基准测试迭代后清理适应度。这种清理是比未缓存基准测试慢的原因。bench_fitness_age
测量在所有基准测试迭代中缓存适应度的性能,因此它非常快。- 未缓存的基准测试测量未缓存的执行性能,在最后一种情况下,有 1 代个体,因此性能相似,但必须应用年龄不适应度功能的基准测试要慢一些。
children_fight_most_similar
和children_replace_most_similar
函数需要调用距离函数c * p
次,其中c
是子代数量,p
是种群大小(基准测试中分别为 255 和 1024)。overpopulation
和competitive_overpopulation
函数与children_replace_most_similar
和children_fight_most_similar
类似,只是它们只与种群中的m
个个体进行比较(m
大于子代数量,小于种群大小,基准测试中为 768)。因此,与children_replace_most_similar
和children_fight_most_similar
相比,这些基准测试中有 3/4 的比较是在这些基准测试中完成的。children_replace_parents_and_the_rest_random_most_similar
与children_replace_parents
类似,但在其后,随机选择个体与种群中最相似的个体进行对抗,直到种群大小恢复到原始大小。这意味着在整个种群中,根据每一代的重复父代进行0到254次随机选择和距离计算。children_replace_parents_and_the_rest_most_similar
类似于前面的函数,但它搜索种群中最相似的个体对,这意味着需要进行 p2 次距离函数调用(基准测试中为 220 次)。
贡献
我们绝对欢迎并鼓励贡献!贡献可以有多种形式。您可以
- 通过 issue 提交功能请求或错误报告。
- 通过 issue 要求改进文档。
- 对需要反馈的问题进行评论。
- 通过 pull requests 贡献代码,提交 PR 之前别忘了运行
cargo fmt
!
我们旨在保持 Rocket 代码的质量处于最高水平。这意味着您贡献的任何代码都必须是
- 注释:公共项必须进行注释。
- 文档化:公开的项必须有 rustdoc 注释,如有适用,需包含示例。
- 格式化:尽可能将您的代码格式化为
rustfmt
。 - 简单:您的代码应尽可能简单且符合惯例地完成任务。
- 测试:当可能时,您应添加(并通过)任何您添加的功能的令人信服的测试。
- 专注:您的代码应仅执行其应执行的任务。
请注意,除非您明确声明,否则您提交给 oxigen 的任何有意贡献将根据 MIT 许可证和 Apache 许可证版本 2.0 双重许可,不附加任何额外条款或条件。
参考
Pozo, M.M. "Oxigen:使用 Rust 编写的快速、并行、可扩展和适应性强的遗传算法库"。
Bibtex
@misc{
title={Oxigen: Fast, parallel, extensible and adaptable genetic algorithm library written in Rust},
author={Pozo M.M.},
howpublised = "\url{https://github.com/Martin1887/oxigen}"
}
许可证
oxigen 根据 Mozilla 公共许可证 2.0 许可。
依赖关系
~2MB
~36K SLoC