Distributed Multi Robot Lunar Cargo Transportation via Phase Decomposed Reinforcement Learning

Ashutosh Mishra\(^{*1}\), Elian Neppel\(^{1}\), Shreya Santra\(^{1}\), Antoine Jonquières\(^{2}\),
Muhammad Athallah Naufal\(^{3}\), Kentaro Uno\(^{1}\), and Kazuya Yoshida\(^{1}\)
12


Abstract

Modular reconfigurable robotic systems provide a scalable solution for cooperative surface operations in future lunar missions. However, cooperative cargo transportation remains challenging due to morphology-dependent topology changes, strong payload-induced coupling, long-horizon decision making, and safety constraints. This paper proposes a phase-decomposed reinforcement learning framework for cooperative cargo transport with distributed robotic units. The task is decomposed into lifting, transportation, and placement, each optimized with a dedicated joint-state policy capturing inter-agent coupling. Centralized training promotes stable convergence, while deployment uses onboard proprioception for control and OptiTrack motion capture for ground-truth evaluation and post-processed metrics. A deterministic phase controller expressed in Markov state representation regulates transitions between stages, and a failure-sensitive synchronization mechanism ensures coordinated progression and safety-aware halting during real-world execution. The framework is evaluated in simulation and through controlled field experiments at a JAXA space exploration test facility. Results demonstrate reliable cooperative transport across all stages in both simulation and hardware experiments.

1 Introduction↩︎

Sustained lunar surface missions require robotic systems capable of transporting construction materials, instruments, and logistical payloads under communication delay, terrain uncertainty, and strict safety constraints [1]. Modular reconfigurable robots are particularly suited to such environments, as they enable task-specific morphologies assembled from reusable units, supporting repairability, redundancy, and functional adaptability [2], [3].

Modularity introduces configuration-dependent kinematics, actuation topology, and inter-module coupling, reducing policy transferability across morphologies. In cooperative transport, additional coupling arises through the shared payload, producing strong dynamic interactions.

Existing decentralized reinforcement learning approaches allow distributed modules to act based on local observations while contributing to global objectives [4], [5]. Centralized training with decentralized execution improves convergence stability in multi-agent settings [6], [7]. Deep reinforcement learning has shown effectiveness for cooperative navigation and transport [8], [9]. However, most formulations assume fixed topology or optimize a single controller for structurally static teams, without addressing physical reconfiguration and topology-dependent coupling during task execution. For modular systems undergoing physical reconfiguration and multi-stage contact transitions, unified policy optimization can induce gradient interference across heterogeneous dynamics and fails to explicitly model configuration-dependent coupling [2], [10].

Figure 1: System overview of the proposed modular multi-phase reinforcement learning framework. Two distributed wheel arm integrated units cooperatively transport a shared payload through lifting, transportation, and placement stages.

In cooperative lunar cargo transport, the problem is compounded by sequential operational stages. Lifting, transportation, and placement involve distinct contact conditions and objective structures. Furthermore, modules must coordinate differently depending on configuration, for example, synchronized actuation during lifting, compliant steering during transit, and precise alignment during placement.

To address this, we propose a phase-decomposed reinforcement learning framework for cooperative lunar cargo transport. The mission is decomposed into lifting, transportation, and placement stages, each optimized using a dedicated joint-state policy. Execution remains distributed across modules. In this study the modular robots are assembled into a transport-capable configuration and the framework evaluates coordinated operation under this configuration. A high-level Markov decision process governs stage transitions, while a synchronization layer coordinates inter-module execution and enforces safety-aware halting. The framework is validated on our in-house modular reconfigurable robotic platform [11] through simulation and controlled experiments at a JAXA lunar-analog facility [12].

The key contributions are:

  1. Phase-structured reinforcement learning for cooperative transport under configuration-dependent dynamics and payload-induced coupling, separating policy optimization across distinct contact regimes.

  2. Centralized joint-state training with distributed execution for mechanically coupled yet physically independent modular robots.

  3. Discrete MDP-based phase gating with synchronization-aware safety coordination for sequential multi-stage execution.

  4. Simulation and lunar-analog hardware validation on a modular wheel–arm robotic system assembled for cooperative cargo transport.

Beyond integrating established components, the central methodological contribution is the phase-structured formulation itself: a long-horizon, contact-rich cooperative-transport task is decomposed into phase-specific MDPs with dedicated reward geometries and coupled by a deterministic, synchronization-aware phase gate that keeps multi-stage sim-to-real execution stable and safety-bounded. This structure is not merely organizational. A monolithic policy trained over the same joint state and action space fails to converge (Sec. 7.2), indicating that the decomposition mitigates gradient interference across heterogeneous contact regimes rather than re-packaging PPO, centralized training with decentralized execution, and MDP gating.

2 Related Work↩︎

Multi-robot collaboration for space and planetary applications has focused on modular and distributed logistics, coordinated construction, and collective manipulation under communication constraints [13], [14]. Model-based decentralized transport strategies have demonstrated feasibility in structured environments [15], [16]. However, these approaches depend on accurate dynamic models and pre-defined coordination logic.

Multi-agent reinforcement learning provides an alternative framework for adaptive coordination. Early decentralized learning formulations established theoretical foundations for distributed control [4]. Recent multi-agent reinforcement learning emphasizes centralized training with decentralized execution [5], [17]. Applications to decentralized navigation and cooperative transport demonstrate scalability benefits [8], [9]. Recent work also investigates communication-efficient decentralized training mechanisms [7], [18]. Despite these developments, most studies optimize a single policy across heterogeneous task phases.

Modular robotics introduces additional structural variability. Learning-based control generalization across reconfigurable morphologies has been explored to enhance adaptability [2], [3]. Morphology-aware co-optimization strategies have further improved transferability across configurations [19]. Robust synchronization strategies for modular systems have addressed distributed actuation coordination [20]. However, the structured decomposition of cooperative transport into phase-specific reinforcement learning problems, combined with synchronization-governed stage transitions, has not been systematically investigated.

The present work addresses this gap by introducing a phase-structured reinforcement learning architecture that integrates joint-state training, distributed execution, and safety-aware synchronization for cooperative lunar cargo transport.

3 Modular Robotic System↩︎

The robotic platform considered in this work [11] consists of physically distinct locomotion and manipulation modules that can be mechanically combined to realize task specific configurations. The architecture separates mobility and manipulation at the hardware level, enabling structural reconfiguration according to operational requirements.

3.1 Wheel Module↩︎

The locomotion subsystem is implemented as a differential drive wheel module. The module comprises a rigid base with two independently actuated wheels, enabling planar motion through differential velocity control. The kinematic structure is nonholonomic and constrained to planar translation and rotation about the vertical axis. The module provides mobility for transport and positioning tasks and serves as the base platform when coupled with a manipulation module.

3.2 Arm Module↩︎

The manipulation subsystem is realized as a serial articulated arm with seven degrees of freedom. The kinematic chain enables three dimensional end effector positioning and orientation control. Each joint is independently actuated, allowing coordinated motion for grasping, lifting, and payload stabilization tasks. The arm module can operate independently or be mechanically attached to a mobility base.

3.3 Wheel and Arm Integrated Configuration↩︎

A mobile manipulation unit is formed by mechanically coupling one differential drive wheel module with one seven degree of freedom arm module. In this configuration, the wheel base provides planar mobility while the arm enables object interaction. The combined system exhibits coupled base and manipulator dynamics during motion and load handling. For cooperative cargo transportation, multiple integrated units physically engage a shared payload. The overall system architecture and phase-structured task execution are illustrated in Fig. 1. The present study evaluates the proposed framework on a transport-capable assembly of the modular system; investigating policy transfer across alternative robot configurations is an important direction for future work. At the module level, learned policies can control new morphologies and generalize to previously unseen assemblies, as we demonstrate in prior work [21]. The arms establish contact and generate interaction forces, while the wheel bases provide coordinated locomotion.

4 Modular Multi-Phase Reinforcement Learning Policies↩︎

The cooperative cargo transport problem is formulated as a phase-structured sequential decision-making process composed of three operational stages: lifting (\(L\)), transportation (\(T\)), and placement (\(P\)). The execution is sequential, whereas policy learning remains modular. Each stage is optimized independently over a shared joint state–action interface, enabling replacement or refinement of individual policies without retraining the full framework.

System-Level Formulation↩︎

Let \(N\) denote the number of integrated wheel–arm units. The coupled system state at time \(t\) is defined as

\[s_t = \left[ x_t^{(1)}, \dots, x_t^{(N)}, x_t^{c} \right],\]

where \(x_t^{(i)}\) represents the pose, velocity, and joint configuration of robot \(i\), and \(x_t^{c}\) denotes the pose and velocity of the cargo.

The joint action vector is

\[a_t = \left[ u_t^{(1)}, \dots, u_t^{(N)} \right],\]

where \(u_t^{(i)}\in\mathbb{R}^{9}\) contains the two differential-drive wheel velocity commands and the seven arm-joint position commands of robot \(i\); for the \(N{=}2\) transport-capable configuration the joint action is therefore \(a_t\in\mathbb{R}^{18}\).

Each operational phase \(k \in \{L, T, P\}\) is modeled as a Markov Decision Process

\[\mathcal{M}_k = (\mathcal{S}, \mathcal{A}, \mathcal{P}, r_k, \gamma),\]

where \(\mathcal{S}\) and \(\mathcal{A}\) are joint state and action spaces, \(\mathcal{P}\) denotes transition dynamics, \(r_k\) is the phase-specific reward, and \(\gamma \in (0,1)\) is the discount factor. Policies are trained under full joint-state observation and executed using local observations with shared cargo state.

Observation Model↩︎

During centralized training:

\[o_t = s_t.\]

During distributed execution, robot \(i\) observes

\[o_t^{(i)} = \left[ x_t^{(i)}, \Delta x_t^{c}, \Delta x_t^{(j)} \right],\]

where \(\Delta x_t^{c}\) denotes relative cargo pose and \(\Delta x_t^{(j)}\) encodes relative neighboring robot states.

Because the system is assembled from a fixed set of known modules, the per-unit observation \(o_t^{(i)}\) and action \(u_t^{(i)}\) have fixed, predetermined dimensions, so the policy input and output dimensions are consistent between training and deployment.

Stage I: Lifting Policy↩︎

The lifting stage aims to raise the cargo to a target height \(h^\ast\) while maintaining balanced load distribution and minimizing oscillatory behavior.

The reward function is \[r_L = - w_{h}\Big(\frac{h_t-h^\ast}{h_{\text{tol}}}\Big)^2 - w_{\dot{h}}\Big(\frac{\dot{h}_t}{\dot{h}_{\text{tol}}}\Big)^2 - w_{f}\Big(\frac{\sigma_f}{\sigma_{f,\text{tol}}}\Big) - w_{\theta}\Big(\frac{\theta_{\text{tilt}}}{\theta_{\text{tol}}}\Big)^2 .\]

where \(h_t\) denotes cargo height, \(\dot{h}_t\) vertical velocity, \(\sigma_f\) the variance of interaction forces across arms, \(\theta_{\text{tilt}}\) the cargo tilt angle, and \(w_h, w_{\dot{h}}, w_f, w_\theta > 0\) are weighting coefficients. The inclusion of \(\sigma_f\) explicitly enforces balanced cooperative lifting across wheel–arm units. Force signal used for \(\sigma_f\). In simulation, interaction forces at each arm-payload contact were obtained directly from the physics engine contact wrench and \(\sigma_f\) was computed across arms at each step. In hardware, direct contact force sensing was not available; therefore, \(\sigma_f\) was used only during simulation training as a regularizer to promote balanced load sharing, while deployment relied on encoder-based state feedback and the synchronization clamp to suppress asymmetric transients.

Stage II: Transportation Policy↩︎

The transportation stage governs cooperative planar motion from \(p_A\) to \(p_B\) while preserving payload stability.

The reward is defined as

\[\label{eq:reward95transport} \begin{align} r_T =\;& - w_p \Big(\frac{\|p_t - p_B\|}{p_{\mathrm{tol}}}\Big)^2 - w_v \Big(\frac{\|\dot{p}_t - v_{\mathrm{cmd}}\|}{v_{\mathrm{tol}}}\Big)^2 \\ &- w_\theta \Big(\frac{\theta_{\mathrm{tilt}}}{\theta_{\mathrm{tol}}}\Big)^2 - w_h \Big(\frac{h_t - h^\ast}{h_{\mathrm{tol}}}\Big)^2 . \end{align}\tag{1}\]

where \(p_t\) denotes cargo planar position, \(\dot{p}_t\) planar velocity, \(v_{\text{cmd}}\) the commanded transport velocity, \(\theta_{\text{tilt}}\) enforces upright stability, and \(h_t\) preserves lifting height. This formulation decouples translational tracking from stability and height regulation.

Stage III: Placement Policy↩︎

The placement stage ensures controlled lowering and precise alignment at the target configuration.

The reward is \[\begin{align} r_P =\;& - w_{p}\Big(\frac{\|p_t-p_B\|}{p_{\text{tol}}}\Big)^2 - w_{h0}\Big(\frac{h_t}{h_{0,\text{tol}}}\Big)^2 \\ &- w_{\dot{h}}\Big(\frac{\dot{h}_t}{\dot{h}_{\text{tol}}}\Big)^2 - w_{\theta}\Big(\frac{\theta_{\text{tilt}}}{\theta_{\text{tol}}}\Big)^2 . \end{align}\]

where \(h_t\) penalizes residual height, \(\dot{h}_t\) ensures smooth descent, \(\theta_{\text{tilt}}\) enforces upright alignment, and \(w_p, w_{h0}, w_{\dot{h}}, w_\theta > 0\) are weighting coefficients.

Policy Optimization↩︎

Each phase policy \(\pi_{\theta_k}\) maximizes expected discounted return

\[J(\theta_k) = \mathbb{E}_{\pi_{\theta_k}} \left[ \sum_{t=0}^{T_k} \gamma^t r_k(s_t, a_t) \right],\]

where \(T_k\) denotes the phase horizon.

Policies are trained independently under centralized observation and deployed in distributed execution governed by a higher-level phase MDP and synchronization layer.

5 Policy Coordination and Sim-to-Real Deployment↩︎

Training in simulation yields stage-specific policies for lifting, transportation, and placement. However, direct deployment can be destabilized by actuator latency, wheel slip, contact uncertainty, and state-estimation noise, due to distributed modular hardware with dynamically coupled subsystems. To reduce this sim-to-real discrepancy, execution is governed by a hybrid coordination layer composed of (i) a discrete phase MDP that gates the active policy and (ii) a joint synchronization mechanism that bounds per-step progression of the coupled wheel–arm–cargo–arm–wheel system.

5.1 Phase Coordination via MDP Gating↩︎

We augment the continuous system state \(s_t\) with a discrete phase variable \[m_t \in \mathcal{M}\triangleq\{0,1,2,3\}, \qquad x_t \triangleq (s_t,m_t), \label{eq:hybrid95mode95state}\tag{2}\] where \(0\) denotes Idle, \(1\) Lift, \(2\) Transport, and \(3\) Place.

As illustrated in Fig. 2, the discrete mode variable gates the active stage policy through \((g_{\text{lift}}, g_{\text{move}}, g_{\text{place}})\), enabling sequential progression with safety-stop overrides.

Only one policy is active at any time, selected by indicator gating: \[a_t = \mathbb{I}[m_t=1]\pi_{\ell}(s_t) + \mathbb{I}[m_t=2]\pi_{t}(s_t) + \mathbb{I}[m_t=3]\pi_{p}(s_t). \label{eq:gated95action95compact}\tag{3}\]

Figure 2: Phase MDP gating for multi-stage cooperative cargo transport. The discrete mode variable m_t \in \{0,1,2,3\} activates exactly one phase policy at a time through indicator gating, with M0 denoting Idle, M1 Lift (\pi_\ell), M2 Transport (\pi_t), and M3 Put Down (\pi_p). The gating variables satisfy (g_{\text{lift}},g_{\text{move}},g_{\text{place}})\in\{(1,0,0),(0,1,0),(0,0,1)\} for active phases and (0,0,0) in Idle. Green transitions denote task-driven progression, whereas red transitions indicate abort or safety-stop overrides that return the system to Idle.

Phase transitions follow measurable task events with a safety override: \[m_{t+1}= \begin{cases} 0, & \sigma_t=1,\\ 1, & m_t=0 \wedge \mathrm{start}(s_t)=1,\\ 2, & m_t=1 \wedge h_{\mathrm{payload}}(s_t)\ge h_{\min},\\ 3, & m_t=2 \wedge d_{\mathrm{goal}}(s_t)\le \tau_{\mathrm{place}},\\ 0, & m_t=3 \wedge \mathrm{released}(s_t)=1,\\ m_t, & \text{otherwise}. \end{cases} \label{eq:mode95transition}\tag{4}\] Here \(\sigma_t\) denotes a safety-stop signal (fault, timeout, or operator intervention), enforcing immediate halt and reset to Idle.

5.2 Joint Synchronization for Cargo Mode↩︎

When the two units carry a shared payload, they form a closed kinematic chain, \[\texttt{Wheel}_1 - \texttt{Arm}_1 - \texttt{Cargo} - \texttt{Arm}_2 - \texttt{Wheel}_2 ,\] so uncoordinated motion of any one subsystem is resisted by the others through the rigid cargo. This layer therefore admits as much of each policy command as possible while ensuring that no subsystem advances far ahead of the others, by clamping the commanded motion to a small admissible region around the current sensed state at every control step \(k\).

We stack the sensed states of the two arms and two wheel bases at step \(k\) into a single vector \[Y_k \triangleq \begin{bmatrix} g_k^{(A_1)}\\ g_k^{(A_2)}\\ p_k^{(W_1)}\\ p_k^{(W_2)} \end{bmatrix}, \qquad g_k^{(A_i)}\in SE(3),\; p_k^{(W_j)}\in\mathbb{R}^2, \label{eq:stacked95state}\tag{5}\] where \(g_k^{(A_i)}\) is the pose of arm \(i\) and \(p_k^{(W_j)}\) the planar position of wheel base \(j\). The policy output \(a_k\) defines a desired increment that, applied to the previous command \(U_{k-1}\), produces the raw target state \[F_k \triangleq U_{k-1}\oplus \Delta(U_{k-1},a_k), \label{eq:target95state}\tag{6}\] where \(\oplus\) denotes composition on the \(SE(3)\) (arm) components and addition on the planar wheel components.

Rather than issuing \(F_k\) directly, we require the commanded stacked state to remain within an admissible set centered on the sensed state \(Y_k\): \[\mathcal{V}_k \triangleq \{ X \mid d(X,Y_k) \le 1 \}, \label{eq:admissible95set}\tag{7}\]

\[\begin{align} d(X,Y) \triangleq \max \Big( & d_{SE}(X^{(A_1)},Y^{(A_1)}), \\ & d_{SE}(X^{(A_2)},Y^{(A_2)}), \\ & \| (X^{(W_1)}-Y^{(W_1)}) \oslash \Delta_w \|_2, \\ & \| (X^{(W_2)}-Y^{(W_2)}) \oslash \Delta_w \|_2 \Big). \end{align} \label{eq:stacked95metric95clean}\tag{8}\] where the distance \(d\) is the worst case (an \(\ell_\infty\) maximum) over the four subsystems, each normalized by its own tolerance, and \(\oslash\) is elementwise division by the per-axis tolerance \(\Delta_w\), so that a wheel term reaches \(1\) when the combined normalized planar deviation attains unit magnitude. The arm term uses a normalized pose distance \[d_{SE}(g_1,g_2)=\max\Big(\frac{\|t_1-t_2\|_2}{d_{\max}},\frac{\angle(R_1^\top R_2)}{\theta_{\max}}\Big),\] with \(g=(R,t)\in SE(3)\) and \(\angle(\cdot)\) the rotation angle, which reaches \(1\) when either the translation error hits \(d_{\max}=0.03\) m or the rotation error hits \(\theta_{\max}=5^\circ\). The wheel deviation bound is \(\Delta_w=[0.02,\,0.02]^\top\) m. Consequently \(d(\cdot,Y_k)\le 1\) holds only if every subsystem stays within its tolerance simultaneously.

Finally, let \(\mathcal{T}(t,S_k,F_k)\) interpolate from the previous command \(S_k\triangleq U_{k-1}\) to the target \(F_k\) as \(t\) ranges over \([0,1]\). We issue the largest feasible step toward the target that still satisfies the admissibility constraint: \[\begin{align} t_k^{\max} &\triangleq \max \Big\{ t \in [0,1] \;\Big|\; \mathcal{T}(t,S_k,F_k)\in \mathcal{V}_k \Big\}, \\ U_k &\triangleq \mathcal{T}\!\left(t_k^{\max}, S_k, F_k\right). \end{align}\]

Because the admissibility constraint is an \(\ell_{\infty}\) maximum over subsystems, the subsystem closest to its per-step deviation bound caps \(t_k^{\max}\) and hence the progress of the whole assembly, preventing the asynchronous arm or wheel motion that would otherwise induce internal cargo stress. If \(t_k^{\max}\) is undefined (no feasible interpolation) or a safety stop is active (\(\sigma_k=1\)), the controller defaults to a hold command, thereby ensuring bounded deviation and safe stabilization.

6 Experimental Setup↩︎

6.1 Simulation Environment and Training Configuration↩︎

Figure 3: IsaacLab simulation and centralized PPO training architecture for phase-specific joint-state policies under articulated multi-body contact dynamics and domain randomization.

All phase-specific policies were trained in NVIDIA IsaacLab [22] using articulated rigid-body simulation with contact dynamics, as illustrated in Fig. 3. The cooperative system consisted of two integrated wheel–arm units mechanically coupled through a rigid payload in the configuration Wheel\(_1\)–Arm\(_1\)–Cargo–Arm\(_2\)–Wheel\(_2\).

The robot model was defined via URDF and deployed in USD-based scenes within Isaac Sim. Each training scene contained a contact-enabled planar ground surface and illumination sources. To improve sample efficiency, 2048 to 4096 parallel environments were instantiated during training depending on the phase (Table 1).

The policy and value functions were implemented as separate multilayer perceptrons with ELU activations, optimized with PPO. The hidden-layer sizes are \([256, 128, 64]\) for the lifting and placement policies and \([128, 128, 128]\) for the transportation policy.

Moderate domain randomization was introduced, including bounded friction variation, terrain perturbations, actuation noise, and initial joint variation. Domain randomization ranges were: wheel-ground friction coefficient \(\mu\sim \mathcal{U}(0.6,1.0)\), terrain heightfield perturbations with peak-to-peak amplitude \(\leq 0.02\) m and spatial correlation length in \([0.05,0.15]\) m, wheel velocity command noise \(\epsilon_\omega\sim\mathcal{N}(0,0.05^2)\) rad/s, and arm joint command noise \(\epsilon_q\sim\mathcal{N}(0,0.01^2)\) (in joint units).

Table 1: Per-phase PPO training configuration. The learning rate is the initial value under an adaptive schedule.
Parameter Lifting Transport Placement
Algorithm PPO PPO PPO
Learning rate \(1\times10^{-4}\) \(1\times10^{-3}\) \(1\times10^{-4}\)
Discount factor \(\gamma\) 0.98 0.99 0.98
GAE parameter \(\lambda\) 0.95 0.95 0.95
Clip ratio 0.2 0.2 0.2
Epochs per update 5 5 5
Mini-batches 4 4 4
Entropy coefficient 0.006 0.005 0.006
Parallel environments 2048 4096 2048

4pt

Table 2: Reward tolerances and weights used in all experiments.
Term Tolerance Weight
Height error \(h_{\text{tol}}\) \(0.03\) m \(w_h=1.0\)
Vertical speed \(\dot h_{\text{tol}}\) \(0.10\) m/s \(w_{\dot h}=0.5\)
Tilt \(\theta_{\text{tol}}\) \(10^\circ\) (\(0.174\) rad) \(w_\theta=0.7\)
Planar position \(p_{\text{tol}}\) \(0.20\) m \(w_p=1.0\)
Planar speed \(v_{\text{tol}}\) \(0.05\) m/s \(w_v=0.5\)
Force-variance \(\sigma_{f,\text{tol}}\) \(1.0\) (normalized) \(w_f=0.3\)
Final height \(h_{0,\text{tol}}\) \(0.02\) m \(w_{h0}=1.0\)

The per-phase training configuration is summarized in Table 1.

6.2 Hardware Platform and Field Setup↩︎

Hardware validation of the proposed multi-phase framework was conducted at JAXA’s Advanced Facility for Space Exploration in a controlled lunar-analog environment. Experiments were performed on a regolith-like sand test field designed to emulate lunar surface conditions. The terrain consisted of loose granular soil with non-uniform compaction and mild surface irregularities, introducing wheel slip, contact uncertainty, and uneven load distribution. This environment enables evaluation under terrain-induced disturbances.

The experimental platform comprised four independently operable robots from our modular reconfigurable system: two wheel bases and two articulated arms. Each wheel–arm pair was physically assembled into a mobile manipulation unit, and two such units engaged a shared payload. Although physically distributed and reconfigurable, the coupled system was trained under centralized joint-state observation and executed in a distributed manner, validating the framework under real hardware reconfiguration. Each unit combined a differential drive wheel base for planar locomotion with a seven degree of freedom articulated manipulator, instrumented with motor encoders for joint position and velocity measurement. The two manipulators engaged the payload from opposing sides, forming a mechanically coupled system throughout execution. No external structural support was provided.

To evaluate robustness across mass and geometry variations, three payload configurations were tested: a Sledge (5.3 kg elongated structure), a Cargo box (7.5 kg), and Cargo with bricks (9.5 kg).

The sledge configuration differed in geometry from the nominal box model used during simulation training, enabling evaluation of policy generalization to previously unseen object shape. All three operational phases were executed sequentially for each payload configuration. The sequential hardware execution of the lifting, transportation, and placement stages under phase-specific policies is illustrated in Fig. 4.

Figure 4: Sequential hardware execution of lifting (M1), transportation (M2), and placement (M3) during cooperative cargo transport in a lunar-analog field experiment. Each row corresponds to a phase-specific policy \pi_{\theta_k} with reward r_k, activated through discrete MDP gating under distributed execution.

Ground-truth pose of the payload and robot bodies was measured using an OptiTrack motion capture system. Reflective markers attached to the payload and robot modules provided real-time pose estimation. Wheel velocities and joint states were recorded from onboard encoders. For each payload configuration, multiple trials were conducted and all state data were logged for post processing and comparison with simulation results.

At deployment, each unit runs its active phase policy onboard at a fixed control rate from local proprioceptive observations \(o_t^{(i)}\), with the relative cargo and neighbor states obtained from joint encoders and the forward kinematics of the rigid grasp. The discrete phase-MDP gate selects the active policy, and the synchronization clamp of Section 5 is applied online before commands are issued to the wheel and arm actuators. The OptiTrack measurements serve only for offline ground-truth evaluation and the reported metrics; they are not used in the control loop.

7 Results and Discussion↩︎

7.1 Policy Performance in Simulation↩︎

Simulation experiments were conducted to evaluate stage-specific performance of the lifting, transportation, and placement policies. Performance was assessed using state-based metrics rather than cumulative reward.

Lifting↩︎

Figure 5: Simulation results for the lifting policy. (a) PPO training convergence for different target heights. (b) Representative rollout showing payload altitude and tilt evolution during lifting.

Figure 5 reports the learning behavior and execution performance of the lifting policy across target heights of 0.3, 0.5, and 0.75 m.

As shown in Fig. 5 (a), reward converges in \(1.72\times10^4\), \(2.28\times10^4\), and \(2.43\times10^4\) timesteps for the 0.3, 0.5, and 0.75 m targets, respectively. The final plateau reward decreases from 105.9 to 94.7 as the lift height increases, reflecting the increased torque demand and sensitivity to coupling at higher elevations.

Trajectory-level evaluation in Fig. 5 (b) shows monotonic height increase without vertical overshoot. The final height tracking error scales with target elevation, remaining at 5.9 mm, 17.4 mm, and 36.0 mm for the 0.3, 0.5, and 0.75 m lifts, respectively. Peak transient tilt occurs during mid-lift torque amplification, where \(\theta_{tol}\) acts as a reward-shaping reference rather than a strict safety bound, reaching 17.9\(^\circ\), 22.8\(^\circ\), and 25.4\(^\circ\) across the three targets. Following this transient phase, the payload stabilizes with steady-state tilt bounded below 2.5\(^\circ\) in all cases.

Transport↩︎

Figure 6 summarizes the learning behavior and closed-loop performance of the transport policy.

As shown in Fig. 6 (a), the mean episodic reward increases steadily and reaches 95% of its final plateau at approximately \(1.74\times10^4\) timesteps, indicating rapid convergence relative to the \(5\times10^4\) training horizon. The peak reward overshoot is limited to 8.9% (122.1 vs.), and the entropy term decays smoothly without collapse.

Figure 6 (b) evaluates trajectory-level behavior under a 0.2 m/s velocity command. The controller achieves an RMS velocity tracking error of 0.023 m/s (approximately 11% of command magnitude), with smooth acceleration and deceleration profiles and no oscillatory instability. Simultaneously, the payload height is maintained at 0.30 m with a maximum deviation of 1.58 mm, demonstrating effective decoupling between longitudinal motion and vertical support.

Figure 6: Simulation results for the transport policy. (a) PPO training convergence showing reward saturation and gradual entropy decay. (b) Forward velocity tracking (v_{\text{cmd}} vs.v_{\text{actual}}) and concurrent payload height regulation during representative execution.

Lowering and Placement↩︎

Figure 7: Simulation results for the lowering policy. (a) PPO training convergence showing rapid reward stabilization and gradual entropy decay. (b) Representative rollout illustrating altitude descent and tilt evolution during payload placement.

Figure 7 presents the learning behavior and execution performance of the lowering policy. As shown in Fig. 7 (a), the policy converges within \(1.42\times10^4\) timesteps, reaching 95% of its final plateau faster than the lifting and transport phases. Reward evolution is smooth and entropy decays without instability.

Trajectory-level evaluation in Fig. 7 (b) shows controlled descent toward ground contact. The final placement height error is 13.3 mm relative to the ground reference. The impact velocity at contact is 0.030 m/s, which is substantially lower than free-fall velocity from 0.3 m, confirming a soft touchdown rather than ballistic descent. Peak transient tilt during mid-descent is limited to 6.0\(^\circ\), and the steady-state tilt after contact is 0.6\(^\circ\), indicating stable upright placement.

7.2 Baseline Training Without Phase Decomposition↩︎

To evaluate the necessity of phase decomposition and joint sync, we trained a single centralized policy for the full mission without stage separation or synchronization. As shown in Fig. 8, the monolithic policy fails to converge even after \(7.5\times10^{4}\) timesteps, which exceeds the training horizon used for the individual phase policies. The reward remains oscillatory and entropy does not exhibit sustained decay, indicating that the controller fails to learn a stable strategy. Qualitative simulation behaviour further shows unstable motion where the robot attempts to move while lifting or placing the payload simultaneously. These results suggest that phase decomposition is necessary for stable learning under configuration-dependent coupling.

Figure 8: Centralized training without phase decomposition.Left: reward and entropy evolution for a monolithic policy trained over the entire task.Right: representative simulation snapshot showing unstable behaviour where translation occurs during lifting or placement.

7.3 Evaluation on Hardware↩︎

Figure 9: Full hardware mission profile showing lifting, transportation, and lowering for three payload masses. Curves represent mean height over three trials per payload. The dashed line indicates the 0.30 m reference height.

Hardware validation was conducted for three payload configurations (5.3 kg, 7.5 kg, and 9.5 kg), with three trials per payload. Figure 9 shows the averaged payload height profile across the full mission (lifting, transport, and lowering). All values reported below are mean \(\pm\) 1\(\sigma\) over trials.

During transportation, the steady-state payload height decreases monotonically with mass, measuring \(272.6 \pm 3.8\) mm (5.3 kg), \(243.0 \pm 5.5\) mm (7.5 kg), and \(209.1 \pm 7.8\) mm (9.5 kg). Relative to the 0.30 m target, these correspond to approximately 91%, 81%, and 70% height retention, respectively. The increased undershoot and variance with payload mass are consistent with terrain sinkage and amplified arm torque coupling under heavier loading.

The time required to reach 95% of steady-state height also increases with mass, from \(21.02 \pm 0.61\) s (5.3 kg) to \(30.42 \pm 0.76\) s (7.5 kg) and \(31.18 \pm 0.56\) s (9.5 kg), indicating slower vertical convergence under higher load.

During placement, all payloads exhibit controlled descent with small residual post-contact height. The final settled height after lowering is \(8.1 \pm 0.7\) mm, \(8.0 \pm 1.0\) mm, and \(12.1 \pm 1.0\) mm for increasing mass, respectively. The slightly higher residual height for the heaviest payload is attributable to deeper sand penetration and local terrain deformation.

Hardware results confirm stable multi-phase execution across payload masses. While absolute lifting height reduces under heavier payloads, vertical regulation remains consistent and bounded across trials, demonstrating robust completion with degraded height regulation under heavier loads in granular terrain conditions.

8 CONCLUSIONS↩︎

This work presented a phase-decomposed reinforcement learning framework for cooperative lunar cargo transport using physically reconfigurable wheel-arm robotic units. Stage-specific policies were trained in simulation and validated through lunar-analog hardware experiments, demonstrating stable multi-phase execution under varying payload masses and terrain conditions. The results indicate that phase decomposition with synchronized execution improves robustness to configuration-dependent dynamics and sim-to-real discrepancies.

References↩︎

[1]
, “Artemis program overview,” 2025. [Online]. Available: https://www.nasa.gov/specials/artemis/.
[2]
J. Whitman, et al., “Learning modular robot control policies,” IEEE Transactions on Robotics, vol. 39, no. 5, pp. 4095–4113, 2023.
[3]
T. Iwata et al., “Adaptive modular reinforcement learning for robot controlled in multiple environments,” IEEE Access, vol. 9, pp. 103 032–103 043, 2021.
[4]
L. Busoniu et al., “Decentralized reinforcement learning control of a robotic manipulator,” in ICARCV, 2006.
[5]
Y. Wang et al., “Distributed reinforcement learning for robot teams: A review,” Current Robotics Reports, vol. 3, no. 4, pp. 239–257, 2022.
[6]
G. Sartoretti et al., “Distributed learning of decentralized control policies for articulated mobile robots,” IEEE Transactions on Robotics, vol. 35, no. 5, pp. 1109–1122, 2019.
[7]
Y. Guo et al., “Decentralized motor skill learning for complex robotic systems,” IEEE Robotics and Automation Letters, vol. 8, no. 9, pp. 5791–5798, 2023.
[8]
L. Zhang et al., “Decentralized control of multi-robot system in cooperative object transportation using deep reinforcement learning,” IEEE Access, vol. 8, pp. 184 109–184 119, 2020.
[9]
R. Han et al., “Reinforcement learned distributed multi-robot navigation with reciprocal velocity obstacle shaped rewards,” IEEE Robotics and Automation Letters, vol. 7, no. 3, pp. 5896–5903, 2022.
[10]
T. Yu, et al., “Gradient surgery for multi-task learning,” in NeurIPS, 2020.
[11]
K. Uno, et al., MoonBot: Modular and on-demand reconfigurable robot toward moon base construction,” IEEE Transactions on Field Robotics, vol. 2, pp. 847–874, 2025.
[12]
, “R&d facility for open innovation and space exploration projects,” 2017. [Online]. Available: https://www.ihub-tansa.jaxa.jp/english/files/JAXA_inb_brochurer.pdf.
[13]
Wang et al., “A collaborative path planning approach for multiple robots persistently building a lunar base,” Acta Astronautica, 2025.
[14]
E. Polizzi et al., “Multi-robot cooperation for lunar in-situ resource utilization,” Frontiers in Robotics and AI, vol. 10, p. 1149080, 2023.
[15]
H. Farivarnejad et al., “Fully decentralized controller for multi-robot collective transport in space applications,” in IEEE Aerospace Conference, 2021.
[16]
R. Wang et al., “Model-based reinforcement learning for decentralized multiagent rendezvous,” in CoRL, 2021.
[17]
S. Gronauer et al., “Multi-agent deep reinforcement learning: A survey,” Artificial Intelligence Review, vol. 55, no. 2, pp. 895–943, 2022.
[18]
J. S. Nair et al., “On decentralizing federated reinforcement learning in multi-robot scenarios,” in SEEDA-CECNSM, 2022.
[19]
Z. Gao, et al., “Co-optimizing reconfigurable environments and policies for decentralized multi-agent navigation,” IEEE Transactions on Robotics, pp. 1–20, 2025.
[20]
E. Neppel et al., “Robust and modular multi-limb synchronization in motion stack for space robots with trajectory clamping via hypersphere,” in IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), 2025, pp. 3767–3772.
[21]
A. Mishra, et al., “Multi-modal decentralized reinforcement learning for modular reconfigurable lunar robots,” in IEEE International Conference on Space Robotics (iSpaRo), 2025, pp. 329–335.
[22]
M. Mittal, et al., “Isaac lab: A gpu-accelerated simulation framework for multi-modal robot learning,” arXiv preprint arXiv:2511.04831, 2025. [Online]. Available: https://arxiv.org/abs/2511.04831.

  1. This work was supported by JST Moonshot R&D Program, Grant Number JPMJMS223B.↩︎

  2. \(^{1}\)A. Mishra, E. Neppel, S. Santra, K. Uno, and K. Yoshida are with the Space Robotics Lab. (SRL), Department of Aerospace Engineering, Graduate School of Engineering, Tohoku University, Sendai 980–8579, Japan. \(^{2}\)A. Jonquières is with École Centrale de Lille, France. \(^{3}\)M. A. Naufal is with Institut Teknologi Bandung, Indonesia. \(^{*}\)Corresponding author: A. Mishra (ashutosh.mishra@dc.tohoku.ac.jp).↩︎