Skip to main content
Advertisement
Browse Subject Areas
?

Click through the PLOS taxonomy to find articles in your field.

For more information about PLOS Subject Areas, click here.

  • Loading metrics

Reinforcement learning-controlled differential evolution with L-BFGS refinements

  • Yang Cao,

    Roles Conceptualization, Funding acquisition, Supervision, Writing – review & editing

    Affiliations School of Computer Science and Engineering, Shenyang Jianzhu University, Shenyang, China, Liaoning Provincial Key Laboratory of Big Data Management and Analysis of Urban Construction, Shenyang, China, Shenyang Branch of National Special Computer Engineering Technology Research Center, Shenyang, China

  • Bingchuan Wu ,

    Roles Data curation, Resources, Software, Writing – original draft

    183740157@qq.com

    Affiliations School of Computer Science and Engineering, Shenyang Jianzhu University, Shenyang, China, Liaoning Provincial Key Laboratory of Big Data Management and Analysis of Urban Construction, Shenyang, China, Shenyang Branch of National Special Computer Engineering Technology Research Center, Shenyang, China

  • Miao Wen

    Roles Investigation, Validation, Visualization

    Affiliations School of Computer Science and Engineering, Shenyang Jianzhu University, Shenyang, China, Liaoning Provincial Key Laboratory of Big Data Management and Analysis of Urban Construction, Shenyang, China, Shenyang Branch of National Special Computer Engineering Technology Research Center, Shenyang, China

Abstract

Differential evolution has become one of the mainstream solvers for complex optimization problems due to its concise structure and strong global search ability. However, the performance of the DE algorithm is highly sensitive to its mutation and crossover strategies and related control parameters. Traditional adaptive strategies are mostly based on heuristic rules and lack online learning capabilities, which limits their further application in expensive black box optimization scenarios. Therefore, this article proposes Reinforcement Learning-controlled Differential Evolution with Limited-memory Broyden-Fletcher-Goldfarb-Shanno Refinement (RL-DE). The algorithm has constructed a three-layer closed-loop framework of “parameters policy refinement”, achieving a paradigm shift from rule driven to data-driven. The system evaluation on the Congress on Evolutionary Computation 2017 (CEC2017) benchmark test set shows that the algorithm is robust in high-dimensional scenarios and achieves better results compared to classical adaptive DE variants; Its successful application in flexible job shop scheduling problems also validates the generalization ability of the framework towards the field of discrete combinatorial optimization. This study provides a learnable unified architecture for parameter adaptation in DE, offering efficient and intelligent solutions for expensive black box optimization and engineering scheduling problems.

1. Introduction

Differential Evolution (DE), as a population-based stochastic optimization algorithm, has established its efficiency and wide application in solving complex, nonlinear, and non convex black box optimization problems [1]. Its success is attributed to the simple yet powerful mechanism of using vector differences to perturb the population, which gives it a strong ability to explore the global search space [2]. However, the performance of the standard DE algorithm is greatly influenced by its control parameter settings [3], including population size (NP), mutation factor (F), and crossover probability (CR). The optimal configuration of these parameters not only depends on the specific problem, but may also vary during different optimization stages of solving the same problem. Finding a universally applicable optimal parameter setting is a very difficult and time-consuming task, often requiring a lot of trial and error work, which weakens the black box characteristics of the algorithm. This dependency has prompted extensive research to focus on developing adaptive or self-adjusting mechanisms to dynamically adjust these parameters during the search process. Although these methods have shown improvement, many of them rely on fixed, predefined rules or heuristic methods, which may not be flexible enough to adapt to the changing terrain in complex optimization problems. The challenge lies in creating a parameter control mechanism that can intelligently learn and adapt to the specific characteristics of the problem at hand in real-time, thereby maximizing the efficiency and robustness of the algorithm.

Reinforcement Learning (RL), as a powerful paradigm for solving sequential decision problems, provides a systematic framework for solving sequential decision problems, in which agents learn to make optimal decisions by interacting with the environment to maximize cumulative rewards [4]. This paradigm has been widely applied in the field of optimization, especially in the automation design and control of metaheuristic algorithms [5]. The core idea is to model the algorithm configuration or parameter control process as a Markov Decision Process (MDP), where the state represents the current state of the optimization process (such as population diversity, fitness terrain features), the action corresponds to the adjustment of algorithm parameters, and the reward comes from the improvement of solution quality or other performance indicators [6]. By using reinforcement learning techniques such as Q-learning [7] or policy gradient methods, agents can be trained to learn a parameter controlled optimal policy that dynamically adapts to the constantly evolving search terrain [8,9]. Compared to the discretization limitations of Q-learning, the policy gradient method can directly optimize the continuous parameter distribution, avoiding the curse of dimensionality, and is particularly suitable for continuous control problems such as DE parameters. This approach goes beyond static or heuristic based rules, allowing algorithms to learn optimization from experience, whether on specific problems or on a set of training problems. Integrating reinforcement learning into evolutionary algorithms such as differential evolution (DE) provides a promising direction for creating smarter and more autonomous optimization systems that can self adjust parameters to achieve superior performance without human intervention.

In recent years, it has been proven that reward shaping and curriculum learning can alleviate the problem of sparse rewards. By adjusting weights in stages, RL agents can remain robust in high-dimensional and multi-modal environments [10]; At the same time, the action space has expanded from simply adjusting F/CR to mutation strategy selection and even when to trigger L-BFGS. Notably, Tan et al. [11] proposed a differential evolution framework with hybrid parameters and mutation strategies based on reinforcement learning, which employs a hybrid policy network to simultaneously control mutation strategies and control parameters, demonstrating the effectiveness of continuous-continuous action space in DE optimization. Ding et al. [12] took the lead in implementing joint adaptive mutation operators and parameters within the DE framework using distributed proximal policy optimization. Yang et al. [13] further incorporated the timing of L-BFGS calls into RL decision-making, providing a scalable architecture template; In terms of multi-agent collaboration, Long et al. [14] systematically reviewed the progress of population level deep reinforcement learning, and Li Yao’s co evolutionary particle swarm framework [15] also showed that sub population information exchange can significantly improve high-dimensional exploration efficiency. Its divide and conquer approach lays a theoretical and empirical foundation for the future evolution of RL-DE towards multi-agent RL-DE. Furthermore, recent comprehensive surveys have systematically established the theoretical frameworks and research paradigms for reinforcement learning-assisted evolutionary algorithms (RL-EA), outlining the integration mechanisms between learning and optimization from both algorithmic design and application perspectives [16].

To address the sensitivity of differential evolution algorithm to parameters in complex optimization tasks, researchers have proposed various adaptive mechanisms. Specifically, JADE [17] demonstrates better convergence performance on various test functions by maintaining external archive and adaptively updating F and CR based on successful historical parameters; SaDE [18] learns mutation strategy selection and adaptive probability distributions of F and CR in real-time during the evolution process, forming an adaptive paradigm based on success history; SHADE [19] has become the basic template for high-performance DE variants by continuously archiving F and CR values and updating them using Lehmer mean; ILSHADE [20] introduces linear population size reduction (LPSR) and stricter external archive maintenance on this basis; JSO [21] further adopts a two-stage mutation pool and improved archive utilization, which is widely recognized as the strongest pure DE algorithm currently available and often serves as a benchmark for performance cap; MPEDE [22] divides the population into three subpopulations and configures different mutation strategies, while dynamically adjusting resources according to contribution, which can horizontally compare the simplicity and efficiency of RL-DE’s single population six dimensional continuous control.

Before the widespread application of reinforcement learning, researchers also attempted to dynamically adjust parameters through fuzzy logic, coding evolution, or random perturbations in order to improve algorithm robustness without the need for human intervention. Compared to fixed value schemes, FADE [3] uses fuzzy logic controllers to adjust F and CR online, making parameter changes more in line with population diversity fluctuations; JDE [23] encodes F and CR into individual genes and evolves synchronously with decision variables to achieve self optimizing parameter updates; RADE samples F through a simple random distribution, providing persistent perturbations without the need for additional storage; The dynamic population DE gradually reduces NP based on convergence progress, while maintaining exploration ability and reducing later computational costs. In addition, Brest et al. [23] systematically studied the impact of control parameter adaptivity on DE performance and verified the effectiveness of parameter adaptivity mechanisms on various numerical benchmark problems; Boggs and Byrd [24] proposed an adaptive finite memory BFGS algorithm from the perspective of quasi-Newton method, providing theoretical basis and algorithm foundation for the subsequent fusion of L-BFGS and DE.

Although significant progress has been made in parameter adaptation in the above research, its limitations are still evident: on the one hand, mainstream methods such as JADE, SaDE, SHADE generally rely on fixed mathematical formulas or statistical models to update F and CR, making it difficult to capture complex nonlinear dynamics in real-time during the optimization process; On the other hand, although early improvements such as FADE, jDE, RADE introduced ideas such as fuzzy logic, coding evolution, or random perturbations, the adjustment policy still belongs to the heuristic category, lacks learning mechanisms, and cannot dynamically optimize decisions based on historical experience, resulting in being easily trapped in inefficient parameter regions in complex or multimodal problems, making it difficult to achieve true intelligent control.

Based on the limitations of the above research, this article proposes RL-DE Algorithm, through the collaboration of reinforcement learning+quasi-Newton dual engines, achieves a leap in parameter adjustment from preset to self-learning. Its core contributions are as follows:

  1. (1) Propose PG-Tuner: a policy gradient based reinforcement learning parameter controller that incorporates six dynamic parameters of differential evolution (DE) – mutation factor F, crossover probability CR, pBest selection ratio p, weight scaling factor Fw, archive disturbance intensity archF, archive usage probability archP - into a unified continuous action space, and models it as a Markov decision process through neural networks. It is worth noting that PG-Tuner focuses on adaptive adjustment of these dynamic parameters, while population size NP is pre specified by the user as a static hyperparameter based on the complexity of the problem. The controller uses the policy gradient method to update the policy online, which can synchronously adjust the six dimensional parameters during operation and gradually release the coupling potential between parameters; No need to rely on manual presets or fixed rules, allowing parameters to self adjust with the search process, providing support for the transition from global coarse exploration to local fine development.
  2. (2) Propose a Two-Stage Mutation Switching mechanism: Design a dual-mode mutation strategy, using a globally exploratory strategy in the early stage (first 50% iterations) and switching to a locally exploratory strategy in the later stage. The switching threshold is based on a fixed iteration progress setting, which preserves the adaptive ability of the policy while avoiding the introduction of additional RL training overhead. It forms orthogonal complementarity with six parameter adjustment and constructs a parameter strategy dual layer collaborative framework, providing a feasible new idea for improving the search efficiency and convergence stability of complex multimodal problems.
  3. (3) Propose a local refinement module LB-Refiner based on the L-BFGS algorithm: This module performs local refinement calculations with a fixed probability in the last 10% iterations of the DE run, quickly polishing the optimal or elite individuals. After the global search has locked in high-quality areas, it improves the accuracy of the solution at low cost and strengthens the algorithm’s local search ability.

The rest of this article is arranged as follows: Chapter 2 introduces the theoretical background and related work of differential evolution algorithm, policy gradient method, and L-BFGS algorithm; Chapter 3 elaborates on the proposed RL-DE algorithm, including the design and implementation of the PG-Tuner parameter controller, two-stage mutation strategy switching mechanism, and LB-Refiner local refinement module; Chapter 4 conducts a comprehensive experimental evaluation using the CEC 2014/2017 benchmark test function, and compares its performance, ablation experiments, and hyperparameter sensitivity analysis with multiple representative algorithms; Chapter 5 applies RL-DE to practical problems in flexible job shop scheduling and verifies its effectiveness in high-dimensional optimization scenarios; Chapter 6 summarizes the research findings and explores the limitations of the current framework and future research directions.

2. Theoretical background and related work

2.1. Differential evolution (DE) algorithm

Differential Evolution (DE) evolves populations through iterative execution of mutation, crossover, and selection operations [25].

  1. (1) Mutation operation

For each target vector (i = 1,2,..., NP) in the population, perform mutation to generate a mutation vector vi according to the following strategy:

(1)

Among them, are three random individuals that are different from each other and have a different index i from the target vector; F∈(0, 1] is the scaling factor; NP is the population size.

  1. (2) Cross operation

Perform binomial crossover on the target vector and the variation vector to generate the experimental vector :

(2)

Among them, is the dimension index; is a uniformly distributed random number; is the cross probability; jrand is a dimension index randomly selected from .

  1. (3) Select operation

Adopting a greedy selection strategy to retain better individuals:

(3)

Among them, f(.) is the objective function; is a new individual of the t + 1 generation. The above iterative process drives the population to move towards a more optimal region.

2.2. Policy gradient method

2.2.1. Markov decision process.

Markov Decision Process (MDP) is a standard mathematical framework for describing sequential decision problems [26], consisting of five tuples (S, A, P, R, γ): S is the state space that summarizes the current observable information of the system; A is the action space, defining the control variables that decision-makers can choose from; P provides the transition probability of state action pairs to the next state, characterizing the uncertainty of the environment; R provides real-time profit signals for evaluating the quality of actions; γ∈[0,1] is a discount factor used to balance immediate and future rewards.

The state transition probability formula of Markov decision process is as follows:

(4)

Among them, st ∈ S is the state of step t; at∈A is the action performed at step t; st+1 is the next state after the transition.

The probability of the reward function formula is as follows:

(5)

Among them, rt ∈ R is the immediate reward obtained in step t; E [.] represents the expected operator.

The goal is to find the optimal policy π*(a | s) to maximize the expected cumulative return:

(6)

Among them, π(a | s) is the policy function, representing the probability of selecting action a in state s; Eπ[.] represents the expectation under policy π; This objective function takes into account both immediate rewards and long-term discount rewards [26].

2.2.2. Gradient of continuous control policy.

The policy gradient method solves the continuous control MDP [27] by directly optimizing the parameterized policy πθ(a | s), where θ ∈ Rd is the policy network parameter. This method uses the policy gradient theorem to calculate the gradient of the objective function J(θ) with respect to the policy parameters:

(7)

Among them, is the state action value function, which represents the expected cumulative return following policy π after executing action a in state s; is the logarithmic policy gradient, which guides the direction of parameter updates.

To reduce the variance of gradient estimation, the dominance function Aπ(s, a) is used instead of Qπ(s, a) in practice:

(8)

Among them, is the state value function, representing the expected return of state s. The advantage function highlights the superiority or inferiority of action a relative to the average level by subtracting the baseline Vπ(s).

Based on Monte Carlo sampling, the policy gradient can be approximated as:

(9)

Among them, N is the number of sampling trajectories; T is the maximum number of steps per trajectory; is the discount return for the i-th trajectory at step t. This estimation enables the policy network to learn optimization from experience and achieve adaptive fine control of DE parameters [28].

2.3. L-BFGS algorithm

L-BFGS(Limited-memory Broyden-Fletcher-Goldfarb-Shanno) Algorithm [29] is an efficient quasi-Newton optimization method specifically designed to solve large-scale unconstrained optimization problems. It is an improved version of the BFGS algorithm that reduces memory usage, allowing it to handle optimization problems with a large number of variables.

The core idea of the L-BFGS algorithm is to accelerate the convergence speed of gradient descent by approximating the inverse of the Hessian matrix. Unlike the traditional BFGS algorithm, the L-BFGS algorithm only stores the gradient information and position change information of the last few iterations, rather than storing the complete inverse of the Hessian matrix. This greatly reduces memory requirements, making it suitable for large-scale problems.

The iterative core of L-BFGS lies in efficiently calculating the search direction and updating the approximation matrix. Each iteration mainly consists of three interrelated steps:

  1. (1) Calculation of search direction

In the kth iteration, the algorithm first approximates the search direction dk based on the gradient gk of the current point and the inverse of the Hessian matrix Hk, and obtains the search direction dk through simple multiplication:

(10)

Among them, is the gradient at the current point xk, and Hk is an approximation of the inverse of the Hessian matrix.

  1. (2) Update of position and gradient

After determining the search direction, the algorithm determines the step size αk through line search (satisfying the Wolfe condition) and updates the position and gradient:

(11)(12)
  1. (3) Implicit update of approximate matrix

L-BFGS does not directly store Hk, but implicitly maintains matrix information by storing the last m pairs of {si, yi} vectors (i = k-m,..., k-1). among which

(13)(14)

These vectors capture key information about the curvature of the objective function. Based on them, BFGS updated the approximate formula for the inverse of the Heisenberg matrix:

(15)

Among them, the scaling factor is used to ensure scale invariance.

To avoid explicit calculation and storage of Hk, L-BFGS uses a double loop recursive algorithm to directly calculate Hkgk, with the following process:

Forward loop: Starting from the most recent iteration, traverse in reverse, calculate intermediate variables and update , gradually stripping away the influence of historical curvature information.

Scaling steps: Use the initial scale to scale q and obtain the initial search direction r.

Reverse loop: Starting from the farthest iteration, traverse forward, calculate and update , re inject historical curvature information, and finally obtain the accurate search direction , which will be directly used for position update (see formula 11).

3. Proposed RL-DE algorithm

The RL-DE algorithm achieves a paradigm shift in differential evolution from heuristic control to intelligent learning through a three-layer collaborative framework of policy parameter refinement. The core idea is: the two-stage mutation strategy provides the basic search skeleton → PG-Tuner dynamically adjusts the six dimensional parameters → LB-Refiner locally refines the solution accuracy, and the three form a closed-loop feedback. The complete architecture and information flow of the three-layer collaborative framework are shown in Fig 1. This chapter first elaborates on the external archive and strategy switching of the two-stage mutation mechanism, then describes the neural network embedding and MDP modeling process of PG-Tuner, and finally explains the local refinement policy and overall algorithm flow of L-BFGS.

3.1. Dual stage mutation strategy switching mechanism

RL-DE balances global exploration and local development through a two-stage mutation strategy that switches at a fixed iteration threshold, with all dynamic parameters adaptively generated by PG-Tuner (Section 3.3). Together with the external archive, these form the search backbone (Fig 1). In the early stage, the DE/rand/2 + A strategy is employed, utilizing double difference vectors and archive perturbations to generate high-intensity randomness for exploring complex multimodal spaces. The mathematical expression for this strategy is:

(16)

Among them, t is the current iteration number and T is the total number of iterations; , , , , , and are six distinct individuals randomly selected from the current population P(t) at generation t, all different from the target individual ; the two difference vector pairs (-) and (-) are mutually exclusive (non-overlapping indices), providing anisotropic perturbations; archF∈[0, 1] is the archive disturbance intensity (scaling factor for archive-based perturbation), archP∈[0, 1] is the archive usage probability (probability of incorporating archive individuals), Xa is an individual randomly selected from the external archive set A, and rand~U(0, 1) is a uniform random number in [0, 1]; F is the mutation scaling factor (differential weight).

As the iteration enters the later stage, the algorithm switches to the DE/current to pBest-w/1 + A strategy, utilizing the current optimal solution information to achieve targeted fine development and accelerate the convergence of high-quality areas. The expression is:

(17)

The implementation details are as follows:

  1. (1) pBest selection: The p∈[0,1] value output by PG-Tuner is the pBest selection ratio, which is used to determine the number of elite individuals . is randomly selected from the top pcnt optimal individuals with replacement;
  2. (2) Weighted guidance: Fw is the weight scaling factor output by PG-Tuner, specifically used to control the step strength of guiding the current individual to pBest;
  3. (3) Archive perturbation: Using probability archP to select and external archive individuals Xa from the pBest set to form a secondary perturbation, maintaining moderate diversity in the later stages of development.

The two-stage mutation strategy shares the same external archive A and randomly introduces archive individuals using probability archP to maintain diversity. The external archive set A enhances population diversity and guides algorithms to avoid invalid search areas through the mechanism of eliminating individuals for reuse. Its construction and maintenance follow the following rules:

  1. (1) Initialization and Capacity Management
  2. When the algorithm starts, the external archive is initialized to an empty list structure, with its maximum capacity set to the problem dimension D, that is, |A|max = D. This list structure supports dynamic append and random access to ensure access efficiency.
  3. (2) Eliminating individual storage mechanisms
  4. In each generation selection stage, when the experimental individual successfully replaces the target individual , the eliminated old individual is stored in the archive with an exponential decay probability. The probability psave(t) dynamically adjusts with the number of iterations t:
(18)
  1. Among them, pmax = 1.0 is the initial storage probability, pmin = 0.01 is the last generation storage probability, and the decay coefficient γarch = 0.98 controls the rate of probability decrease.
  2. (3) Archive overflow handling strategy
  3. When the file reaches the capacity limit, the first in first out strategy is no longer used, but the maximum and minimum distance replacement is performed to maintain population spatial diversity:

Let the set of individuals in the current file be , and the newly eliminated individual be Xnew. Firstly, calculate the minimum distance from the new individual to all individuals in the archive:

(19)

Calculate the minimum distance from each individual in the archive to the remaining individuals:

(20)

Identify the nearest pair of individual indices in the archive and randomly select one as a candidate replacement index . If is satisfied, indicating that the new individual is more dispersed in spatial distribution, then replace with Xnew; Otherwise, abandon storage. This strategy prioritizes preserving historical information with a more dispersed spatial distribution while controlling the size of archives, improving the efficiency of archive utilization.

3.2. Local refinement (LB-Refiner) module

PG-Tuner achieves efficient global search of differential evolution (DE) through dynamic parameter generation, but after locking high-quality regions in the global search, it is difficult to further improve the accuracy of the solution solely through parameter adjustment. To this end, the LB-Refiner module – a heuristic local refinement component based on the L-BFGS algorithm – is introduced to perform low-cost refinement in local areas, forming a closed-loop optimization of parameter control+local refinement. As shown in Fig 1, LB-Refiner intervenes as an independent module in the final stage of optimization, forming a three-layer collaborative framework with PG-Tuner and two-stage mutation.

The specific implementation of the LB-Refiner module in the RL-DE algorithm has been described in detail in Algorithm 3 (lines 23–28). When the iteration reaches the last 10% (t ≥ 0.9 T), each generation triggers LB-Refiner with a probability of LS_P = 0.2. If triggered, randomly select an individual from the current best or top 5% elite individual as the L-BFGS initial point, and use the L-BFGS algorithm to locally optimize the selected initial point (see Algorithm 1 for the specific process). After the optimization is completed, if the new solution is better than the population optimum, the external archive will be directly replaced and synchronously updated, and only one individual will be refined per generation to maintain population diversity. This policy is expected to improve convergence speed and resolution accuracy in local stages without the need for additional manual intervention.

Algorithm 1 LB-Refiner local refinement process

Input:

 • Initial point x0 (currently the best or elite individual in the population)

 • f(x) (fitness function)

 • g(x) (The gradient vector of f(x))

 • MaxIter (maximum number of L-BFGS iterations, maxIter=30)

 • εg (gradient norm convergence threshold, εg=1e-5)

Output:

 • Refine the solution x*

1: x ← x₀

2: g ← g(x)

3: If ||g|| ≤ εg then return x # The first-order optimum has been satisfied, and there is no need for local optimization

4: for k = 1 to maxIter do

5: d ← - Hk g # Calculate the search direction recursively in a double loop according to formula 10

6: α ← LineSearchWolfe(f, g, x, d) # satisfies formulas 11 and 12

7: x ← x + αd

8: g ← g(x) # Update gradient

9: Update memory for {sk, yk} # according to formula (13) (14)

10: If ||g|| ≤ εg then return x # Convergence judgment

11:end for

12:return x

3.3. Policy gradient tuner (PG-Tuner)

PG-Tuner is the intelligent decision-making core of RL-DE, which is essentially a fully connected neural network that learns the optimal control policy for six key parameters (F, CR, p, Fw, archF, archP) online through policy gradient method. As shown in Fig 1, the PG-Tuner takes the current population statistical characteristics as input and outputs a six dimensional parameter distribution, directly driving the DE engine’s mutation and crossover operations, forming a closed loop of state observation → policy decision-making → environmental feedback. Unlike traditional adaptive methods that rely on fixed statistical rules, PG-Tuner models parameter control as a continuous Markov decision process, achieving a transition from rule driven to data-driven. The core embedding mechanism is shown in Fig 2.

thumbnail
Fig 2. Schematic diagram of embedding PG-Tuner neural network into DE framework.

https://doi.org/10.1371/journal.pone.0347860.g002

Unlike traditional adaptive methods that rely on fixed statistical rules, PG-Tuner models parameter control directly as a continuous MDP, achieving a transition from rule driven to data-driven; To achieve this, it is necessary to first embed the neural network into the DE framework and then provide a complete formal description of MDP.

In the standard DE framework, parameters F, CR, etc. are usually generated by fixed rules or random perturbations, and this decoupling method ignores the nonlinear coupling effects between parameters. PG-Tuner embeds a parameterized neural network πθ to map the state observation st to a Gaussian distribution N(μ, σ2) of six parameters, achieving collaborative optimization among parameters. In each generation t, PG-Tuner takes the current population statistical characteristics as input and outputs the distribution parameters of six dimensional parameters at=[F, CR, p, Fw, archF, archP]; Obtain specific action values through reparameterization sampling, and guide DE mutation and crossover operations after truncation; After executing the operation, the environment (DE engine) returns a new state st+1 and a reward rt, forming a closed loop. This embedding method deeply integrates parameter control with DE evolution, making the policy network an inseparable decision brain of the algorithm.

3.3.1. MDP modeling.

Next, the DE parameter control problem will be formalized as a Markov decision process for direct optimization using the policy gradient method. To apply the policy gradient method, the DE parameter control problem needs to be formalized as a Markov decision process MDP=(S, A, P, R, γ). The specific construction of the quintuple is given in order below. First, the state space (S) is described.

The state vector st ∈ R8 is constructed from the population fitness distribution and search progress in each generation. Specifically, the state consists of seven normalized fitness statistics plus a logarithmic progress indicator, defined as follows:

  1. (1) Logarithmic scaling: Performing a unified logarithmic scaling on population fitness to compress differences in magnitude.
(21)
  1. Among them, sgn is the sign function; log represents the common logarithm base 10 (log10), the same throughout the text; To prevent it from being 0, an additional 1e-8 is added, the same throughout the text.
  2. (2) Statistical extraction: Extract 7 statistical measures on logarithmic fitness:
    • fmin: minimum fitness value
    • f25%: 25th percentile fitness (first quartile, Q1)
    • fmed: median fitness (50th percentile, Q2)
    • f75%: 75th percentile fitness (third quartile, Q3)
    • fmax: maximum fitness value
    • fmean: mean fitness value (average)
    • fstd: standard deviation of fitness distribution
(22)
  1. (3) Relative normalization: Based on the median, normalize the six statistical measures other than the median to make the state values independent of the problem scale.
(23)
  1. (4) Progress feature: Introducing search progress feature – taking the logarithm of the proportion of currently consumed generations to the total generations, reflecting the iteration progress.
  2. (5) Final state: Assemble into an 8-dimensional vector st – 6 normalized statistics+median fixed to 0 + logarithmic progression:
(24)

Among them, t/T represents the iteration progress, where t denotes the current iteration number and T denotes the total number of iterations.

Once the state is determined, the six dimensional continuous control variables are defined as a unified action vector at once. The action space is composed of six continuous parameter vectors:

(25)

Each component is controlled separately: F∈[0,1] is the variation factor, CR∈[0,1] is the crossover probability, p∈[0,1] is the pBest selection ratio, Fw∈[0,1] is the weight scaling factor, archF∈[0,1] is the archive disturbance intensity, and archP∈[0,1] is the archive usage probability. The policy network πθ outputs the Gaussian distribution mean μ ∈ R6 of each parameter, and then obtains the valid action values through sampling and truncation.

Once the action is executed, the population enters the next generation according to DE dynamics, which implies a state transition. The state transition is implicitly defined by the dynamics of the DE algorithm:

(26)

Among them, DE(.) represents the new state generated after performing one generation mutation, crossover, and selection operations. The transfer is deterministic, completely determined by the current population and parameters.

To guide policy learning, it is also necessary to immediately return a concise and effective reward signal in each generation. To balance computational efficiency and policy learnability, a minimalist reward design is adopted here. Each generation only monitors three front-end indicators of population fitness distribution – optimal value, 10% percentile value, and 25% percentile value, compares them with the corresponding indicators of the previous generation for relative improvement, and then logarithmically scales them as immediate rewards. This signal reflects the overall progress of the population within a single generation, enabling the reinforcement learning agent to quickly and stably capture the instantaneous changes in optimized terrain.

(27)

Among them, , , and represent the best fitness value, the 10th percentile, and the 25th percentile of the population at generation t, respectively; , , and denote the corresponding values at generation t-1 (previous generation); the superscripts (t) and (t-1) indicate the current and previous generation indices, respectively; log denotes the base-10 logarithm.

With instant rewards, the ultimate goal can be expressed as maximizing the expected discount cumulative return. The output action distribution of the policy network πθ is πθ(a | s)=N(μθ(s), σ2). The goal of PG-Tuner is to maximize the expected cumulative discount return from the initial state s0:

(28)

The trajectory is generated by the interaction between the policy πθ and the environment, with γ as the discount factor used to balance immediate and future rewards, making the agent more focused on long-term optimization performance. Here, γ = 0.9. Update the parameter θ through the policy gradient theorem.

3.3.2. Policy network training and embedding.

With precise definitions of state, action, and reward, the training details of the policy network can be derived. The policy network is trained using the REINFORCE algorithm, which is a classic policy gradient method. The training process includes running the DE algorithm using the current policy to complete a complete scenario (performing a complete optimization run on the problem), recording the sequence of states, actions, and rewards, estimating the policy gradient at the end of the scenario, and updating the network weight θ to maximize the expected cumulative reward. To this end, follow the three steps of network structure gradient calculation training process in sequence.

Firstly, the structure of the policy gradient network is explained. The parameterized policy πθ is implemented by a fully connected neural network, as shown in Fig 3. The network takes an 8-dimensional state vector st as input, and after two hidden layers (64 neurons per layer, ReLU activated), outputs a six dimensional Gaussian distribution mean vector μθ(st). The logarithmic standard deviation logσ is stored independently as a trainable parameter to ensure exploration stability.

thumbnail
Fig 3. PG-Tuner policy network architecture diagram.

Note: In Fig 3, (s1, s2,..., s8) are 8-dimensional state vectors, (y1, y2,... y6) are 6-dimensional control vectors, corresponding to (F, CR, p, Fw, archF, archP), and N(.) represents Gaussian sampling based on μ and σ, respectively.

https://doi.org/10.1371/journal.pone.0347860.g003

Once the network structure is established, the direction of parameter updates can be derived based on the policy gradient theorem. According to the policy gradient theorem, the gradient of the objective function J(θ) with respect to the policy parameters is:

(29)

To mitigate the high variance issue inherent in Monte Carlo policy gradient estimates, we employ the advantage function as a bias-free variance reduction technique [30]. Specifically, serves as a state-dependent baseline (also known as the critic) that estimates the expected return starting from state s, while quantifies the relative advantage of action a compared to the average performance at state s. This baseline subtraction significantly reduces the variance of gradient estimates without introducing bias, thereby preventing policy collapse in high-dimensional parameter optimization [31]. In practice, is approximated using the mean fitness of the current population as an unbiased estimator, which has been verified effective for maintaining convergence stability in our 6-dimensional continuous control scenario.

Based on Monte Carlo sampling, the gradient can be approximated as:

(30)

Among them, N is the number of sampled trajectories and is the discounted return. In practical implementation, Adam optimizer is used for stable updates.

The gradient formula has been clarified, and a complete training algorithm pseudocode will be provided for engineering implementation. Algorithm 2 provides the complete training process for PG-Tuner, ensuring that the policy learning process strictly corresponds to the DE evolution dynamics.

Algorithm 2 PG-Tuner training algorithm based on policy gradient

Input:

 • Policy network θ (fully connected neural network, randomly initialized)

 • Training function set M

 • The number of trajectories per function L

 • Number of iterations per trajectory T

 • Population size NP

 • Learning rate α

 • Discount factor γ

Output:

 • Trained policy network parameters θ

1: for epoch = 1 to MaxEpoch do

2:  for each function fm in M do

3:   Randomly initialize population P0 and calculate fitness f0

4:   for l = 1 to L do

5:    t ← 0

6:    repeat

7:     Build state st (according to formula 24)

8:     Forward propagation: (calculated from left to right according to the data in Fig 3)

9:     Sampling action

10:    Crop action to legal range ()

11:    Execute DE Generation 1:

12:    Calculate the reward rt (according to formula 27)

13:    Storage experience

14:    t ← t + 1

15:   until t ≥ T

16:  end for

17:  Update θ using REINFORCE (according to formula 30)

18:  end for

19:end for

20:return θ

Explanation: State construction, action sampling, and reward calculation all correspond strictly to the previous MDP modeling, ensuring that policy learning is evidence-based.

After training, embed the obtained policy network θ into the main loop, so that the six dimensional parameters are generated in real-time for each generation and updated synchronously with DE evolution. The trained PG-Tuner has been integrated into the main loop of Algorithm 3. The embedding logic corresponds to the following:

  1. (1) Each generation’s six dimensional parameters are generated in real-time by πθ – lines 14–16;
  2. (2) Six dimensional parameter driven algorithm two-stage mutation/crossover operation – lines 20–23;
  3. (3) Environmental feedback rewards and online fine-tuning of πθ every 10 generations – lines 28–32.

3.4. Algorithm flow

Algorithm 3 provides the complete process of RL-DE, clearly demonstrating the integration of PG-Tuner, two-stage mutation, and LB-Refiner.

Algorithm 3 RL-DE main algorithm process

Input:

 • Objective function f(x)

 • Population size NP

 • Maximum iteration times T

 • L-BFGS activation probability LS_P

 • Reinforcement learning policy network parameter θ

Output:

 • Optimal solution xbest

1: # Initialize

2: Randomly initialize population P0={x1, x2,..., xNP}

3: Initialize policy network πθ (randomly initialized)

4: Initialize external archive set A=∅

5: Initialize reward accumulator Raccum = 0

6: for t = 1 to T do

7: # PG-Tuner observes the current population status:

8: Construct state vector st (according to formula 24)

9:  Forward propagation: (calculated from left to right according to the data in Fig 3)

10: Sampling action

11: # DE engine executes a generation of evolution

12: if t ≤ 0.5 * MaxCycle then

13:  Using the DE/rand/2 + A strategy (Formula 16) for mutation

14: else

15:  Use DE/current to pBest-w/1 + A strategy (Formula 17) for mutation

16: Perform cross operation to generate test vectors

17: Perform selection operation to update population Pt

18: Update external archive set A based on probability (as described in section 3.1)

19: # Calculate rewards and update strategies

20: Calculate the immediate reward rt (according to formula 27)

21: Calculate cumulative rewards Raccum ← Raccum+rt

22: If t mod 10=0 then use policy gradient to update πθ (according to algorithm 1)

23: # Check LB-Refiner activation

24: if t ≥ 0.9 * T and rand() <LS_P then

25:  Select the current optimal solution xbest and a random individual from the top 5% elite individual as the initial point for L-BFGS

26: Perform L-BFGS local optimization to obtain xrefined

27: if f(xrefined)<f(xbest) then

28:   xbest ← xrefined

29:end for

30:Return the optimal solution xbest

Instructions:

  1. (1) State action reward closed loop: Each generation’s state construction, action sampling, and reward calculation strictly correspond to the previous MDP modeling;
  2. (2) Policy network embedding: The decision output (at) of PG-Tuner directly drives the parameters in mutation formulas 16 and 17;
  3. (3) Online fine-tuning: policy updates are performed every 10 generations to adapt to the current problem characteristics;
  4. (4) Resource allocation: L-BFGS is only activated in the last 10% of iterations to avoid premature consumption of computational budget.

4. Experimental setup and results

4.1. Benchmark function suite

In order to rigorously evaluate the generalization performance of the proposed RL-DE framework, the experiment used a completely separate training-test benchmark partition: the policy network training phase only used the Congress on Evolutionary Computation 2014 (CEC2014) suite (30 functions) as the training set; The performance evaluation phase is conducted on the CEC 2017 suite (29 functions) without overlapping instances, fundamentally eliminating the risk of information leakage. The CEC 2017 suite includes unimodal, multimodal, mixed, and composite feature functions with domains of [−100, 100] and dimensions set to 10, 30, 50, and 100 to test the algorithm’s scalability. To ensure reproducibility and fair comparison, all experiments in Section 4 employ consistent settings across all dimensions: population size NP = 100, maximum function evaluations MaxFES = 5000 × D (where D is the problem dimension), and 30 independent runs for each algorithm on each test function.

To further validate the robustness of RL-DE on more recent and challenging multimodal landscapes, and to address potential concerns regarding overfitting to the CEC2017 characteristics, we additionally evaluate the proposed framework on the IEEE CEC2022 benchmark suite. The CEC2022 suite comprises 12 composite functions with complex terrain features specifically designed to test algorithm scalability in 10-dimensional and 20-dimensional spaces. Following the standard experimental protocol, we set MaxFES = 200,000 for D = 10 and MaxFES = 1,000,000 for D = 20, conducting 30 independent runs per function. Comparative analyses against SaDE, SHADE, ILSHADE, JSO, MPEDE, and LSHADE demonstrate that RL-DE maintains consistent performance advantages on this newer benchmark, corroborating the high-dimensional robustness claims presented in Section 4.2. Detailed numerical results, including mean values, standard deviations, and convergence curves, are provided in S5 and S6 Tables of the Supporting Information.

4.2. Comparison and analysis of effects

To comprehensively evaluate the performance of the proposed RL-DE framework, this section compares and analyzes the comprehensive performance of RL-DE with six representative differential evolution variants (DE, SaDE, SHADE, LSHADE, ILSHADE, jSO, MPEDE) on the CEC2017 test suite from three dimensions: solution accuracy, convergence speed, and statistical significance. The experiment covers four dimensions: 10, 30, 50, and 100, and each configuration is independently run 30 times to ensure the statistical reliability of the results.

4.2.1. Accuracy comparison and analysis.

Tables 1-4 show the average optimal values obtained by each algorithm on 29 test functions of CEC2017 in 10, 30, 50, and 100 dimensions (average of 30 independent runs). To facilitate comparison, all optimal values are kept in the Scientific notation format, and the algorithm results showing the best performance on each function are marked in bold (the complete data including standard deviations are provided in S1-S4 Tables).

As shown in Table 1, for the 10-dimensional CEC2017 functions, all algorithms achieve identical optimal values on F2 and F5, which are therefore excluded from the comparative analysis. Among the remaining 27 functions, RL-DE ranks first on 11 functions, followed by LSHADE with 7 functions, while MPEDE, ILSHADE, and JSO each secure the top position on 5 functions. Notably, RL-DE and LSHADE demonstrate the strongest overall performance, ranking as the top two algorithms in the 10-dimensional test suite.

thumbnail
Table 1. Comparison of optimal values for 10 dimensional CEC2017 functions (30 runs).

https://doi.org/10.1371/journal.pone.0347860.t001

As shown in Table 2, among the 29 functions, all algorithms achieve identical optimal values on F2 and F5, which is excluded from the count. Among the remaining 27 functions, the RL-DE algorithm ranks first in 12 functions; the MPEDE algorithm ranks first in 8 functions; and the JSO algorithm ranks first in 6 functions. Notably, RL-DE, MPEDE, and JSO demonstrate the strongest overall performance, securing the top three positions in the 30-dimensional CEC2017 function test.

thumbnail
Table 2. Comparison of optimal values for 30 dimensional CEC2017 functions (30 runs).

https://doi.org/10.1371/journal.pone.0347860.t002

As shown in Table 3, for the 50-dimensional CEC2017 functions, all algorithms achieve identical optimal values on F2 and F5, which are therefore excluded from the count. Among the remaining 27 functions, the RL-DE algorithm ranks first in 13 functions; the SHADE algorithm ranks first in 4 functions; and the SaDE algorithm ranks first in 4 functions. The RL-DE, SHADE, and SaDE algorithms demonstrate the strongest overall performance, securing the top three positions in the 50-dimensional test suite.

thumbnail
Table 3. Comparison of optimal values for 50 dimensional CEC2017 functions (30 runs).

https://doi.org/10.1371/journal.pone.0347860.t003

As shown in Table 4, among the 29 functions, all algorithms achieve identical optimal values on F5, which are therefore excluded from the count. Among the remaining 28 functions, the RL-DE algorithm ranks first in 14 functions; the SaDE algorithm ranks first in 6 functions; and the SHADE algorithm ranks first in 5 functions. The RL-DE, SaDE, and SHADE algorithms demonstrate the strongest overall performance, securing the top three positions in the 100-dimensional CEC2017 function test.

thumbnail
Table 4. Comparison of optimal values for 100 dimensional CEC2017 function (30 runs).

https://doi.org/10.1371/journal.pone.0347860.t004

To quantitatively summarize the performance advantages of RL-DE, we conducted a comprehensive statistical analysis of the ranking results across all dimensions. Tables 1-4 show that RL-DE achieved the best performance on 11/27 (40.7%), 12/27 (44.4%), 13/27 (48.1%), and 14/28 (50.0%) of functions in 10D, 30D, 50D, and 100D, respectively (with F2 and F5 excluded from the 10D-50D comparisons as all algorithms achieved identical optimal values on these functions, and F5 excluded from the 100D comparison). Furthermore, supported by the Wilcoxon signed-rank test results presented in Section 4.2.3, RL-DE demonstrates statistically significant superiority (p < 0.05) over baseline algorithms on the majority of functions. Additionally, the function-level statistical significance analysis in Table 10 reveals that RL-DE achieves statistically significant superiority over traditional DE on 70.4%−81.5% of functions across the four dimensions, with the proportions of significant superiority over other adaptive DE variants (SaDE, SHADE, LSHADE, ILSHADE, jSO, MPEDE) ranging from 44.4% to 70.4%. These quantitative metrics, combined with the Friedman test results showing RL-DE consistently ranking first across all dimensions, validate the robustness of its performance advantage from both ranking-based and statistical significance perspectives.

4.2.2. Comparison and analysis of convergence speed.

To visually evaluate the convergence behavior of RL-DE and seven other comparison algorithms (DE, SaDE, SHADE, LSHADE, ILSHADE, jSO, MPEDE) during the search process, this section selects a representative function F23 from the CEC2017 test function set and plots its convergence curve in 10, 30, 50, and 100 dimensions. F23 is a multimodal composite function with complex terrain and multiple local extrema, which can effectively test the algorithm’s ability to explore and balance development in different dimensions.

The horizontal axis in the figure represents the number of function evaluations (recorded every 100 times), and the vertical axis represents the current optimal solution error value (taking the average of 30 independent runs, using logarithmic coordinates to highlight differences). Figs 47 respectively show the convergence trends of the F23 function in four dimensions.

thumbnail
Fig 4. 10 dimensional F23 function convergence curve (average).

https://doi.org/10.1371/journal.pone.0347860.g004

thumbnail
Fig 5. 30 dimensional F23 function convergence curve (average).

https://doi.org/10.1371/journal.pone.0347860.g005

thumbnail
Fig 6. 50 dimensional F23 function convergence curve (average).

https://doi.org/10.1371/journal.pone.0347860.g006

thumbnail
Fig 7. 100 dimensional F23 function convergence curve (average).

https://doi.org/10.1371/journal.pone.0347860.g007

From the figure, it can be observed that:

  1. (1) The convergence speed in the early stage varies significantly with dimensionality
  2. In the 10 dimensional stage, LSHADE, ISHADE, and RL-DE rank among the top three; At 30 dimensions, RL-DE ranks first, followed closely by LSHADE and ISHADE, indicating that the policy gradient controller can enter the efficient search phase ahead of schedule in the medium dimension; In 50 dimensions, SHADE achieved first place, while RL-DE and SaDE ranked second and third respectively; After entering the 100 dimensions, SHADE, SaDE, and MPEDE occupy the top three, while RL-DE ranks fourth.
  3. (2) Outstanding ability to continuously decline in the later stage
  4. Although RL-DE starts up slightly slower than some comparison algorithms in the high-dimensional early stage, its error curve can still maintain a stable decrease after 90% evaluation budget, verifying the improvement effect of LB-Refiner on the final solution accuracy.
  5. (3) High dimensional adaptability gradually emerges
  6. As the dimensionality increases, although the RL-DE ranking fluctuates in the early stage, its later convergence stability is better than most comparison algorithms, reflecting the robustness of the reinforcement learning parameter controller in complex terrain.

In summary, RL-DE exhibits a convergence feature of competitive in the early stage and stable decline in the later stage on F23. Its six parameter adaptive mechanism based on policy gradient and L-BFGS local reinforcement module work together to effectively balance global exploration and local development, and maintain continuous evolution ability in complex multimodal and high-dimensional scenarios.

4.2.3. Robustness comparison and analysis.

To evaluate the stability and consistency of RL-DE in multiple runs, this paper conducted non parametric Friedman test and paired Wilcoxon rank test (α = 0.05) on the average optimal values of 29 functions and 30 independent runs of each function in CEC2017. The experiment covers 10, 30, 50, and 100 dimensions, and the results are as follows:

  1. (1) Overall ranking: Friedman test

The results in Table 5 show that the Friedman test gives p < 0.01 for all four dimensions (specifically, p = 3.85 × 10-3 for 10D, p = 6.28 × 10-7 for 30D, p = 3.46 × 10-9 for 50D, and p = 6.56 × 10-14 for 100D), indicating statistically significant differences between the algorithms. Further observation of the average rank reveals that:

thumbnail
Table 5. Friedman average ranking and overall ranking (10/30/50/100 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t005

RL-DE always maintains first place:

Starting from the 10-dimensional scenario, it ranks first with an average rank of 3.17, and then further improves to 2.57 and 2.40 in the 30-dimensional and 50-dimensional tests, respectively, indicating that as the dimension increases, its relative advantage actually expands; Although it has slightly rebounded to 2.88 at 100 dimensions, it still firmly holds the top spot.

Other algorithms drift with dimensionality, while RL-DE maintains a stable position:

In 10 dimensions, LSHADE (3.62) and MPEDE (4.60) are ranked second and third, respectively; After 30 dimensions, MPEDE remained stable in the top three with average ranks of 3.50 (30D), 3.57 (50D), and 3.21 (100D), demonstrating good cross-dimensional stability; SaDE jumped to second place (2.98) in 100 dimensions, trailing RL-DE by 0.10, indicating that its adaptive mechanism has also begun to exert force in high-dimensional scenarios, but it has not yet surpassed RL-DE.

In summary, the Friedman test consistently confirms from an overall ranking perspective that the six parameter adaptive mechanism of policy gradient proposed by RL-DE can provide the most stable and significantly optimal solution quality in the 10–100 dimensional range, laying a solid robustness foundation for its subsequent application in real high-dimensional engineering problems.

  1. (2) Pairwise significance: Wilcoxon signed-rank test

To verify whether RL-DE is significantly superior to other comparison algorithms, this study employed the Wilcoxon signed-rank test for pairwise comparisons. This non-parametric test is specifically designed for paired samples, where each algorithm’s performance on the same test function constitutes a natural pair, effectively controlling for inter-function variability and isolating algorithmic performance differences. The test evaluates whether the median performance difference across the 29 benchmark functions (each assessed via 30 independent runs) significantly deviates from zero, with the null hypothesis stating no difference between algorithms and the alternative hypothesis stating RL-DE performs significantly better. Statistical significance is determined at the α = 0.05 level.

As shown in Table 6, in the 10 dimensional scenario, RL-DE achieved p < 0.05 compared to traditional DE, ILSHADE, MPEDE, SHADE, and SaDE, indicating significant differences; Compared with jSO, p = 0.135 > 0.05, The difference between the two has not yet reached a significant level, but the average rank of RL-DE is still lower than that of jSO, indicating that its overall solution quality is better. Entering the 30th dimension (Table 7), the significant advantages of RL-DE are further expanded: except for MPEDE which is close to it in some functions (p = 0.023), RL-DE is significantly better than other algorithms, especially with a p-value lower than DE of 2.16 × 10−4, indicating that as the dimension increases, the performance gain brought by RL-DE’s policy gradient six parameter adaptive mechanism becomes more significant.

thumbnail
Table 6. Pairwise Wilcoxon signed-rank p-value matrix (paired-sample test, < 0.05 indicates significant difference) (10 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t006

thumbnail
Table 7. Pairwise Wilcoxon signed-rank p-value matrix (paired-sample test, < 0.05 indicates significant difference) (30 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t007

The 50 dimensional results (Table 8) show that RL-DE outperforms the control algorithm in all aspects: p-values with DE, ILSHADE, JSO, LSHADE, SHADE, and SaDE are all less than 0.01, and p = 0.047 with MPEDE is also significantly critical. It is worth noting that in 100 dimensions (Table 9), although SaDE and SHADE closely follow RL-DE in Friedman mean rank, the Wilcoxon test still gave p = 0.22 and p = 0.11, which did not reach a significant level; On the other hand, DE, ILSHADE, LSHADE, and JSO continued to be significantly surpassed by RL-DE (p < 0.02). This phenomenon indicates that as the dimensionality continues to increase, the gap between RL-DE and the strongest baseline narrows, but still maintains statistical significance, reflecting the robust generalization ability of its reinforcement learning controller in high-dimensional space.

thumbnail
Table 8. Pairwise Wilcoxon signed-rank p-value matrix (paired-sample test, < 0.05 indicates significant difference) (50 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t008

thumbnail
Table 9. Pairwise Wilcoxon signed-rank p-value matrix (paired-sample test, < 0.05 indicates significant difference) (100 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t009

To further quantify the statistical advantages of RL-DE relative to baseline algorithms, we conducted function-level significance tests using Welch’s t-test on the mean and standard deviation of 30 independent runs for each function (α = 0.05). Unlike the Wilcoxon signed-rank test, the t-test performs statistical inference based on mean and standard deviation, which is suitable for evaluating significant differences in means between algorithms. Table 10 summarizes the function-level statistical significance comparison results between RL-DE and comparison algorithms across 10D-100D dimensions, including the number and percentage of functions categorized as ‘statistically significantly better’, ‘no significant difference’, and ‘statistically significantly worse’.

thumbnail
Table 10. Summary of Function-Level Statistical Significance Comparison Between RL-DE and Baseline Algorithms.

https://doi.org/10.1371/journal.pone.0347860.t010

Based on the function-level statistical significance analysis results in Table 10, the following conclusions can be drawn: (1) RL-DE’s advantage over DE is the most stable and significant: Across the 10D-100D range, RL-DE achieves statistically superior performance on 70.4%−81.5% of functions and never performs significantly worse than DE, validating the substantial improvement of the PG-Tuner parameter adaptation mechanism over traditional DE; (2) RL-DE’s advantage over LSHADE increases noticeably with dimensionality: rising from 44.4% (10D) to 70.4% (50D) and maintaining 67.9% (100D), demonstrating the adaptive advantage of the reinforcement learning controller in high-dimensional scenarios; (3) RL-DE never performs statistically significantly worse than ILSHADE and LSHADE in 50D and 100D tests (significantly worse ratio: 0%), showing its robust handling capability for complex multimodal problems; (4) Challenges and opportunities coexist in 100D: Although the advantage ratios over SaDE, SHADE, and MPEDE decrease (46.4%−50.0%), RL-DE maintains significant superiority ratios of 64.3%−78.6% over DE, ILSHADE, LSHADE, and JSO, indicating its competitiveness in ultra-high-dimensional problems. These function-level statistical significance data corroborate the overall results from Friedman and Wilcoxon tests, collectively validating the effectiveness and robustness of RL-DE’s parameter adaptation mechanism.

4.2.4. Computational cost analysis and performance-efficiency trade-off.

Although RL-DE demonstrates superior optimization performance, the integration of PG-Tuner inevitably introduces additional computational overhead compared to traditional adaptive DE variants. To quantify this trade-off, we conducted a runtime comparison experiment between RL-DE (including both training and inference phases), SHADE, and MPEDE across 10, 30, 50, and 100 dimensions on the CEC2017 benchmark. All experiments were executed on a standard computing platform equipped with an Intel Core i5-1135G7 processor (2.4 GHz) and 16 GB RAM, implemented in Python.

As shown in Table 11, the RL-DE framework involves two distinct stages: policy network training (RL-DE-Train) and deployment inference (RL-DE-Inference). The training phase, conducted once per dimension on the CEC2014 training set, requires substantial computational investment (ranging from 456 seconds for 10D to 12,721 seconds for 100D) due to the extensive policy gradient updates and episode sampling. However, this constitutes a one-time offline cost; once trained, the policy network is fixed and deployed for inference without requiring retraining for new problem instances within the same dimensionality.

thumbnail
Table 11. Runtime comparison (seconds) for a single run across different dimensions.

https://doi.org/10.1371/journal.pone.0347860.t011

During the inference phase, RL-DE exhibits moderate runtime increments compared to traditional methods. Specifically, RL-DE-Inference consumes approximately 20%, 60%, 39%, and 73% more time than SHADE for dimensions 10, 30, 50, and 100, respectively. The overhead primarily stems from (1) neural network forward propagation for generating six-dimensional parameters (F, CR, p, Fw, archF, archP) each generation, and (2) additional fitness evaluations required for state construction and reward calculation. Notably, the runtime gap narrows relative to MPEDE in lower dimensions (10D-30D), as MPEDE’s multi-population maintenance also incurs significant computational cost.

From a performance-efficiency trade-off perspective, the additional computational investment yields substantial returns in solution quality. As demonstrated in Tables 1-4 and the statistical significance analysis (Table 10), RL-DE achieves statistically significant superiority over baseline algorithms on 44.4%−81.5% of test functions across all dimensions. Specifically, in high-dimensional scenarios (100D), RL-DE secures the best results on 50% of functions (14/28), while SHADE and SaDE achieve only 17.9% and 21.4%, respectively.

We characterize this relationship using a Performance-Cost Ratio (PCR) defined as the ratio of function-level significant improvement percentage to relative runtime overhead. In 100D, RL-DE achieves a PCR of approximately 0.68 (50.0% improvement rate/ 73.4% overhead) against SHADE, indicating that each 1% of additional computational cost yields roughly 0.68% improvement in significant win rate. This validates that the neural network-based parameter control provides favorable cost-benefit characteristics for expensive black-box optimization scenarios where solution quality dominates computational budget concerns.

4.3. Ablation experiment

To quantitatively evaluate the contribution of each module, we compared the performance of five algorithm variants on a 30 dimensional CEC2017 test set. Table 12 shows the average optimal results on 8 representative functions (covering unimodal, multimodal, mixed, and composite terrain features). The meaning of each column is as follows: RF-DE-full is the complete algorithm; RL-DE-w/o-LB removes the L-BFGS local refinement module; RL-DE-w/o-Phase uses a fixed single mutation strategy (eliminating the exploration development switch); RL-DE-w/o-RL replaces PG-Tuner with traditional SHADE adaptive mechanism; RL-DE-w/o-A removes external archive disturbances; The best value is the theoretical optimal value as a reference benchmark. All experiments were conducted with a population size of NP = 100, a maximum evaluation frequency of MaxFES = 150000, and 30 independent runs.

thumbnail
Table 12. Comparison of ablation experiment results (30 dimensional CEC2017 partial function).

https://doi.org/10.1371/journal.pone.0347860.t012

Result analysis:

  1. (1) Contribution of LB-Refiner module
  2. On multi-modal composite functions such as F7, F9, and F11, removing the LB-Refiner module resulted in a performance decrease of 8.3% −26.9%, confirming its significant improvement in solution accuracy in the optimization stage. This module effectively compensates for the shortcomings of DE in the local refinement stage through the convergence characteristics of L-BFGS.
  3. (2) Two-stage strategy switching mechanism
  4. The fixed single policy resulted in a performance decrease of 40.0% −97.2% for F11, F14, and F17, indicating that the exploration development phased policy is crucial for complex terrains. The strong exploration ability of early DE/rand/2 can avoid premature convergence, while the targeted development of DE/current to pBest-w/1 in the later stage accelerates the mining of high-quality regions.
  5. (3) PG-Tuner Reinforcement Learning Controller
  6. After replacing with traditional SHADE adaptation, the performance of all functions decreased, with F11 showing the most significant decrease (130.5%). This indicates that fixed rules are difficult to capture nonlinear coupling between parameters, while policy gradient methods can learn more refined adaptability through neural network modeling.
  7. (4) Impact of external archive collection mechanism
  8. Removing historical archives leads to a loss of global diversity, with the most significant performance degradation observed in highly multimodal problems such as F9 and F11 (45.5% and 152.4% respectively), verifying the crucial role of archives in maintaining population diversity and escaping local optima.

The Friedman test showed significant differences among the variants (p = 3.21 × 10−6), with an average rank of 1.25 for RL-DE-full, which was significantly better than other variants (p < 0.01), indicating the existence of orthogonal complementary effects among the modules.

4.4. Hyperparameter analysis

To investigate the impact mechanism of RL-DE core hyperparameters on algorithm performance in depth, this section conducted a systematic sensitivity analysis on the 30 dimensional CEC 2017 test set. Focus on three key hyperparameters: L-BFGS activation probability LS_P, L-BFGS activation threshold, and exploration development strategy switching threshold. Set 4–5 typical value levels for each parameter, fix other parameters as default settings (LS_P = 0.2, L-BFGS startup threshold = 0.9, strategy switching threshold = 0.5), run independently 30 times, calculate the average optimal value and standard deviation, and determine the optimal configuration through non parametric testing.

4.4.1. The influence of L-BFGS activation probability LS_P.

LS_P controls the intervention frequency of the local refinement module, directly affecting the balance of global local search. The experimental setup includes four levels of LS_P∈{0.1, 0.2, 0.3, 0.5} to investigate the performance differences on different terrain functions. As shown in Table 13.

thumbnail
Table 13. Comparison of average optimal values under different LS_P settings (30 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t013

Result analysis: When LS_P is too low (0.1), the activation of L-BFGS is insufficient, especially in the later stages of functions such as F20 and F23, where convergence is slow and the error increases by 4.5% −7.2%; If LS_P is too high (0.5), it will lead to premature local search, resulting in premature depletion of population diversity on highly multimodal functions such as F9 and F11, and a performance decline of 9.8% −15.3%. LS-P = 0.2 achieves the optimal balance between development efficiency and global exploration, and Wilcoxon test shows that it is significantly better than LS_P = 0.1 (p = 0.012) and LS-P = 0.5 (p = 0.008), confirming the key role of moderate local reinforcement in hybrid algorithms.

4.4.2. The influence of L-BFGS startup threshold.

The activation threshold of L-BFGS determines the timing of local refinement intervention. Early activation may waste exploration ability, while late activation may miss the opportunity for refinement. Test startup threshold∈{0.7, 0.8, 0.9, 0.95} (i.e., final 30%, 20%, 10%, 5% iteration stages). As shown in Table 14.

thumbnail
Table 14. Performance comparison under different L-BFGS startup thresholds (30 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t014

As shown in Table 14, the sensitivity of L-BFGS activation threshold varies significantly with problem multimodality. For unimodal or simple multimodal functions (e.g., F3, F20), performance differences within the threshold range of 0.8–0.95 are statistically insignificant (p > 0.05), as the global optimum is easily located regardless of refinement timing. However, for highly multimodal complex functions (e.g., F11, F17), early activation (0.7) severely weakens late-stage exploration capability, causing performance degradation of 40.2% and 30.3%, respectively, due to premature consumption of computational resources in unexplored regions. Conversely, late activation (0.95) results in insufficient available budget for L-BFGS, leaving functions such as F20 and F23 not fully refined. The default threshold of 0.9 (i.e., the final 10% of iterations) achieves the optimal balance on most functions, with Friedman’s test showing an average rank of 1.72 significantly superior to other settings (p = 3.7 × 10−4). This configuration maintains robustness across both simple and complex landscapes, consistent with Tanabe & Fukunaga’s suggestion [19] and Wu & Zhang’s experimental conclusion [32], validating the effectiveness of this heuristic threshold in the exploration-exploitation trade-off.

4.4.3. The impact of strategy switching threshold.

The strategy switching threshold controls the transition node from strong exploration (DE/rand/2) to fine development (DE/current to pBest-w/1). Test switching threshold∈{0.3, 0.4, 0.5, 0.6, 0.7}. As shown in Table 15.

thumbnail
Table 15. Comparison of average optimal values under different strategy switching thresholds (30 dimensions).

https://doi.org/10.1371/journal.pone.0347860.t015

Result analysis: Premature threshold switching (0.3) severely weakened the global exploration ability, resulting in a performance decline of 18.8% and 56.1% (p < 0.001) on highly multimodal functions such as F9 and F11. Due to the algorithm entering the development oriented stage too early, high-quality regions were not fully discovered. Switching too late (0.7) slows down the fine development process and reduces convergence speed by 12.4% on relatively simple functions such as F3 and F4. The default threshold of 0.5 performs the best in the exploration development trade-off, with Friedman’s average rank of 1.96 significantly better than other settings (p = 8.6 × 10−4), confirming the orthogonal complementary effect of this fixed threshold and PG-Tuner dynamic parameter adjustment.

4.4.4. Comprehensive analysis and recommended settings.

Based on the sensitivity analysis above, the recommended configuration for RL-DE hyperparameters is LS_P = 0.2, L-BFGS activation threshold = 0.9, and strategy switching threshold = 0.5. The average error of this combination on the 30 dimensional CEC 2017 test set is 3.58 × 103, significantly better than all other parameter combinations (p < 0.001).

It is worth noting that there is a weak interaction effect between the hyperparameters: when the switching threshold is too early (such as 0.4), LS_P can be appropriately reduced to 0.15 to avoid premature localization; When the L-BFGS startup threshold is delayed to 0.95, LS_P needs to be increased to 0.25 to compensate for the insufficient budget for refinement. However, the default settings exhibit the strongest robustness across a wide range of problem types and are insensitive to hyperparameter perturbations (performance fluctuations<5%), validating the engineering practicality of the RL-DE framework. Subsequent research can explore meta learning based hyperparameter adaptive frameworks to achieve dynamic configuration related to problems and further enhance algorithm autonomy.

5. Application of RL-DE in flexible job shop scheduling problem

5.1. Background and challenges

The Flexible Job Shop Scheduling Problem (FJSP) is a core NP hard problem in intelligent manufacturing systems, which directly affects production efficiency and equipment utilization [33]. The complexity of resource allocation and task sequencing in such problems also appears in emerging applications such as communication satellite networks, where evolutionary algorithms with memory guidance have been recently employed to optimize data transmission scheduling under dynamic constraints [34]. In FJSP, the feature of allowing processes to be processed on multiple optional machine tools greatly increases the complexity of the problem and the size of the solution space [35]. Although traditional metaheuristic algorithms can achieve good results, they generally rely on manually designed heuristic rules and static parameter configuration, making it difficult to adapt to real-time scheduling requirements in dynamic production environments. The migration of the RL-DE framework for continuous optimization to FJSP faces three challenges: firstly, there are essential differences between FJSP’s discrete decision space (process machine allocation and sorting) and DE’s real vector encoding, requiring the design of an effective mapping mechanism; Secondly, the state representation of scheduling problems needs to capture multimodal information such as machine load, process priority, and topological constraints, rather than a single fitness statistic; Finally, the reward function needs to balance the conflicting objectives of makespan and workload balance, and explicitly guide the search to converge towards the Pareto front. This chapter extends the successful experience of RL-DE to the field of discrete combinatorial optimization through a three-dimensional transformation of encoding state reward, and verifies its generalization ability on the Brandimarte benchmark dataset.

5.2. FJSP optimization problem modeling

This section first establishes the standard mathematical model of FJSP, laying the foundation for embedding the RL-DE framework. FJSP can be formalized as a three-stage decision-making process: process allocation, machine tool selection, and time sorting. Let the homework set J={1,..., n}, each homework j contains Oj processes, and the machine tool set M={1,..., m}. Define binary decision variables:

  1. (1) αk, i∈{0,1}: whether process i is assigned to machine tool k
  2. (2) βi, i’∈{0,1}: Does process i and i’ have priority on shared machine tools
  3. (3) ti ∈ R+: Start time of process i
  4. (4) Cmax: Maximum completion time (makespan)
  5. (5) Lmax: Maximum machine load (workload balance)

The objective function and constraints are shown in formula 31:

(31)

Among them, Ri is the set of optional machine tools for process i, pk, i is the processing time, pr(i) is the preceding process, and H is a sufficiently large constant.

Makespan (cmax) is the primary optimization objective, reflecting the total time required for the production system to complete all tasks. This indicator is directly related to the delivery cycle of enterprise orders and the utilization rate of production capacity, and is the core indicator for measuring the efficiency of scheduling plans. Workload Balance (Lmax) is used as a secondary objective to quantify the workload imbalance among different machine tools. Its optimization helps to balance equipment utilization, avoid local overload, and has important value in improving the overall stability of manufacturing systems.

To adapt the above mixed integer programming model to the continuous optimization framework of RL-DE, this section further proposes an encoding policy based on priority rules. The priority rule is embedded into a continuous vector using the Random Key representation method, as shown in Fig 8. The encoding vector x∈[0, 1]D is divided into two segments with total dimension D = n + O, where n is the number of jobs and O is the total number of operations:

  1. (1) Job Priority Segment (xjob): The first n dimensions [x1, x2, …, xn] represent the scheduling priorities of n jobs. During decoding, jobs are sorted in ascending order of xj (smaller values indicate higher priority), and all operations of a job inherit this priority. This ensures that when multiple operations are eligible for scheduling, the one belonging to the job with the smallest xj is selected first.
  2. (2) Machine Selection Segment (xop): The remaining O dimensions [xn+1, …, xD] correspond to the machine selection preferences for each operation. For operation Oi (the i-th operation in the global operation list), its machine assignment is determined by its encoding value xn+i and the eligible machine set Ri. Specifically, for each machine k ∈ Ri, we calculate a composite weight:
(32)

where loadk is the current workload of machine k. The machine with the minimum wk is selected. This formula ensures that: (i) when xn+i is small, the term 1/(1 + loadk) dominates, favoring machines with lighter loads; (ii) when xn+i is large, the random key value dominates, enabling exploration of different machine assignments. Through the evolutionary optimization of xn+i, the algorithm automatically learns the trade-off between processing efficiency and load balancing.

Operation Sequence Generation: Based on the Serial Scheduling Generation Scheme (SSGS) [36], the decoding process maintains a schedulable operation set Ω (initialized with the first operation of each job). At each iteration:

  1. (i) Select operation i* = argmini∈Ω{xjob[i]}, where xjob[i] denotes the priority value of the job to which operation i belongs;
  2. (ii) Assign machine k* to operation i* using formula 32;
  3. (iii) Insert i* into the earliest available time slot of machine k* while satisfying precedence constraints;
  4. (iv) Remove i* from Ω and add its successor operation (if any) to Ω.

Illustrative Example: Taking the MK01 instance (10 jobs, 55 operations, 6 machines) as an example, the encoding vector has dimension D = 10 + 55 = 65. The first 10 dimensions xjob=[0.12, 0.85, 0.34, 0.67, 0.23, 0.91, 0.45, 0.78, 0.56, 0.32] determine the job scheduling sequence: Job 1 (x1 = 0.12) is scheduled first, followed by Job 5 (x5 = 0.23), Job 10 (x10 = 0.32), etc. For the first operation of Job 1 (global index i = 1, encoded by x11 = 0.40), assuming its eligible machines are R1={M1(3 min), M3(5 min)} with current loads loadM1 = 10 and loadM3 = 8, the weights are calculated as: wM1 = 0.40 + 1/11 ≈ 0.491 and wM3 = 0.40 + 1/9 ≈ 0.511. Therefore, M1 is selected due to the smaller weight.

The complete decoding procedure is formally described in Algorithm 4.

Algorithm 4 FJSP decoder based on priority rules

Input:

 • Random key vector x

 • Processing time matrix p_k, i (processing time of process i on machine tool k)

Output:

 • Scheduling Plan Sch

 • Maximum completion time c_max

 • Maximum machine load L_max

1. # Step 1: Extract homework priority and sort it

2. π = argsort(x[1: n]) # π is the arrangement of job indexes, and the smaller x_j, the higher the priority of job j

3. # Step 2: Initialize data structure

4. Ω={O_{j, 1} | For all jobs j ∈ J} # schedulable process set (the first process of each job)

5. load_k = 0 for all machine tools k ∈ M # machine tool current cumulative load

6. t_i = 0 for all processes i ∈ O # process start time

7. c_max = 0 # Initialize makespan

8. L_max = 0 # Initialize workload balance

9. # Step 3: Main Loop - Serial Scheduling Generation Scheme (SSGS)

10.while Ω ≠ ∅ do:

11.  # Step 4: Select the highest priority schedulable process

12.  i* = argmin_{i ∈ Ω}x_job[i] # x_job[i] represents the priority weight of the job to which process i belongs

13.  # Step 5: Computer bed selection preference weights

14.  for k in R_i*: # Optional Machine Tool Set for Traversing Process i*

15.    w_k = x_op[i*]+1/(1 + load_k) # Preference = encoding weight+load penalty term

16.  k* = argx_{k ∈ R_i*}w_k # Choose the machine tool with the highest comprehensive weight

17. # Step 6: Schedule the earliest feasible time window to machine tool k*

18. t_i* = max(

19.    t_{pr(i*)} + Σ_{k ∈ R_{pr(i*)}} p_{k,pr(i*)}·α_{k,pr(i*)}, # Completion time of preceding process

20.      max_{i’ scheduled on k*} {t_i’ + Σ_{k ∈ R_i’} p_{k,i’}·α_{k,i’}·δ_{k*,k}} # Machine tool k* idle time

21.)

22. # Step 7: Update System Status

23.   Ω = Ω \{i *} # Remove the scheduled process

24. If i* has a subsequent process succ(i*) then:

25.   Ω = Ω∪{succ(i*)} # Add subsequent processes to the schedulable set

26. end if

27. load_k* = load_k* + p_ {k*, i*} # Update the load of machine tool k*

28. c_max = max(c_max, t_i* + p_{k*, i*}) # Update global completion time

29. L_max = max(L_max, load_k*) # Update maximum machine load

30.end while

31.# Step 10: Return to the scheduling plan

32.Sch={(i, k_i, t_i) | For all processes i} # Record the triple of process machine time

33.return Sch, c_max, L_max

This encoding policy transforms discrete scheduling decisions into optimization problems in a continuous space, allowing the mutation crossover operation of DE to directly affect priority weights while ensuring the feasibility of the decoded solution.

5.3. RL-DE application policy

To successfully migrate the RL-DE framework to the FJSP discrete domain, this section systematically adapts and transforms it from four dimensions: state representation, reward design, evolutionary operations, and local refinement.

This chapter uses the 8-dimensional state vector st defined in Section 3.3. Firstly, the dual objectives of FJSP (makespan and workload balance) are linearly scalarized as individual fitness.

(33)

Subsequently, statistical measures are extracted from the standardized fitness distribution after each generation of evolution, and the state st is constructed according to formula 24. Then, the reward signal rt is generated using formula 27 in section 3.3.

This minimalist reward avoids artificial preferences by monitoring the relative improvement of the population front-end. In the FJSP experiment, PG-Tuner’s adaptive adjustment of archive probability archP and elite individual ratio p is expected to achieve an implicit trade-off of dual objectives: when the workload improves stagnation, moderately increasing archP to introduce archive diversity can help break local optima; When the makespan decreases too rapidly, reducing p and weakening elite individual guidance can help alleviate load imbalance. This unbiased reward+parameter adaptation mechanism is an important factor for RL-DE to be effective in the continuous domain, and it has also demonstrated some multi-objective coordination ability in discrete scheduling scenarios.

Then, keeping the six parameter control architecture and two-stage switching logic of RL-DE unchanged (using DE/rand/2 + Archive strategy in the early stage and DE/current to pBest-w/1 + Archive strategy in the later stage) (see formulas 16 and 17 in section 3.1), the continuous mutation operation is transformed into priority perturbation on random key vectors.

All differential operations are performed in a random key space, and the generated test vectors are mapped to a valid scheduling scheme using Algorithm 4 decoder to evaluate fitness. The crossover operation adopts a priority weight composite binomial crossover, and the crossover probability CR directly controls the inheritance ratio between the job priority and the machine tool weight segments, achieving a parameter structure co evolution consistent with the continuous domain.

After completing the breadth exploration of the global evolution operation, the local refinement module further develops high-quality solutions in depth. The LB-Refiner module is transformed into neighborhood exchange refinement: for each generation’s optimal scheduling scheme, the bottleneck machine set Mbottleneck on the critical path is identified, and 2-opt local exchange is performed within its machining process block. After each exchange, call the decoder to quickly evaluate, and execute L = 30 exchanges at most or Nno-improve = 5 consecutive times before terminating. This policy avoids the gradient calculation dilemma of L-BFGS in discrete space, ensuring that the global search has locked in high-quality regions before low-cost refinement.

5.4. Experimental results and analysis

5.4.1. Experimental setup.

The benchmark test uses the Brandimarte dataset MK01-MK15, covering instances of different scales from 10 × 6–20 × 15 [33]. Comparative algorithms include:

  1. (1) Accurate methods: MILP (Gurobi 30 min), CP (OR Tools 5 min, DoCplex 5 min)
  2. (2) Heuristic rule: Shortest Processing Time (SPT)
  3. (3) Meta heuristic: jSO, SHADE (adapted to FJSP version)

RL-DE parameters: population NP = 60, evolutionary generation T = 30, L-BFGS probability LS-P = 0.2, activation threshold 0.9. The policy network training was completed on the Dauzère-Pérès and Barnes datasets [35,37], with no overlap with Brandimarte. Run each instance independently 10 times (Note that 10 independent runs are adopted for Brandimarte experiments rather than 30 runs used in CEC2017, because each FJSP evaluation involves computationally expensive encoding-decoding and discrete-event simulation. This configuration aligns with common practice in FJSP literature where 10–20 runs are typically reported for large-scale scheduling benchmarks [38,39].) and record the optimal mean of makespan and workload balance.

5.4.2. Performance comparison analysis.

Key findings:

  1. (1) Makespan Competitiveness (Table 16): RL-DE performs well overall on small and medium-sized instances (MK01-MK08), with an average GAP of 4.8%. Although slightly inferior to the CP method (1.43%), it is significantly better than traditional heuristic algorithms. On the MK10-MK15 large-scale instances, the reinforcement learning policy of RL-DE demonstrated a certain degree of generalization ability, with an average result about 2.3 percentage points lower than jSO, but still maintained stable solution quality.
  2. (2) Advantages of Workload Balance (Table 17): In terms of load balancing objectives, RL-DE has a GAP of 5.2%, which is overall better than jSO (7.8%) and SHADE (9.2%). Especially on instances such as MK06 and MK13, there is a significant improvement, indicating that statistical data in the state space can effectively capture load distribution characteristics and guide the algorithm to search towards more balanced solution space regions.
  3. (3) Calculation efficiency: RL-DE takes an average of 148.3 seconds per run (Intel i7-12700H), which is slower than SPT (0.002 seconds) and OR Tools (45 seconds), but 11 times faster than MILP (1661.5 seconds), meeting the real-time requirements of industrial applications. Equivalent to jSO (132 seconds), the additional overhead brought by reinforcement learning components is within an acceptable range.
thumbnail
Table 16. Makespan optimization comparison (Brandimarte dataset) (unit: minutes).

https://doi.org/10.1371/journal.pone.0347860.t016

thumbnail
Table 17. Workload balance optimization comparison (unit: minutes).

https://doi.org/10.1371/journal.pone.0347860.t017

5.4.3. Ablation and sensitivity analysis.

As shown in Table 18, the Friedman test showed significant differences among the variants (p = 2.8 × 10-3). The experiment shows that the external archive mechanism contributes the most (with a performance decrease of 10.3% after removal), verifying the key role of historical elimination solutions in maintaining diversity in discrete scheduling problems and effectively avoiding premature convergence to local optimal scheduling modes; The two-stage policy switch came second (down 7.4%), indicating that the dynamic balance between exploration and development is particularly important for FJSP. Early and thorough exploration can help discover high-quality machine tool allocation patterns; The PG-Tuner controller brings a 5.9% improvement, which is lower than the continuous optimization scenario (about 15–20%), but still confirms the effectiveness of policy gradient learning in discrete space. Its adaptive p-value and archP can effectively handle instance features of different scales; The local refinement module brings a 4.4% improvement, which has practical significance for fine adjustment of makespan, especially in critical path optimization.

thumbnail
Table 18. Ablation experiment (MK06 example, cmax target).

https://doi.org/10.1371/journal.pone.0347860.t018

6. Conclusion

RL-DE proposes a three-layer closed-loop framework consisting of policy gradient six dimensional continuous parameter control+two-stage mutation skeleton+final L-BFGS local refinement: SG Turner models one iteration of differential evolution as an 8-dimensional state → 6-dimensional continuous action MDP, and learns the optimal distribution of F, CR, p, Fw, archF, and archP online; In the early stage, DE/rand/2 utilizes external archive to maintain global diversity, and in the later stage, it switches to DE/current to pBest-w/1 and introduces pBest weight scaling; The last 10% of iterations initiate LB-Refiner with a probability of 0.2, and perform 30 steps of L-BFGS rapid local refinement on the optimal or elite individuals. Compared with DE, SaDE, SHADE, LSHADE, ILSHADE, jSO, and MPEDE on 29 functions and 10–100 dimensions in CEC2017, this design achieved 11, 13, 16, and 15 first places respectively, with Friedman’s average rank consistently ranking first; In the MK01-MK15 flexible job shop scheduling instance, RL-DE, as a meta heuristic algorithm, outperforms comparative algorithms such as jSO and SHADE in both makespan and workload balance objectives, but its solution accuracy is lower than that of CP solvers (OR Tools/DoCplex), verifying the relative effectiveness of the framework in discrete combinatorial optimization scenarios.

Although RL-DE has achieved significant statistical advantages, there are still three obvious shortcomings: (1) the current 8-dimensional state representation relies solely on logarithmic fitness statistics and progress indicators, which may not adequately capture the population distribution morphology in the decision space. Preliminary theoretical analysis suggests that critical indicators for local optima or flat terrain (e.g., population diameter, pairwise distance entropy) are potentially under-represented, warranting future empirical validation on multimodal functions; (2) The reward function focuses primarily on fitness improvements among the top-tier individuals within a single generation, without explicitly monitoring population diversity metrics. This design potentially risks premature convergence on complex multimodal terrains, as the correlation between reward signals and diversity maintenance remains insufficiently quantified. The quantitative relationship between reward shaping and diversity preservation remains to be characterized through stagnation analysis; (3) The triggering probability (LS_P) and activation threshold of L-BFGS are currently fixed at 0.2 and 0.9T, respectively, and may not be optimally adaptive to dynamic convergence states. Heuristic analysis suggests such fixed parameters could lead to suboptimal resource allocation, meriting adaptive extensions based on real-time population statistics. In addition, the action space is limited to six continuous parameters, excluding discrete decisions such as mutation strategy selection. While ensuring learning stability, this simplification potentially constrains rapid adaptation to heterogeneous terrains, highlighting the need for hybrid continuous-discrete architectures.

To address the aforementioned limitations, we propose the following concrete and empirically verifiable research roadmap:

  1. (1) State Space Augmentation: We will design a 20-dimensional state vector (extending the current 8D representation by adding population diameter, pairwise distance entropy, fitness landscape gradient estimate, stagnation counter, and 8 additional topology-aware features). The enhanced RL-DE will be validated on the CEC2022 benchmark suite (12 functions at 10D and 20D) to verify whether the added diversity-aware features improve convergence on multimodal terrains (target: 10% reduction in average error on hybrid composite functions).
  2. (2) Multi-Agent RL-DE with Dynamic FJSP Testing: We will implement a 3-subpopulation multi-agent architecture where each subpopulation operates an independent PG-Tuner with asynchronous updates via a shared experience replay buffer. This architecture will be tested on FJSP instances with dynamic machine failures (simulated using Brandimarte MK10-MK15 instances with 10% random machine failure probability per generation) to validate robustness in non-stationary scheduling scenarios (target: < 5% makespan degradation under dynamic conditions).
  3. (3) Extended Action Space with Strategy Selection: The action space will be expanded from 6D to a 9-dimensional continuous-discrete hybrid space, adding learnable decisions for mutation strategy selection (3 optional strategies), L-BFGS trigger timing, and external archive update intensity. We will validate this design on CEC2017 100-dimensional constrained optimization problems to demonstrate integrated global-local scheduling capability.
  4. (4) Meta-Learning for Hyperparameter Self-Tuning: We will employ Model-Agnostic Meta-Learning (MAML) to enable the PG-Tuner to adapt its initialization to new problem classes within 5 generations. Testing will cover 100-dimensional expensive optimization scenarios to achieve plug-and-play capability without manual calibration.
  5. (5) TCN-Based Non-Stationary Optimization: To address temporal heterogeneity in wind power forecasting and renewable energy scheduling, we will integrate MNN-based Rolling Decomposition with Temporal Convolutional Networks (TCN) (inspired by Cai et al. [40]). This approach will be validated on real-world wind speed datasets (e.g., hourly meteorological data from NCEI/NOAA stations) to demonstrate superior adaptability compared to static RL-DE in dynamic environments (target: 10% improvement in rolling horizon prediction efficiency).

Supporting information

S1 Table. Comparison of optimal values and standard deviations for the 10-dimensional CEC2017 function (30 runs).

The complete numerical results include the average optimal values and standard deviations obtained from 30 runs of 8 algorithms (DE, SaDE, SHADE, ILSHADE, jSO, MPEDE, LSHADE, RL-DE) on 29 test functions across 10 dimensions.

https://doi.org/10.1371/journal.pone.0347860.s001

(XLSX)

S2 Table. Comparison of optimal values and standard deviations for the 30-dimensional CEC2017 function (30 runs).

The complete numerical results include the average optimal values and standard deviations obtained from 30 runs of 8 algorithms (DE, SaDE, SHADE, ILSHADE, jSO, MPEDE, LSHADE, RL-DE) on 29 test functions across 30 dimensions.

https://doi.org/10.1371/journal.pone.0347860.s002

(XLSX)

S3 Table. Comparison of optimal values and standard deviations for the 50-dimensional CEC2017 function (30 runs).

The complete numerical results include the average optimal values and standard deviations obtained from 30 runs of 8 algorithms (DE, SaDE, SHADE, ILSHADE, jSO, MPEDE, LSHADE, RL-DE) on 29 test functions across 50 dimensions.

https://doi.org/10.1371/journal.pone.0347860.s003

(XLSX)

S4 Table. Comparison of optimal values and standard deviations for the 100-dimensional CEC2017 function (30 runs).

The complete numerical results include the average optimal values and standard deviations obtained from 30 runs of 8 algorithms (DE, SaDE, SHADE, ILSHADE, jSO, MPEDE, LSHADE, RL-DE) on 29 test functions across 100 dimensions.

https://doi.org/10.1371/journal.pone.0347860.s004

(XLSX)

S5 Table. Comparison of optimal values and standard deviations for the 10-dimensional CEC2022 function (30 runs).

The complete numerical results include the average optimal values and standard deviations obtained from 30 runs of 7 algorithms (SaDE, SHADE, ILSHADE, jSO, MPEDE, LSHADE, RL-DE) on 12 test functions across 10 dimensions.

https://doi.org/10.1371/journal.pone.0347860.s005

(XLSX)

S6 Table. Comparison of optimal values and standard deviations for the 20-dimensional CEC2022 function (30 runs).

The complete numerical results include the average optimal values and standard deviations obtained from 30 runs of 7 algorithms (SaDE, SHADE, ILSHADE, jSO, MPEDE, LSHADE, RL-DE) on 12 test functions across 20 dimensions.

https://doi.org/10.1371/journal.pone.0347860.s006

(XLSX)

S7 Table. Computational time comparison of RL-DE and comparison algorithms across different dimensions.

The table shows the running time (in seconds) for a single run of RL-DE and comparison algorithms on the CEC2022 benchmark functions in 10D and 20D.

https://doi.org/10.1371/journal.pone.0347860.s007

(XLSX)

S8 Table. Statistical significance test results on the CEC2022 benchmark suite (Friedman test and Wilcoxon signed-rank test).

This table presents the non-parametric statistical test results for RL-DE and six comparison algorithms (SaDE, SHADE, ILSHADE, JSO, MPEDE, LSHADE) on the 12 composite functions of CEC2022 at D = 10 and D = 20. The Friedman test reports the average ranks of each algorithm across all functions and the overall p-value (significant at the α = 0.05 level). The Wilcoxon signed-rank test matrices provide pairwise p-values between algorithms, where values less than 0.05 indicate statistically significant performance differences.

https://doi.org/10.1371/journal.pone.0347860.s008

(XLSX)

References

  1. 1. Sun J, Liu X, Back T, Xu Z. Learning Adaptive Differential Evolution Algorithm From Optimization Experiences by Policy Gradient. IEEE Trans Evol Computat. 2021;25(4):666–80.
  2. 2. Sopov A, Sherstnev P. An investigation of the hybridization of DE and BFGS algorithms. Eur Proc Comput Technol. 2022.
  3. 3. Liu J, Lampinen J. A Fuzzy Adaptive Differential Evolution Algorithm. Soft Comput. 2004;9(6):448–62.
  4. 4. Sutton RS, Barto AG. Reinforcement learning: an introduction. Cambridge (MA): MIT Press. 1998.
  5. 5. Yang X, Wang R, Li K, Wang L. Reinforcement learning based automated design of differential evolution algorithm for black-box optimization preprint. arxiv. 2025.
  6. 6. Guo H, Ma S, Huang Z, Hu Y, Ma Z, Zhang X, et al. Reinforcement Learning-based Self-adaptive Differential Evolution through Automated Landscape Feature Learning. In: Proceedings of the Genetic and Evolutionary Computation Conference, 2025. 1117–26.
  7. 7. Huynh TN, Do DTT, Lee J. Q-Learning-based parameter control in differential evolution for structural optimization. Applied Soft Computing. 2021;107:107464.
  8. 8. Watkins CJ, Dayan P. Q-learning. Mach Learn. 1992;8(3–4):279–92.
  9. 9. Williams RJ. Simple statistical gradient-following algorithms for connectionist reinforcement learning. Mach Learn. 1992;8(3–4):229–56.
  10. 10. Ibrahim S, Mostafa M, Jnadi A, Salloum H, Osinenko P. Comprehensive Overview of Reward Engineering and Shaping in Advancing Reinforcement Learning Applications. IEEE Access. 2024;12:175473–500.
  11. 11. Tan Z, Tang Y, Li K, Huang H, Luo S. Differential evolution with hybrid parameters and mutation strategies based on reinforcement learning. Swarm and Evolutionary Computation. 2022;75:101194.
  12. 12. Ding W, Qian M, Lu C, Yi J, Pu H, Luo J. Differential Evolution with Joint Adaptation of Mutation Strategies and Control Parameters via Distributed Proximal Policy Optimization. Tsinghua Science and Technology. 2025;31(1):101–24.
  13. 13. Yang X, Wang R, Li K, Li W, Wang L. Deep reinforcement learning-assisted component auto-configuration of differential evolution algorithm for constrained optimization: a foundation model preprint. arxiv. 2025.
  14. 14. Long W, Hou T, Wei X, Yan S, Zhai P, Zhang L. A Survey on Population-Based Deep Reinforcement Learning. Mathematics. 2023;11(10):2234.
  15. 15. Xiaodong Li, Xin Yao. Cooperatively Coevolving Particle Swarms for Large Scale Optimization. IEEE Trans Evol Computat. 2012;16(2):210–24.
  16. 16. Song Y, Wu Y, Guo Y, Yan R, Suganthan PN, Zhang Y, et al. Reinforcement learning-assisted evolutionary algorithm: A survey and research opportunities. Swarm and Evolutionary Computation. 2024;86:101517.
  17. 17. Zhang J, Sanderson AC. JADE: Adaptive Differential Evolution With Optional External Archive. IEEE Trans Evol Computat. 2009;13(5):945–58.
  18. 18. Qin AK, Suganthan PN. Self-adaptive Differential Evolution Algorithm for Numerical Optimization. In: 2005 IEEE Congress on Evolutionary Computation. 1785–91.
  19. 19. Tanabe R, Fukunaga A. Success-history based parameter adaptation for Differential Evolution. In: 2013 IEEE Congress on Evolutionary Computation, 2013. 71–8.
  20. 20. Brest J, Maucec MS, Boskovic B. iL-SHADE: Improved L-SHADE algorithm for single objective real-parameter optimization. In: 2016 IEEE Congress on Evolutionary Computation (CEC), 2016. 1188–95.
  21. 21. Brest J, Maucec MS, Boskovic B. Single objective real-parameter optimization: Algorithm jSO. In: 2017 IEEE Congress on Evolutionary Computation (CEC), 2017. 1311–8.
  22. 22. Li X, Wang L, Jiang Q, Li N. Differential evolution algorithm with multi-population cooperation and multi-strategy integration. Neurocomputing. 2021;421:285–302.
  23. 23. Brest J, Greiner S, Boskovic B, Mernik M, Zumer V. Self-Adapting Control Parameters in Differential Evolution: A Comparative Study on Numerical Benchmark Problems. IEEE Trans Evol Computat. 2006;10(6):646–57.
  24. 24. Boggs PT, Byrd RH. Adaptive, Limited-Memory BFGS Algorithms for Unconstrained Optimization. SIAM J Optim. 2019;29(2):1282–99.
  25. 25. Storn R, Price K. Differential evolution-a simple and efficient heuristic for global optimization over continuous spaces. J Glob Optim. 1997;11(4):341–59.
  26. 26. Li XQ, Huang H, Hao ZF. Evolutionary algorithm for single-objective optimization based on neighborhood difference and covariance information. J Softw. 2018;29(9):2606–15.
  27. 27. Schulman J, Levine S, Abbeel P, Jordan M, Moritz P. Trust region policy optimization. In: Proceedings of the 32nd International Conference on Machine Learning, Lille, France, 2015. 1889–97.
  28. 28. Guo H, Ma Y, Ma Z, Chen J, Zhang X, Cao Z, et al. Deep Reinforcement Learning for Dynamic Algorithm Selection: A Proof-of-Principle Study on Differential Evolution. IEEE Trans Syst Man Cybern, Syst. 2024;54(7):4247–59.
  29. 29. Saputro DRS, Widyaningsih P. Limited memory Broyden-Fletcher-Goldfarb-Shanno (L-BFGS) method for the parameter estimation on geographically weighted ordinal logistic regression model (GWOLR). In: AIP Conf Proc, 2017. 040009.
  30. 30. Mao H, Venkatakrishnan SB, Schwarzkopf M, Alizadeh M. Variance reduction for reinforcement learning in input-driven environments. arXiv. 2018.
  31. 31. Wu C, Rajeswaran A, Duan Y, Kumar V, Bayen AM, Kakade S. Variance reduction for policy gradient with action-dependent factorized baselines. arXiv. 2018.
  32. 32. Wu X, Zhang K. A limited-memory BFGS-based differential evolution algorithm for optimal control of nonlinear systems with mixed control variables and probability constraints. Numer Algor. 2022;93(2):493–542.
  33. 33. Brandimarte P. Routing and scheduling in a flexible job shop by tabu search. Ann Oper Res. 1993;41(3):157–83.
  34. 34. Ren Y, Li Q, Zhang Y, Li J, Pedrycz W, Nagaratnam Suganthan P, et al. An Evolutionary Algorithm With Memory Guidance for Data Transmission Scheduling Optimization in Communication Satellite Network. IEEE Trans Syst Man Cybern, Syst. 2026;56(2):1354–68.
  35. 35. Dauzère-Pérès S, Ding J, Shen L, Tamssaouet K. The flexible job shop scheduling problem: a review. Eur J Oper Res. 2024;314(2):409–32.
  36. 36. Hartmann S, Kolisch R. Experimental evaluation of state-of-the-art heuristics for the resource-constrained project scheduling problem. Eur J Oper Res. 2000;127(2):394–407.
  37. 37. Barnes JW, Chambers JB. Solving the job shop scheduling problem with tabu search. IIE Transactions. 1995;27(2):257–63.
  38. 38. Zheng C, Xie Z. Research on the Flexible Job Shop Scheduling Problem with Job Priorities Considering Transportation Time and Setup Time. Axioms. 2025;14(12):914.
  39. 39. Luan F, Li R, Liu SQ, Tang B, Li S, Masoud M. An Improved Sparrow Search Algorithm for Solving the Energy-Saving Flexible Job Shop Scheduling Problem. Machines. 2022;10(10):847.
  40. 40. Cai X, Li D, Zou Y, Liu Z, Heidari AA, Chen H. A hybrid wind speed forecasting model with rolling mapping decomposition and temporal convolutional networks. Energy. 2025;324:135673.