mlsanityData to 14 Oct 2026

Expected Gain-based Escalation in Vertical Federated Learning

arXiv 2606.31331 ↗First posted 30 Jun 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 17 references that are in the map. First posted on arXiv 30 Jun 2026.

Builds on

11 of 17 references.

References · 11

  1. 2017On Calibration of Modern Neural Networkscited by 4,960
  2. 2019Advances and Open Problems in Federated Learningcited by 3,046
  3. 2019Beyond temperature scaling: Obtaining well-calibrated multiclass probabilities with Dirichlet calibrationcited by 364
  4. 2020Consistent Estimators for Learning to Defer to an Expertcited by 271
  5. 2017Predict Responsibly: Improving Fairness and Accuracy by Learning to Defercited by 233
  6. 2021Machine Learning with a Reject Option: A surveycited by 109
  7. 2021Optimal strategies for reject option classifierscited by 61
  8. 2023When Does Confidence-Based Cascade Deferral Suffice?cited by 60
  9. 2022Vertical Federated Learning: A Structured Literature Reviewcited by 17
  10. 2023LESS-VFL: Communication-Efficient Feature Selection for Vertical Federated Learningcited by 14
  11. 2023Communication-Efficient Vertical Federated Learning with Limited Overlapping Samplescited by 13

Built on by

Not cited yet.