---
title: Multi-hop Split Learning (MSL)
url: https://www.emergentmind.com/topics/multi-hop-split-learning-msl
type: topic
---

# Multi-hop Split Learning (MSL)

Searching arXiv for recent papers on multi-hop split learning, pipelined split learning, and SFC-based split learning.
Multi-hop Split Learning (MSL) is a form of split learning in which a global neural network is partitioned into multiple sequential sub-models and executed across a client, a server, and one or more intermediate nodes, so that forward activations and backward gradients traverse a multi-hop path rather than a single client–server cut. In the Service Function Chaining (SFC)-based formulation, split sub-models are interpreted as service functions and their composition forms a service chain representing the global model; in parallel and pipelined formulations, the same basic idea appears as a multistage training pipeline across multiple compute nodes [2509.10001][2402.00208][2505.04368][2604.25197]. Across these strands, MSL is studied as a joint systems problem involving model partitioning, placement, communication, synchronization, routing, latency, memory footprint, and, in some variants, privacy against eavesdroppers [2507.07323].

## 1. Formal model and notation

A standard formalization considers a client, a server, and $H-1$ intermediate nodes, each hosting one split sub-model. The global model $F$ with $L$ layers is written as
$$
F = f^1 \circ f^2 \circ \dots \circ f^L,
$$
and partitioned into $H$ sub-models
$$
F_h = f^{l_{h-1}+1} \circ \dots \circ f^{l_h},
$$
with $\sum_{h=1}^H |\mathrm{layers}(F_h)| = L$. If $x \in \mathbb{R}^d$ is the input, $y \in \mathbb{R}^C$ the ground-truth label, $W_h$ the parameters of sub-model $F_h$, $a_h$ the activation at the output of $F_h$, and $\hat y = a_H$ the network output, then forward propagation is
$$
a_0 = x,\qquad
a_h = F_h(a_{h-1};W_h),\; h=1,\dots,H,\qquad
\hat y = a_H,
$$
with global loss
$$
L(W;x,y)=\ell(a_H,y).
$$
Backward propagation follows the chain rule:
$$
\delta_H = \nabla_{a_H}\ell(a_H,y),\qquad
\delta_h = \left(\frac{\partial F_{h+1}}{\partial a_h}\right)^T \delta_{h+1},\; h=H-1,\dots,1,
$$
and
$$
\nabla_{W_h}L = \left(\frac{\partial F_h}{\partial W_h}\right)^T \delta_h.
$$
Under this model, each hop $h$ receives $a_{h-1}$, computes $a_h$ in the forward pass, then receives $\delta_h$ in the backward pass, computes $\nabla_{W_h}$, and passes $\delta_{h-1}$ upstream [2509.10001].

Related formulations use different symbols but the same sequential dependency structure. MP-SL describes a deep network split into $P$ model-segments
$$
M_1 \rightarrow M_2 \rightarrow \dots \rightarrow M_P,
$$
where the data owner hosts both $M_1$ and $M_P$, and intermediate compute nodes host $M_2,\dots,M_{P-1}$; activations at cut layer $\ell_p$ are denoted $A_k^{(\ell_p)}$ and gradients $\partial L/\partial A^{(\ell_p)}$ [2402.00208]. A pipelined edge-network formulation instead assumes an $I$-layer model split into up to $K$ submodels with binary cut variables $x_{ik}$ and placement variables $y_{kn}$, together with a micro-batch size $b \in \{1,\dots,B\}$ [2505.04368]. This suggests that “multi-hop” is not tied to a single implementation pattern: it is the broader condition that training-state transfer is distributed across more than one inter-node boundary.

## 2. SFC-based architecture and neural service functions

In the SFC-based architecture, each sub-model $F_h$ is encapsulated in a Neural Service Function (NSF), implemented as a Linux user-space process acting as a transparent TCP proxy over SRv6. The stated components are SRv6 encapsulation/decapsulation through an eBPF-based SFC proxy, TPROXY rules for intercepting end-to-end TCP flows, buffering logic to reassemble byte streams from packets, and libtorch sub-model execution [2509.10001].

The forward packet flow is defined procedurally. The client sends an original TCP SYN to the SR source node, which encapsulates it in an outer IPv6+SRH header whose segment list contains the $H-1$ NSFs and the server. At each NSF, eBPF at TC ingress matches the local SID, decapsulates the outer header, stores the $5$-tuple $\rightarrow$ (outer IPv6+SRH) mapping in a BPF map, and intercepts the flow via TPROXY into NSF user space. The NSF buffers TCP segments, reassembles the byte stream into a tensor $x_h$, runs $a_h = F_h(x_h;W_h)$, retrieves the stored SRH from the BPF map keyed by the new $5$-tuple, re-encapsulates $a_h$ into SRv6 using the outer IPv6/SRH, and writes TCP segments to the next NSF. The backward pass reverses the segment list in the SRH; each NSF performs the eBPF decapsulation, TProxy interception, gradient computation, and forwarding of $\delta_{h-1}$ upstream [2509.10001].

This architecture is explicitly designed to maintain compatibility with existing applications while enabling dynamic routing of smashed data. In the later optimization framework, the same SFC intuition is elevated into an augmented graph $G^+=(V\cup \hat V,E\cup \hat E)$ by introducing one imaginary node $\hat v_k$ per sub-model and linking each $\hat v_k$ to its candidate host nodes $V^k$; the ordered execution $F^1 \rightarrow F^2 \rightarrow \dots \rightarrow F^K$ is treated as the service chain itself [2604.25197]. A plausible implication is that the SFC view turns MSL from a purely model-parallel training protocol into a network-native service orchestration problem.

## 3. Training protocols, communication, and pipelining

For SFC-based MSL, training is given per mini-batch $\{(x_i,y_i)\}_{i=1}^b$. The forward pass initializes $A_0 \leftarrow X$ and sequentially invokes
$$
A_h \leftarrow \mathrm{NSF}_h.\mathrm{forward}(A_{h-1};W_h),\qquad h=1,\dots,H.
$$
At the server, the loss gradient is computed as $\delta_H=\partial \ell(A_H,Y)/\partial A_H$. The backward pass then iterates from $h=H$ to $1$:
$$
(\nabla_{W_h},\delta_{h-1}) \leftarrow \mathrm{NSF}_h.\mathrm{backward}(\delta_h),\qquad
W_h \leftarrow W_h - \eta \nabla_{W_h}.
$$
The communication-cost model for forward activations of size $s_h$ bytes is
$$
T_{\mathrm{comm}}^{\mathrm{fwd},h}=t_{\mathrm{ser}}(s_h)+\frac{s_h}{B_e},
$$
with backward gradients of the same order of magnitude. Per-hop computation time is denoted $T_{\mathrm{comp}}^{\mathrm{fwd},h}(b)$ and $T_{\mathrm{comp}}^{\mathrm{bwd},h}(b)$, and overall per-batch training latency is
$$
T_{\mathrm{batch}}=
\sum_{h=1}^H \left[T_{\mathrm{comp}}^{\mathrm{fwd},h}+T_{\mathrm{comm}}^{\mathrm{fwd},h}\right]
+
\sum_{h=1}^H \left[T_{\mathrm{comp}}^{\mathrm{bwd},h}+T_{\mathrm{comm}}^{\mathrm{bwd},h}\right]
+
T_{\mathrm{wait}},
$$
where $T_{\mathrm{wait}}$ is the idle time an NSF spends waiting for upstream gradients [2509.10001].

A distinct line of work treats MSL as a pipeline. MP-SL uses asynchronous non-blocking send/receive threads so that while one mini-batch is computing on hop $p$, the next batch can be transmitted from $p-1$. After an initial fill-up phase of $P-1$ forward steps, the pipeline runs full and each mini-batch completes in
$$
T_{\mathrm{batch}}^{\mathrm{pipefull}} = L_{\mathrm{fwd}} + L_{\mathrm{back}},
$$
where
$$
L_{\mathrm{fwd}} = \max_{p=2,\dots,P-1} \mathrm{procT}_p^{\mathrm{fwd}}(n_p),\qquad
L_{\mathrm{back}} = \max_{p=2,\dots,P-1} \mathrm{procT}_p^{\mathrm{back}}(n_p).
$$
Warm-up and unwind incur extra sums of per-hop forward and backward processing times [2402.00208].

The edge-network pipelining framework makes the micro-batch explicit. A mini-batch of $B$ samples is divided into micro-batches of size $b$, with forward and backward passes overlapping across micro-batches. The total time to train one mini-batch is
$$
L_t(x,y,b)=T_f(x,y,b)+\left\lceil \frac{B-b}{b}\right\rceil T_i(x,y,b),
$$
where $T_f$ is the latency of the first micro-batch and $T_i$ is the per-micro-batch pipeline latency, defined as the bottleneck over all compute and communication stages [2505.04368]. Taken together, these formulations separate two major execution regimes in MSL: synchronous hop-by-hop training, where waiting time is prominent, and pipelined training, where bottleneck-stage latency and fill/unwind overhead become central.

## 4. Model splitting, placement, and chaining optimization

Once MSL is distributed across multiple hops, three coupled design decisions arise repeatedly: where to cut the model, where to place each sub-model, and how to route smashed data through the network. The SFC optimization work formulates these choices as ILPs for inference and training. With routing and placement variables
$$
x^k_{i,j}\in\{0,1\}
$$
and model-splitting variables
$$
y_{\hat v_k,l}\in\{0,1\},
$$
the objective minimizes total latency
$$
\min_{x,y}\; T(x,y,b,\mathrm{FW}) + \mathbb{I}(\mathrm{mode}=\mathrm{TR})\,T(x,y,b,\mathrm{BW}),
$$
subject to flow conservation, connectivity between successive subpaths, exact layer coverage, nonempty submodels, contiguity of layers, ordering across submodels, and storage and memory capacity constraints [2604.25197].

The same paper proposes a Block Coordinate Descent heuristic that alternates between optimizing the split and optimizing routing/placement. Model splitting is solved via a $K$-sequence segmentation dynamic program with
$$
dp_{k,l} = \min_{l'<l}\{dp_{k-1,l'} + T^k(l',l)\},
$$
in $O(KL^2)$, while placement and chaining are solved via a depth-first tour search on a modified augmented graph. The reported total complexity is approximately
$$
O(T_{\max}\cdot K\cdot L^2 + (T_{\max}+1)(K+1)(V^+ + E^+\log V^+)),
$$
and global optimality is not guaranteed because the procedure is a non-convex BCD on an ILP [2604.25197].

A graph-theoretic optimization appears in the pipelined edge-network setting. There, for fixed $b$, the model splitting and placement subproblem maps to minimizing a weighted sum of a bottleneck cost and a linear cost:
$$
\mathcal{P}4:\quad
\min_{x,y,T_1}\; T_f(x,y)+\xi(b)T_1
\quad\text{s.t.}\quad
T_i(x,y)\le T_1,
$$
with $\xi(b)=\left\lceil (B-b)/b\right\rceil$. A bottleneck-aware shortest-path algorithm enumerates candidate bottleneck edges, solves a shortest-path problem on the corresponding subgraph, and returns the globally optimal solution of $\mathcal{P}4$ by Minoux’s min-max/min-sum theory. Its time complexity is
$$
O\bigl(E(V+E)\log V\bigr)
$$
for graph sizes $V$ and $E$ [2505.04368].

MP-SL adds yet another optimization layer: an ILP in which decision variables $x_{ij}\in\{0,1\}$ indicate whether layer $j$ is assigned to compute node $i$, under memory and contiguity constraints, so as to minimize the maximum per-hop processing time [2402.00208]. This suggests that optimization in MSL is not monolithic. Depending on the system model, the dominant objective may be bottleneck equalization, end-to-end latency minimization, memory balancing, or network-path selection.

## 5. Empirical behavior and system-level trade-offs

The SFC-based architecture is evaluated on hardware consisting of an Intel i9-10980XE with 36 cores, an NVIDIA RTX A6000, and 128 GB RAM, using a Mininet emulation with NSFs at $v6$ and $v7$, an SR source at $v2$, a server at $v8$, and 1 Gbps links with jumbo frames enabled. The model is a ResNet-101-$\tfrac14$ width network with $L=37$ layers split into $H=4$ sub-models $(L_1=2,L_2=8,L_3=9,L_4=18)$, trained on CIFAR-100 with SGD and learning-rate schedule $\eta=0.1 \rightarrow 0.02 \rightarrow 0.004 \rightarrow 0.0008$, momentum $0.9$. For inference with mini-batch $b=1$, the proposed SFC-MSI records $38.7$ ms/round versus $39.2$ ms/round for a baseline TCP-proxy MSI, supporting the claim that SRv6 overhead is much smaller than compute time and is effective for real-time $b=1$. For training with $b=128$, per-round latency is approximately $173$ ms for both SFC-MSL and the baseline; total training for $200$ epochs is approximately $13\,000$ s with final test accuracy $71.1\%$. Under simulated 1 Mbps congestion on one link, the baseline latency spikes from $40 \rightarrow 100$ ms in inference and $170 \rightarrow 1000$ ms in training until congestion is removed, whereas SFC-MSL detects congestion with a $10$ s window, installs a detour, and recovers to baseline performance within approximately $20$ s [2509.10001].

MP-SL emphasizes memory footprint and heterogeneous-device robustness. On a Raspberry Pi 4, full-model FL uses approximately $80\%$ of RAM for VGG-19, while MP-SL reduces data-owner memory by up to $76\%$. With two compute nodes, MP-SL improves over one-hop split learning by $8.5\%$ for ResNet and $19\%$ for VGG when split points are optimizer-chosen rather than naive. Under data-owner heterogeneity with a mix of Pi 4 and Pi 3 devices, MP-SL’s epoch time varies by less than $1\%$, whereas SplitNN varies by up to $30\%$. Pipeline steady-state throughput improves roughly by a factor of approximately $1/(P-2)$ if hops are perfectly balanced, and accuracy is unchanged because the functional DNN is identical to the one trained centrally [2402.00208].

The pipelined edge-network study reports that pipeline parallelism cuts total latency by $3\times$ to $7\times$ over no-pipeline split learning with identical final accuracy. It also reports that optimal model splitting and placement plus batching differs from a suboptimal BCD by less than $2\%$ in latency, while BCD runs $50\times$ faster when $N=10$ [2505.04368]. In the SFC optimization study, ILP-optimal inference and training latencies indicate a consistent trade-off: for small batch sizes $(b\le 2)$, the optimum is at $K=2$, whereas for larger $b$ or training, the optimum shifts to $K=3$; beyond $K=3$, added communication delays dominate and total latency rises. BCD remains within $2\%$ of ILP across all $(b,K)$ and is roughly two orders of magnitude faster than ILP in the larger network instances evaluated [2604.25197].

A common misconception is that adding more hops necessarily improves performance. The reported results point to a more constrained conclusion: multi-hop partitioning can reduce memory pressure and computation bottlenecks, but beyond the system-dependent “sweet spot,” added smashed-data transmission and synchronization costs dominate [2604.25197].

## 6. Privacy, limitations, and open directions

Privacy is not inherent to MSL, but it is an active design axis. In MP-SL, multihop splitting with at least two cuts is described as raising the bar against label-inference attacks since no single party sees end-to-end activations, while existing defenses such as NoPeek are said to complement MP-SL [2402.00208]. A more explicit privacy model appears in deceptive signal-assisted private multi-hop split learning, where a global model $\boldsymbol{\Theta}_G$ is partitioned into $S$ sequential sub-models $\theta_1,\dots,\theta_S$, hosted on selected edge devices and the server, while subsets of nontraining devices concurrently emit jamming signals whenever genuine sub-model outputs or gradients are transmitted. The optimization objective is to minimize expected information leakage to eavesdroppers subject to delay and energy constraints, jointly over the training-device set $\mathcal{S}$, deceptive-jammer sets $\{\mathcal{D}_{i,j}\}$, sub-model splits $\{\theta_k\}$, and power levels $\mathbf{p}$ [2507.07323].

Because eavesdropper positions and monitoring probabilities are unknown, that work adopts a model-free reinforcement-learning solution: Soft Actor-Critic augmented with an Intrinsic Curiosity Module and a Cross-Attention mechanism. The state is
$$
s(n)=\bigl[E_R(n),\,T_R(n),\,\Theta'_G(n),\,r(n),\,v(n),\,\ell_M(n),\,\ell_D(n)\bigr],
$$
the action is
$$
a(n)=\bigl[u(n),\,\theta_n,\,\mathcal D(n),\,p(n),\,x(n)\bigr],
$$
and the reward is
$$
R(s,a)=R_E(s,a)+\zeta R_C(s,a).
$$
Simulation results report convergence up to $3\times$ faster and information leakage reduced by up to $13\%$ versus traditional SAC, with additional reductions of up to $18\%$ versus SAC when the number of eavesdroppers is $E=4$ [2507.07323]. This suggests that privacy-preserving MSL can be framed not only through cryptographic or statistical protections but also through device-role assignment, physical-layer signaling, and adaptive model partitioning.

Several limitations recur across the literature. In the SFC-based implementation, training overhead is dominated by inter-hop waiting time as $H$ grows, motivating asynchronous variants and stale-gradient MSL [2509.10001]. In pipelined systems, initial warm-up and final unwind remain unresolved overheads [2402.00208]. More hops increase point-to-point transfers even when communication overlaps with compute [2505.04368]. Joint optimization formulations remain computationally difficult at scale, which is why heuristic methods such as BCD are used despite the absence of global optimality guarantees [2604.25197].

Future directions are correspondingly consistent across papers: compression of smashed data to reduce communication cost; asynchronous or pipelined MSL to better utilize compute resources; adaptive splitting under dynamic node availability and workload shifts; incorporation of secure aggregation or differential privacy; multi-path chaining to load-balance smashed-data traffic; privacy-aware split-point selection; and online adaptation that combines decomposition-based optimization with reinforcement learning under varying batch sizes, link states, and node loads [2509.10001][2402.00208][2505.04368][2604.25197]. Collectively, these directions indicate that MSL has evolved from a simple extension of two-party split learning into a broader research area at the intersection of distributed training, network service chaining, systems optimization, and privacy engineering.

Source: https://www.emergentmind.com/topics/multi-hop-split-learning-msl