CIM-PPO: Correntropy-Induced Metric PPO
- CIM-PPO is a PPO variant that replaces the KL divergence with a symmetric, bounded Liu–Correntropy induced metric, ensuring robust trust-region updates.
- It leverages a kernel-based strategy with linear computational complexity, avoiding costly matrix operations and enhancing scalability.
- Empirical results demonstrate CIM-PPO’s superior stability and faster convergence over traditional PPO methods on various continuous control benchmarks.
Correntropy-Induced Metric Proximal Policy Optimization (CIM-PPO) is a Proximal Policy Optimization (PPO) variant that replaces the standard Kullback-Leibler (KL) divergence penalty with a Liu–Correntropy Induced Metric (CIM), yielding a symmetric, bounded, and computationally efficient trust-region penalty. This approach rectifies the pathological behavior introduced by the asymmetric, unbounded nature of KL divergence in PPO-KL variants, leading to improved update efficiency and policy stability across continuous-control tasks (Guo et al., 2021).
1. Liu–Correntropy Induced Metric: Definition and Properties
The Liu–Correntropy Induced Metric is constructed from the correntropy between two vectors using a (normalized) Gaussian kernel . The correntropy is , and the induced metric is:
This metric is symmetric, positive-definite, bounded ( when ), and satisfies the triangle inequality. The feature map into a Hilbert space enables an inner-product view:
such that
Verifying that 0 is a bona-fide metric establishes a geometric foundation for trust-region analysis (Guo et al., 2021).
2. KL Divergence as a Special Case and Asymmetry Issue
Local approximations relate the KL divergence penalty to the Fisher information:
1
However, for general distributions, especially multivariate Gaussians with mismatched covariances, the KL divergence exhibits pronounced asymmetry, given by:
2
The paper establishes that this asymmetry can be arbitrarily large in high-dimensional settings or with differing variances. Theorems 1 and 2 demonstrate that such asymmetry can misguide the surrogate gradient, causing updates in a direction that actually decreases the advantage, reducing policy improvement efficiency (Guo et al., 2021).
3. Extension to Reproducing Kernel Hilbert Space and Trust-Region Guarantee
CIM is extended to the case of distributions in Reproducing Kernel Hilbert Space (RKHS). For two policies 3 and 4, define:
5
with
6
This metric's triangle-inequality and boundedness permit a trust-region guarantee via a modified TRPO argument:
7
where 8, yielding
9
This provides a rigorous trust-region bound leveraging the CIM (Guo et al., 2021).
4. CIM–PPO Surrogate Objective and Policy Update
The CIM–PPO algorithm replaces the KL divergence penalty in the PPO surrogate with the CIM penalty. The CIM–PPO surrogate is:
0
where 1 is the probability ratio and 2 the penalty coefficient. The policy-gradient becomes:
3
A clipping variant analogous to PPO-Clip can optionally be used for the advantage term. This objective maintains proximity via a direct metric on policy distributions and eliminates pathology from unbounded, asymmetric KL penalties (Guo et al., 2021).
5. Computational Complexity Analysis
A comparative assessment of per-sample computational requirements for Gaussian policy parameterizations:
| Method | Core Operation | Complexity (per sample) |
|---|---|---|
| PPO-KL | 4, 5 | 6 (full), 7 (diag) |
| PPO-KL | 8 | 9 |
| CIM–PPO | Elementwise kernel ops | 0 |
CIM–PPO's kernel-based penalty calculation relies on elementwise operations and a small number of exponentials or powers, in contrast with the matrix inversions and determinants required by KL-based PPO variants. As a result, actor-update cost for CIM–PPO is linear in the action dimension, offering scalability advantages for high-dimensional control (Guo et al., 2021).
6. Algorithm Structure: CIM–PPO in Practice
The CIM–PPO workflow can be summarized as:
- Initialize policy 1, value function 2, penalty 3, and kernel bandwidth 4.
- For each iteration:
- Run 5 for 6 timesteps to collect trajectories 7.
- Compute rewards-to-go 8 and advantage estimates 9 using 0.
- Update actor 1 via stochastic gradient ascent on
2
for 3 epochs with minibatches, optionally using clipping. - Update critic 4 to minimize
5
The design mirrors canonical actor–critic frameworks, replacing the trust-region penalty with the CIM (Guo et al., 2021).
7. Empirical Results on Continuous-Control Benchmarks
Experiments on four OpenAI Gym continuous-control tasks (Pendulum-v0, LunarLanderContinuous-v2, BipedalWalker-v3, BipedalWalkerHardcore-v3) and six Mujoco tasks (Hopper, Walker2d, HalfCheetah, Ant, Humanoid, Swimmer) assess CIM–PPO versus PPO-KL and PPO-Clip under standardized hyperparameters: discount 6, actor LR 7, critic LR 8, minibatch size 32, and 10 update epochs per iteration. PPO-KL targets 9, PPO-Clip uses 0, CIM–PPO sets 1, 2 (triangular kernel for low-dimension, biweight for higher).
Reported learning curves demonstrate:
- On Pendulum-v0, PPO-KL policy collapse after 31200 episodes, PPO-Clip plateaus, while CIM–PPO maintains improvement and higher reward.
- On LunarLanderContinuous, CIM–PPO achieves stable, high reward (4) throughout training, surpassing both baselines.
- BipedalWalker: CIM–PPO converges fastest to 5 reward; PPO-Clip plateaus at 6; PPO-KL lags at 7.
- BipedalWalkerHardcore: CIM–PPO reaches 8 reward, exceeding PPO-Clip (9) and PPO-KL (0 with high variance).
- Six Mujoco tasks: CIM–PPO matches or exceeds PPO-Clip by 1 in final return and attains that return in 2 fewer timesteps.
The experiments suggest that the boundedness and symmetry of CIM provide a more stable trust-region mechanism than asymmetric or heuristic penalties (Guo et al., 2021).
8. Theoretical Guarantee and Significance
CIM–PPO's use of a true metric on distribution space restores formal trust-region guarantees analogous to those in TRPO. The proof requires only that the kernel be symmetric and positive-definite; the advantage function must be uniformly bounded. The result:
3
furnishes a principled way to control policy deviation and performance degradation in each update, avoiding pitfalls associated with the asymmetric, unbounded KL penalty or the heuristic nature of clipping (Guo et al., 2021).
CIM–PPO constitutes a direct application of distributional metric theory to reinforcement learning, improving both theoretical rigor and empirical robustness in deep continuous control.