MIMO-LP: A Multi-Input Multi-Output Framework for Subgraph-based Link Prediction

Link prediction (LP) is a fundamental problem in graph learning and can be broadly categorized into node-based and subgraph-based approaches. While subgraph-based LP methods often achieve superior predictive performance by exploiting localized structural information, they suffer from efficiency bottlenecks due to the high computational cost of per-query subgraph message-passing during both training and inference. To address this challenge, we propose MIMO-LP, a Multi-Input Multi-Output (MIMO) framework that accelerates subgraph-based LP. Given a batch of query node pairs and their corresponding subgraphs extracted from a shared full graph, MIMO-LP superposes their message-passing processes into a shared latent space while ensuring minimal interference among them. This design enables MIMO-LP to multiplex multiple queries within a single forward pass during both training and inference, substantially reducing redundant message-passing computations in overlapping subgraph regions. Extensive experiments demonstrate that MIMO-LP achieves a 14x-44x speedup over existing one-to-one subgraph-based methods, while maintaining comparable predictive performance. The code for MIMO-LP will be released publicly.