Search papers, labs, and topics across Lattice.
This paper investigates a selective escalation approach in vertical federated learning (VFL) to optimize communication and computational efficiency during collaborative inference. By implementing a two-round inference protocol, the authors propose a method that only triggers an additional embedding-fusion round when the expected gain in prediction accuracy justifies the communication cost. Their results demonstrate that this expected-gain score estimation method significantly enhances the communication-accuracy trade-off compared to existing baseline strategies.
Selective escalation in VFL can dramatically reduce unnecessary communication while boosting predictive accuracy, outperforming traditional methods.
Collaborative inference can improve predictive performance by integrating complementary information across agents, but applying collaborative fusion to every sample can incur unnecessary communication and computational overhead. This trade-off is particularly relevant in vertical federated learning (VFL), where clients observe different views of the same sample and fusion typically requires transmitting intermediate representations to a server. We study selective escalation in a two-round VFL inference protocol, in which a low-cost first round produces a prediction from client posteriors and a second embedding-fusion round is invoked only when it is expected to improve the final decision. We formulate routing as expected-gain score estimation: a sample is escalated when a predicted improvement in correctness justifies the additional communication. The proposed analytical score combines a calibrated pooled posterior with classwise reliability estimates of the VFL model, both obtained from held-out calibration data, yielding an interpretable router that requires no separately trained routing network. Experiments on multi-view classification benchmarks, including controlled test--time view degradation settings, show that the proposed router improves the communication-accuracy trade-off over confidence-, learned-gain-, and deferral-based baselines.
