Source code for desdeo_emo.EAs.PPGA

from random import choice, sample
import numpy as np

# from pygmo import fast_non_dominated_sorting as nds
from desdeo_tools.utilities import fast_non_dominated_sort as nds
from desdeo_emo.EAs.BaseEA import BaseEA, eaError


[docs]class PPGA(BaseEA): """Predatory-Prey genetic algorithm. A population of prey signify the various models or solutions to the problem at hand. Weaker prey, i.e. bad models or solutions, are killed by predators. The predators and prey are placed in a lattice, in which they are free to roam. In each generation, each predator gets a certain number of turns to move about and hunt in its neighbourhood, killing the weaker prey, according to a fitness criteria. After this, each prey gets a certain number of moves to pursue a random walk and to reproduce with other prey. Each reproduction step generates two new prey from two parents, by crossing over their attributes and adding random mutations. After each prey has completed its move, the whole process starts again. As the weaker individuals get eliminated in each generation, the population as a whole becomes more fit, i.e. the individuals get closer to the true pareto-optimal solutions. If you have any questions about the code, please contact: Bhupinder Saini: bhupinder.s.saini@jyu.fi Project researcher at University of Jyväskylä. Parameters ---------- population : object The population object Notes ----- The algorithm has been created earlier in MATLAB, and this Python implementation has been using that code as a basis. See references [4] for the study during which the original MATLAB version was created. Python code has been written by Niko Rissanen under the supervision of professor Nirupam Chakraborti. For the MATLAB implementation, see: N. Chakraborti. Data-Driven Bi-Objective Genetic Algorithms EvoNN and BioGP and Their Applications in Metallurgical and Materials Domain. In Datta, Shubhabrata, Davim, J. Paulo (eds.), Computational Approaches to Materials Design: Theoretical and Practical Aspects, pp. 346-369, 2016. References ---------- [1] Laumanns, M., Rudolph, G., & Schwefel, H. P. (1998). A spatial predator-prey approach to multi-objective optimization: A preliminary study. In International Conference on Parallel Problem Solving from Nature (pp. 241-249). Springer, Berlin, Heidelberg. [2] Li, X. (2003). A real-coded predator-prey genetic algorithm for multiobjective optimization. In International Conference on Evolutionary Multi-Criterion Optimization (pp. 207-221). Springer, Berlin, Heidelberg. [3] Chakraborti, N. (2014). Strategies for evolutionary data driven modeling in chemical and metallurgical Systems. In Applications of Metaheuristics in Process Engineering (pp. 89-122). Springer, Cham. [4] Pettersson, F., Chakraborti, N., & Saxén, H. (2007). A genetic algorithms based multi-objective neural net applied to noisy blast furnace data. Applied Soft Computing, 7(1), 387-397. """ def __init__( self, problem, population_size: int = 100, population_params=None, initial_population=None, n_iterations: int = 10, n_gen_per_iter: int = 10, predator_pop_size: int = 50, prey_max_moves: int = 10, prob_prey_move: float = 0.3, offspring_place_attempts: int = 10, kill_interval: int = 7, max_rank: int = 20, neighbourhood_radius: int = 3, ): super().__init__(n_gen_per_iter=n_gen_per_iter, n_iterations=n_iterations) if initial_population is None: msg = "Provide initial population" raise eaError(msg) self.population = initial_population self.target_pop_size = population_size self.predator_pop_size: int = predator_pop_size self.prey_max_moves: int = prey_max_moves self.prob_prey_move: float = prob_prey_move self.offspring_place_attempts: int = offspring_place_attempts self.kill_interval: int = kill_interval self.max_rank: int = max_rank self.neighbourhood_radius: int = neighbourhood_radius self.lattice = Lattice( size_x=60, size_y=60, population=self.population, predator_pop_size=predator_pop_size, target_pop_size=self.target_pop_size, prob_prey_move=prob_prey_move, prey_max_moves=prey_max_moves, offspring_place_attempts=offspring_place_attempts, neighbourhood_radius=neighbourhood_radius, )
[docs] def _next_gen(self): """Run one generation of PPGA. Intended to be used by next_iteration. Parameters ---------- population: "Population" Population object """ # Move prey and select neighbours for breeding mating_pop = self.lattice.move_prey() offspring = self.population.mate(mating_pop) # Try to place the offspring to lattice, add to population if successful placed_indices = self.lattice.place_offspring(len(offspring)) # Remove from offsprings the ones that didn't get placed mask = np.ones(len(offspring), dtype=bool) mask[placed_indices] = False offspring = np.asarray(offspring)[~mask] # Add the successfully placed offspring to the population self.population.add(offspring) self._current_gen_count += 1 self._gen_count_in_curr_iteration += 1 self._function_evaluation_count += offspring.shape[0] # Kill bad individuals every n generations if self._current_gen_count % self.kill_interval == 0: selected = self.select(self.population, self.max_rank) self.lattice.update_lattice(selected) self.population.delete(selected) # Move predators self.lattice.move_predator()
[docs] def select(self, population, max_rank=20) -> list: """Of the population, individuals lower than max_rank are selected. Return indices of selected individuals. Parameters ---------- population : Population Contains the current population and problem information. max_rank : int Select only individuals lower than max_rank Returns ------- list List of indices of individuals to be selected. """ # Calculating fronts and ranks # _, _, _, rank = nds(population.fitness) fronts = nds(population.fitness) index_ranks = np.argwhere(fronts) rank = np.full(fronts.shape[1], np.inf, dtype=int) rank[index_ranks[:, 1]] = index_ranks[:, 0] selection = np.nonzero(rank > max_rank) return selection[0]
[docs] def manage_preferences(self, preference=None): return
[docs]class Lattice: """The 2-dimensional toroidal lattice in which the predators and prey are placed. Attributes ---------- size_x : int Width of the lattice. size_y : int Height of the lattice. lattice : ndarray 2d array for the lattice. predator_pop : ndarray The predator population. predators_loc : list Location (x, y) of predators on the lattice. preys_loc : list Location (x, y) of preys on the lattice. """ def __init__( self, size_x, size_y, population, predator_pop_size, target_pop_size, prob_prey_move, prey_max_moves, offspring_place_attempts, neighbourhood_radius, ): self.size_x = size_x self.size_y = size_y self.population = population self.predator_pop_size = predator_pop_size self.target_pop_size = target_pop_size self.prob_prey_move = prob_prey_move self.prey_max_moves = prey_max_moves self.offspring_place_attempts = offspring_place_attempts self.neighbourhood_radius = neighbourhood_radius self.lattice = np.zeros((self.size_y, self.size_x), int) self.predator_pop = np.empty((0, 1)) self.predators_loc = [] self.preys_loc = [] self.mating_pop = [] self.init_predators() self.init_prey()
[docs] def init_predators(self): """Initialize the predator population, linearly distributed in [0,1] and place them in the lattice randomly.""" # Initialize the predator population self.predator_pop = np.linspace(0, 1, num=self.predator_pop_size) # Take random indices from free (==zero) lattice spaces free_space = np.transpose(np.nonzero(self.lattice == 0)) indices = sample(free_space.tolist(), self.predator_pop.shape[0]) for i in range(self.predator_pop.shape[0]): # Keep track of predator locations in a list self.predators_loc.append([indices[i][0], indices[i][1]]) # +1 to offset zero index individual, set negative number to # identify predators from prey in the lattice self.lattice[indices[i][0]][indices[i][1]] = int(-1 * (i + 1))
[docs] def init_prey(self): """Find an empty position in the lattice and place the prey.""" # Take random indices from free (==zero) lattice spaces free_space = np.transpose(np.nonzero(self.lattice == 0)) indices = sample(free_space.tolist(), len(self.population.individuals)) for i in range(len(self.population.individuals)): # Keep track of preys in a list self.preys_loc.append([indices[i][0], indices[i][1]]) # +1 to offset zero index individual self.lattice[indices[i][0]][indices[i][1]] = int(i + 1)
[docs] def move_prey(self): """Find an empty position in prey neighbourhood for the prey to move in, and choose a mate for breeding if any available. Returns ------- mating_pop : list List of parent indices to use for mating """ mating_pop = [] for prey, pos in enumerate(self.preys_loc): if np.random.random() < self.prob_prey_move: for i in range(self.prey_max_moves): neighbours = self.neighbours(self.lattice, pos[0], pos[1]) dy = np.random.randint(neighbours.shape[0]) dx = np.random.randint(neighbours.shape[1]) # If neighbouring cell is occupied, skip turn if neighbours[dy][dx] != 0: continue dest_y = dy - 1 + pos[0] dest_x = dx - 1 + pos[1] # Check boundaries of the lattice if dest_y not in range(self.size_y) or dest_x not in range( self.size_x ): dest_y, dest_x = self.lattice_wrap_idx( (dest_y, dest_x), np.shape(self.lattice) ) # Move prey, clear previous location self.lattice[dest_y][dest_x] = int(prey + 1) self.lattice[pos[0]][pos[1]] = 0 # Update prey location in the list pos[0], pos[1] = dest_y, dest_x neighbours = self.neighbours( self.lattice, self.preys_loc[prey][0], self.preys_loc[prey][1] ) mates = neighbours[(neighbours > 0) & (neighbours != prey + 1)] if len(mates) < 1: continue else: # -1 for lattice offset mate = int(choice(mates)) - 1 mating_pop.append([prey, mate]) if mating_pop == []: raise eaError("What's ahppening?!") return mating_pop
[docs] def place_offspring(self, offspring): """Try to place the offsprings to the lattice. If no empty spot found within number of max attempts, do not place. Parameters ---------- offspring : int number of offsprings Returns ------- list Successfully placed offspring indices. """ # Keep track of offspring in a list placed_offspring = [] for i in range(offspring): y = np.random.randint(self.size_y) x = np.random.randint(self.size_x) for j in range(self.offspring_place_attempts): if self.lattice[y][x] != 0: continue else: if self.lattice[y][x] == 0: # Append the offspring to the list of preys. # len(self.preys_loc) is the index of the current # last prey in the list self.lattice[y][x] = int(len(self.preys_loc) + 1) self.preys_loc.append([y, x]) placed_offspring.append(i) return placed_offspring
[docs] def move_predator(self): """Find an empty position in the predator neighbourhood for the predators to move in, move the predator and kill the weakest prey in its neighbourhood, if any. Repeat until > predator_max_moves.""" predator_max_moves = int( (len(self.population.individuals) - self.target_pop_size) / self.predator_pop_size ) # Track killed preys in list and remove them at the end of the function to_be_killed = [] for predator, pos in enumerate(self.predators_loc): for i in range(predator_max_moves): neighbours = self.neighbours( self.lattice, pos[0], pos[1], n=self.neighbourhood_radius ) targets = neighbours[neighbours > 0] # If preys found in the neighbourhood, # calculate their fitness and kill the weakest if len(targets) > 0: fitness = [] weakest_prey = None for target in targets: obj1 = self.population.fitness[target - 1][0] obj2 = self.population.fitness[target - 1][1] fc = ( self.predator_pop[predator] * obj1 + (1 - self.predator_pop[predator]) * obj2 ) fitness.append((fc, target)) fitness.sort() weakest_prey = fitness[-1][1] - 1 # Kill the weakest prey and move the predator to its place self.lattice[self.preys_loc[weakest_prey][0]][ self.preys_loc[weakest_prey][1] ] = -1 * (predator + 1) # Set the old predator location to zero self.lattice[pos[0]][pos[1]] = 0 # Update predator location in the list of predators pos[0], pos[1] = ( self.preys_loc[weakest_prey][0], self.preys_loc[weakest_prey][1], ) # Set the killed prey as dead in the list of prey locations and # end predator turn to_be_killed.append(weakest_prey) self.preys_loc[weakest_prey] = None else: dy = np.random.randint(neighbours.shape[0]) dx = np.random.randint(neighbours.shape[1]) # If neighbouring cell is occupied by another predator, skip turn if neighbours[dy][dx] < 0: continue dest_y = dy - 1 + pos[0] dest_x = dx - 1 + pos[1] # Check boundaries of the lattice if dest_y not in range(self.size_y) or dest_x not in range( self.size_x ): dest_y, dest_x = self.lattice_wrap_idx( (dest_y, dest_x), np.shape(self.lattice) ) # Move predator, clear previous location self.lattice[dest_y][dest_x] = -1 * (predator + 1) self.lattice[pos[0]][pos[1]] = 0 # Update predator location in the list pos[0], pos[1] = dest_y, dest_x # Remove killed prey from population self.population.delete(to_be_killed) self.update_lattice()
[docs] def update_lattice(self, selected=None): """Update prey positions in the lattice. Parameters ---------- selected : list Indices of preys to be removed from the lattice. """ # Remove selected individuals from the lattice if selected is not None: for i in selected: self.lattice[self.preys_loc[i][0]][self.preys_loc[i][1]] = 0 self.preys_loc[i] = None # Update the list of prey locations updated_preys = [x for x in self.preys_loc if x is not None] self.preys_loc = updated_preys # Update lattice for prey, pos in enumerate(self.preys_loc): self.lattice[pos[0]][pos[1]] = prey + 1
@staticmethod
[docs] def lattice_wrap_idx(index, lattice_shape): """Returns periodic lattice index for a given iterable index. Parameters ---------- index : tuple one integer for each axis lattice_shape : tuple the shape of the lattice to index to """ if not hasattr(index, "__iter__"): return index # handle integer slices if len(index) != len(lattice_shape): return index # must reference a scalar if any(type(i) == slice for i in index): return index # slices not supported if len(index) == len(lattice_shape): # periodic indexing of scalars mod_index = tuple(((i % s + s) % s for i, s in zip(index, lattice_shape))) return mod_index raise ValueError("Unexpected index: {}".format(index))
@staticmethod
[docs] def neighbours(arr, x, y, n=3): """Given a 2D-array, returns an n*n array whose "center" element is arr[x,y] Parameters ---------- arr : ndarray A 2D-array where to get the neighbouring cells x : int X coordinate for the center element y : int Y coordinate for the center element n : int Radius of the neighbourhood Returns ------- The neighbouring cells of x, y in radius n*n. Defaults to Moore neighbourhood (n=3). """ arr = np.roll(np.roll(arr, shift=-x + 1, axis=0), shift=-y + 1, axis=1) return arr[:n, :n]