Inferensys

Glossary

Vertical Federated Learning

A federated learning scenario where datasets share overlapping sample spaces but differ in their feature spaces, enabling two organizations holding different attributes about the same users to collaboratively train a model without exposing their respective columns.
Data engineer managing feature store on laptop, feature definitions visible, casual data engineering session.
FEATURE-PARTITIONED COLLABORATION

What is Vertical Federated Learning?

A distributed machine learning paradigm where two or more entities hold different attributes about the same set of overlapping users, enabling collaborative model training without exposing raw feature columns.

Vertical Federated Learning is a privacy-preserving machine learning scenario where participating datasets share an overlapping sample space but differ in their feature spaces. Unlike horizontal federated learning, which distributes data across different users, vertical FL applies when organizations possess disjoint sets of attributes about the same individuals—such as a telecom operator holding browsing data and a bank holding transaction records for identical subscribers. The core technical challenge lies in aligning samples across silos using Private Set Intersection protocols before jointly computing a model without any party revealing its proprietary columns to the others.

The training process typically employs a split neural network architecture with a privacy-preserving entity resolution layer. Each participant maintains its own bottom model to process local features, transmitting only encrypted intermediate embeddings to a neutral third-party server that computes the forward pass and backpropagates gradients. Security is enforced through homomorphic encryption or secure multi-party computation, ensuring that even the aggregator cannot reverse-engineer individual feature values. This architecture is critical for telecom operators collaborating with financial institutions to build fraud detection models while maintaining strict data sovereignty and regulatory compliance.

ARCHITECTURAL PRIMITIVES

Key Characteristics of Vertical Federated Learning

Vertical Federated Learning (VFL) addresses the scenario where two or more institutions hold different features for the same set of entities. Unlike horizontal federated learning, VFL requires secure entity alignment and intermediate computation exchange to jointly train a model without exposing raw feature columns.

01

Entity Alignment via Private Set Intersection

Before training begins, participants must identify overlapping samples without revealing non-overlapping entities. Private Set Intersection (PSI) protocols enable two parties to compute the intersection of their sample IDs cryptographically, ensuring that only the common user base is used for training while keeping exclusive customers confidential.

Zero
Non-Overlapping IDs Exposed
02

Split Neural Network Architecture

In VFL, the model is physically partitioned across participants. Each party maintains a bottom model that processes its local features into intermediate representations called embeddings. These embeddings, not raw data, are exchanged and fed into a top model hosted by a central server or a neutral third party to compute the final prediction.

03

Secure Gradient and Loss Exchange

During backpropagation, the top model computes the loss and sends encrypted gradients back to each participant's bottom model. To prevent gradient leakage attacks that could reconstruct private features, VFL implementations often employ homomorphic encryption or differential privacy to protect the backward pass.

04

Asymmetric Feature Ownership

A defining characteristic of VFL is that one party typically holds the label data (the active party), while collaborators provide complementary features (passive parties). This asymmetry requires specialized loss computation protocols where the label owner securely shares the loss gradient without exposing the ground truth labels to passive participants.

05

Inference with Multi-Party Computation

Post-training inference requires real-time coordination between all parties. When a new user arrives, each participant runs its bottom model locally and encrypts the output. A Secure Multi-Party Computation (SMPC) protocol aggregates these encrypted embeddings to produce the final prediction without any single party seeing the full feature vector.

06

Communication-Computation Trade-off

VFL exchanges intermediate activations and gradients for every batch, creating a synchronous communication bottleneck unlike the asynchronous nature of horizontal FL. Optimizations include quantization of embeddings and reduced precision training to minimize the data transfer overhead between collaborating organizations.

TOPOLOGY COMPARISON

Vertical vs. Horizontal Federated Learning

Structural comparison of the two primary federated learning partitioning strategies based on how data is distributed across participating entities.

FeatureVertical FLHorizontal FLFederated Transfer Learning

Data Partitioning Axis

By feature space (columns)

By sample space (rows)

By both feature and sample space

Overlapping Entity Space

Same users, different attributes

Different users, same attributes

Minimal overlap in users and attributes

Typical Participants

2-10 institutional silos

100s to millions of edge devices

2 distinct organizational domains

Entity Alignment Requirement

Private Set Intersection required

Not required

Limited co-occurrence alignment

Primary Cryptographic Primitive

Secure Multi-Party Computation

Secure Aggregation

Homomorphic Encryption

Communication Pattern

Peer-to-peer or coordinator-mediated

Hub-and-spoke (server-centric)

Coordinator-mediated with domain adaptation

Sample ID Overlap

Feature Space Overlap

VERTICAL FEDERATED LEARNING

Frequently Asked Questions

Clear, technically precise answers to the most common questions about training models across organizations that hold different attributes for the same set of users.

Vertical Federated Learning (VFL) is a privacy-preserving machine learning paradigm where two or more parties hold datasets with overlapping sample spaces but disjoint feature spaces, enabling collaborative model training without exposing raw data columns. Unlike Horizontal Federated Learning where participants share the same schema for different users, VFL applies when organizations possess different attributes about the same set of entities—for example, a telecom operator holding call detail records and a bank holding transaction histories for overlapping subscribers. The process begins with Private Set Intersection (PSI) to cryptographically identify common samples without revealing non-overlapping identities. Training then proceeds using split neural network architectures where each party maintains its own bottom model to process local features, exchanging only intermediate embeddings or encrypted gradients through a neutral coordinator. The core challenge lies in aligning samples across silos and computing loss gradients without any party gaining visibility into another's proprietary feature columns, typically solved through homomorphic encryption or secure multi-party computation protocols during the forward and backward passes.

Prasad Kumkar

About the author

Prasad Kumkar

CEO & MD, Inference Systems

Prasad Kumkar is the CEO & MD of Inference Systems and writes about AI systems architecture, LLM infrastructure, model serving, evaluation, and production deployment. Over 5+ years, he has worked across computer vision models, L5 autonomous vehicle systems, and LLM research, with a focus on taking complex AI ideas into real-world engineering systems.

His work and writing cover AI systems, large language models, AI agents, multimodal systems, autonomous systems, inference optimization, RAG, evaluation, and production AI engineering.