Search papers, labs, and topics across Lattice.
This paper introduces PRIME, a novel approach for enhancing multi-agent reinforcement learning in non-stationary environments, specifically targeting UAV-assisted emergency communication networks. It addresses the critical issue of dormant neurons in shared-parameter settings by implementing a bidirectional verification process that ensures only truly inactive neurons are reset, thereby preserving useful representations while restoring learning capacity. The results demonstrate a significant 24.9% improvement in performance over existing methods, with a notable reduction in dormant neuron fractions, highlighting the effectiveness of gradient-silent neuron identification and team-level aggregation.
Dormant neurons in multi-agent systems can be safely reactivated without losing valuable learned representations, leading to a 24.9% boost in performance.
Most reinforcement learning controllers for these networks assume stationary conditions, and the few that handle change react to the external environment while leaving the network's internal state unexamined. We show that sustained non-stationarity damages this internal state directly: as objectives shift, neurons progressively fall dormant and the shared policy loses the capacity to learn. The obvious remedy, resetting dormant neurons, is unsafe under shared-parameter multi-agent training: many neurons that appear inactive are still receiving strong training gradients, and whether a neuron appears dormant depends on which agent's observations it processes. PRIME (Plasticity Recovery In Multi-agent Environments) therefore verifies both directions before intervening. Extending the bidirectional Silent Neuron framework to cooperative multi-agent reinforcement learning, it aggregates activation and gradient statistics over the full team batch, reads the backward signal from the gradient the training loss has already deposited , not from a hand-crafted proxy, and reinitializes only neurons that are simultaneously activation-dormant and gradient-silent. Useful representations are preserved while learning capacity is restored. On a phase-switching UAV emergency communication simulator, PRIME improves interquartile mean return by 24.9\% over MAPPO and holds dormant neuron fractions at 10--20\% versus 40--45\%; ablations attribute the gains to the gradient signal and team-level aggregation rather than to the specific reset operator. A dynamic regret bound shows that the perturbation cost scales with the small silent-subspace dimension rather than the full parameter count.
