Search papers, labs, and topics across Lattice.
The paper proposes LRD-MPC, a method to accelerate secure multi-party computation (MPC) inference of deep neural networks by applying low-rank decomposition (LRD) to linear layers. To mitigate the overhead of additional communication rounds and truncation steps introduced by LRD, the authors introduce truncation skipping and efficient linear layer concatenation. Experiments demonstrate significant speedups (up to 33% in 3-PC) and energy savings (up to 52% GPU) compared to full-rank baselines, alongside a substantial reduction in offline-phase latency (up to 88%).
Decomposing linear layers in MPC inference can actually speed things up, but only if you carefully manage the extra communication rounds and truncation steps.
Secure Multi-party Computation (MPC) enables untrusted parties to jointly compute a function without revealing their inputs. Its application to machine learning (ML) has gained significant attention, particularly for secure inference services deployed across multiple cloud virtual machines (VMs), where each VM acts as an MPC party. Model providers secret-share model weights, and users secret-share inputs, ensuring that each server operates only on random shares. While MPC provides strong cryptographic guarantees, it incurs substantial computational and communication overhead. Deep neural networks rely heavily on convolutional and fully connected layers, which require costly matrix multiplications in MPC. To reduce this cost, we propose leveraging low-rank decomposition (LRD) for linear layers, replacing one large matrix multiplication with two smaller ones. Each matrix multiplication in MPC incurs a round of communication, meaning decomposing one matrix multiplication into two leads to an additional communication round. Second, the added matrix multiplication requires an additional truncation step to maintain numerical precision. Since truncation itself requires communication and computation, these overheads can offset the gains from decomposition. To address this, we introduce two complementary optimizations: truncation skipping and efficient linear layer concatenation. Truncation skipping removes the extra truncation induced by LRD, while linear layer concatenation pipelines operations to hide the additional communication round. Together, these techniques mitigate the main overheads of LRD in MPC and improve overall efficiency. Our approach is broadly applicable across MPC protocols. Experiments show up to 25% speedup in n-PC and 33% in 3-PC protocols over full-rank baselines, along with up to 52% GPU energy savings and 88% reduction in offline-phase latency.
