Vertical data partition is a dataset split where different features (columns) describing the same set of entities (rows) are held by separate, non-overlapping parties. This structure is the foundational data architecture for Vertical Federated Learning (VFL), enabling collaborative model training across organizations—such as a bank and an e-commerce platform sharing insights on mutual customers—without any party exposing its raw, proprietary feature data. The critical prerequisite for this collaboration is entity alignment, which uses protocols like Private Set Intersection (PSI) to securely identify the overlapping samples across the distributed datasets.
Glossary
Vertical Data Partition

What is Vertical Data Partition?
A vertical data partition is the specific arrangement of data that enables collaborative machine learning without sharing raw data between organizations.
In this paradigm, one party typically holds the label space (the target variable, like a loan default flag), while others are feature owners possessing complementary attributes (like transaction history or browsing behavior). Training a model requires a split neural network architecture, where computations are divided at a cut layer. Feature owners compute intermediate outputs from their local data and share these encrypted results to allow the label owner to complete the forward pass and coordinate vertical backpropagation. This structure directly creates the core challenges of VFL: managing vertical communication overhead from exchanging intermediate results and ensuring privacy-preserving computation through cryptographic techniques.
Key Characteristics of a Vertical Partition
A vertical data partition is the foundational data arrangement for Vertical Federated Learning (VFL). It is defined by how features (columns) and samples (rows) are distributed across different, mutually distrusting data owners.
Feature-Based Splitting
A vertical partition splits a dataset by features (columns), not by samples. Different parties hold different attributes for the same set of entities (rows). For example, a bank holds a customer's credit history (features F1-F3), while a retailer holds the same customer's purchase history (features F4-F6). The complete feature vector for any entity is distributed across multiple silos.
Sample ID Alignment Requirement
For VFL to be possible, the parties must share a common set of entity identifiers (e.g., user IDs, device IDs). The core challenge is performing entity alignment—finding these overlapping samples—without exposing the full list of IDs each party holds. This is typically solved using cryptographic protocols like Private Set Intersection (PSI). Only data for the aligned intersection is used for training.
Asymmetric Role Distribution
Roles in a vertical partition are inherently asymmetric:
- Label Owner: Typically one party holds the target variable (e.g., loan default flag, purchase prediction). This party often acts as the computation coordinator.
- Feature Owners: One or more other parties hold subsets of the input features but lack the labels. They collaborate to provide predictive features for the aligned samples. This asymmetry dictates the training protocol and flow of information.
Statistical Heterogeneity (by Design)
Unlike horizontal FL, where data heterogeneity is a challenge, vertical partitions are inherently statistically heterogeneous by design. Each party's feature subset comes from a different domain, distribution, and semantic meaning (e.g., financial vs. behavioral features). The learning objective is to fuse these complementary feature spaces to build a more powerful model than any single party could create alone.
Foundation for Split Model Architectures
The vertical data structure directly informs the model architecture. The standard approach is a split neural network, where the model is divided at a cut layer. Feature owners compute the forward pass on their local features up to the cut layer, producing encrypted intermediate outputs. These are sent to the label owner, which completes the forward pass, calculates the loss, and coordinates backpropagation. The model's architecture mirrors the data partition.
High Privacy Sensitivity
Because each party holds a different view of the same entities, even sharing intermediate results (like embeddings or gradients) can risk leaking raw feature information through inference attacks. This necessitates strong privacy-preserving techniques as a core characteristic, including:
- Secure Multi-Party Computation (MPC) for joint calculations.
- Homomorphic Encryption (HE) for computing on ciphertext.
- Differential Privacy (DP) to add statistical noise to shared values.
How Vertical Partitioning Enables Federated Learning
Vertical data partitioning is the foundational data structure that makes Vertical Federated Learning (VFL) possible, enabling collaborative model training across organizations that hold different attributes about the same entities.
A vertical data partition is a dataset split where different features (columns) for the same set of samples (rows) are held by separate, non-colluding parties. This structure is the core enabler for Vertical Federated Learning (VFL), a paradigm where entities like a bank and a retailer can jointly train a model—the bank holds credit history, the retailer holds purchase history—without ever pooling their raw, sensitive customer data. The shared entity keys (e.g., customer IDs) must first be aligned using a privacy-preserving protocol like Private Set Intersection (PSI) to establish the common sample set for training.
During VFL training, a split neural network architecture is used, with a cut layer dividing the model between parties. Each party computes forward passes on its local features, producing intermediate outputs that are securely shared to continue computation, typically with the label owner. Gradients are then propagated back through the network in a coordinated vertical backpropagation process. This partitioned computation, secured via techniques like homomorphic encryption or secure multi-party computation, allows the collective feature space to be leveraged for a powerful global model while maintaining strict data isolation and privacy for each participant.
Vertical vs. Horizontal Data Partitioning
This table contrasts the two fundamental data partitioning schemes that define different federated learning paradigms, based on how samples and features are distributed across parties.
| Characteristic | Vertical Partitioning | Horizontal Partitioning |
|---|---|---|
Core Partitioning Logic | Features (columns) are split across parties | Samples (rows) are split across parties |
Sample Overlap | Parties share the same set of entities (after alignment) | Each party holds different, non-overlapping entities |
Feature Overlap | Each party holds a unique, disjoint subset of features | All parties share the same feature set |
Typical Federated Paradigm | Vertical Federated Learning (VFL) | Horizontal Federated Learning (HFL) |
Primary Use Case | Collaboration between organizations with different data modalities about the same users (e.g., bank & retailer) | Collaboration across many devices/organizations with similar data schemas (e.g., smartphones, hospitals) |
Model Architecture | Split neural network; model is divided across parties | Single, shared global model; each client has a full copy |
Privacy Challenge | Secure entity alignment and leakage from intermediate outputs | Information leakage from shared model updates (gradients) |
Cryptographic Prerequisites | Private Set Intersection (PSI) for sample alignment | Secure Aggregation for update combination |
Communication Pattern | Point-to-point or star topology between heterogeneous parties | Star topology between a server and many similar clients |
Example Scenario | A bank (credit history) and an e-commerce site (purchase history) jointly train a fraud model on their shared customers. | Thousands of smartphones collaboratively train a next-word prediction model using their local typing history. |
Common Use Cases for Vertical Data Partitions
Vertical data partitions are the foundational structure enabling collaborative machine learning across organizations that hold different attributes about the same entities. These use cases highlight scenarios where data privacy, regulatory compliance, and feature exclusivity are paramount.
Cross-Industry Credit Scoring
A bank holds core financial transaction history and credit labels (loan default), while an e-commerce platform possesses detailed purchase behavior and product affinity data. A vertical partition allows them to train a more robust default prediction model without the bank seeing purchase data or the e-commerce platform accessing sensitive financial records.
- Key Driver: Regulatory compliance (e.g., GDPR, financial privacy laws) prevents raw data sharing.
- Technical Approach: The bank acts as the label owner, and both parties use Private Set Intersection (PSI) to align common customers before training a split neural network.
Healthcare Diagnostics & Research
A hospital holds labeled medical imagery (e.g., MRI scans with diagnoses) and genomic data, while a pharmaceutical research firm possesses detailed molecular compound libraries and assay results. A vertical partition enables the joint development of a diagnostic model linking drug efficacy to patient biomarkers.
- Key Driver: Patient privacy laws (HIPAA, GDPR) and intellectual property protection for proprietary compounds.
- Technical Approach: Homomorphic encryption or secure multi-party computation (MPC) is used to compute on encrypted intermediate outputs, ensuring neither raw patient data nor molecular structures are exposed.
Personalized Retail & Advertising
A retailer holds in-store and online purchase history (labels: customer lifetime value), while a social media platform has rich user engagement, demographic, and interest graph data. They collaborate to build a hyper-personalized recommendation or churn prediction model.
- Key Driver: Competitive data is a core asset; neither party wants to enrich a competitor's central database.
- Technical Approach: A vertical training protocol coordinates the exchange of embeddings or gradients from the social platform's model portion to update the retailer's overall model, preserving user anonymity.
Smart City & IoT Sensor Fusion
A municipal traffic department holds traffic camera feeds (labels: congestion incidents), while a telecom operator possesses anonymized cell phone location pings showing population density and movement. A vertical partition allows for real-time traffic flow prediction and incident detection.
- Key Driver: Data sovereignty and contractual agreements prevent the sharing of raw telecom user location data with government entities.
- Technical Approach: Federated feature selection identifies the most predictive location-derived features, which are then used as input to a vertically split model hosted partly on the telecom's secure edge servers.
Fraud Detection in Financial Networks
A payment processor holds transaction amounts, times, and merchant codes (labels: fraudulent transactions), while a network of issuing banks possesses individual cardholder spending patterns and account histories. A vertical partition enables a consortium to build a superior fraud model that sees a holistic view of behavior without any single entity pooling all data.
- Key Driver: Competitive secrecy between banks and the need to minimize data breach risk by avoiding a central repository of sensitive financial data.
- Technical Approach: Vertical secure aggregation cryptographically combines gradient updates from each bank, and differential privacy noise is added to provide formal privacy guarantees against inference attacks.
Manufacturing Supply Chain Optimization
An automotive OEM holds assembly line sensor data and final quality control labels (defect/no defect). A parts supplier holds detailed sensor data from its own manufacturing process for the supplied component. A vertical partition allows them to jointly model the propagation of sub-component defects into final assembly failures.
- Key Driver: Protection of proprietary manufacturing processes and sensor configurations, which are trade secrets.
- Technical Approach: The supplier and OEM align components by serial number using secure entity resolution. They then employ a vertical inference protocol to run predictions without exposing the supplier's real-time process data to the OEM.
Frequently Asked Questions
A vertical data partition is the foundational data structure for Vertical Federated Learning (VFL). This FAQ addresses common technical questions about how data is split, aligned, and used in this privacy-preserving machine learning paradigm.
A vertical data partition is a dataset split where different features (columns) of the same samples (rows) are held by different, mutually distrusting parties. It forms the core data structure for Vertical Federated Learning (VFL), enabling collaborative model training without sharing raw, sensitive feature data.
For example, in a financial and retail collaboration, Bank A may hold credit history and income (features F1, F2) for a set of customers, while Retailer B holds purchase history and browsing data (features F3, F4) for the same customers. The rows (customer IDs) align, but the columns are partitioned. A third party, like a credit bureau, might hold the labels (e.g., loan default status). The goal of VFL is to train a model that uses Bank A's features and Retailer B's features to predict the credit bureau's labels, without any party exposing its raw data table.
Enabling Efficiency, Speed & Accuracy
Intelligent Analysis, Decision & Execution
We build AI systems for teams that need search across company data, workflow automation across tools, or AI features inside products and internal software.
Talk to Us
Search across company data
Give teams answers from docs, tickets, runbooks, and product data with sources and permissions.
Useful when people spend too long searching or get different answers from different systems.

Automate internal workflows
Use AI to route work, draft outputs, trigger actions, and keep approvals and logs in place.
Useful when repetitive work moves across multiple tools and teams.

Add AI to products and internal tools
Build assistants, guided actions, or decision support into the software your team or customers already use.
Useful when AI needs to be part of the product, not a separate tool.
Related Terms
These terms define the core components, protocols, and challenges of training machine learning models on vertically partitioned data across multiple parties.
Entity Alignment
The foundational process of identifying and matching the same real-world entities (e.g., customer IDs, device serial numbers) across the vertically partitioned datasets held by different parties. This must be done without revealing the full datasets.
- Purpose: Creates the aligned sample space required for model training.
- Methods: Often employs cryptographic protocols like Private Set Intersection (PSI).
- Challenge: A prerequisite for VFL; misalignment leads to training on incorrect data pairs.
Split Neural Network
The model architecture used in VFL where the neural network is divided into multiple segments, each residing on a different party that holds a specific subset of the features.
- Structure: The label owner typically holds the deeper layers near the output, while feature owners hold the initial layers.
- Cut Layer: The specific layer where the model is split. The output of this layer (the intermediate output) is shared to continue computation.
- Example: Feature Owner A computes layers 1-3 on its features, sends the intermediate result to the Label Owner, who computes layers 4-5 to produce the final prediction.
Secure Aggregation
A cryptographic protocol for combining model updates (e.g., gradients or intermediate outputs) from multiple parties without revealing any individual party's contribution. This protects the privacy of each participant's data.
- Core Idea: Uses techniques like secret sharing or homomorphic encryption so that only the aggregated sum is revealed to the coordinator.
- Benefit: Prevents the label owner or other parties from inferring a feature owner's raw data from their update.
- Vertical Context: Applied to gradients from feature owners' model segments or to encrypted intermediate outputs.
Private Set Intersection (PSI)
A cryptographic protocol that allows two or more parties, each holding a private set of items (e.g., user IDs), to compute the intersection of their sets without revealing any information about items not in the intersection.
- Primary Use in VFL: The standard method for performing entity alignment in a privacy-preserving manner.
- Output: All parties learn only which IDs they have in common, and nothing else.
- Variants: Includes PSI-CA (Cardinality) where parties learn only the size of the intersection.
Homomorphic Encryption
An encryption scheme that allows specific types of computations to be performed directly on ciphertext, generating an encrypted result which, when decrypted, matches the result of the operations as if they had been performed on the plaintext.
- Use in VFL: Enables the label owner to perform computations on encrypted intermediate outputs from feature owners.
- Benefit: The label owner never sees the raw intermediate values, enhancing privacy.
- Trade-off: Introduces significant computation overhead due to the complexity of encrypted arithmetic.
Vertical Training Protocol
The defined sequence of communication and computation steps that coordinating parties follow to execute a complete training round in a VFL system.
- Typical Steps: 1) Entity Alignment via PSI. 2) Vertical Forward Propagation: Feature owners compute to the cut layer and send outputs. 3) Loss & Gradient Calculation by the label owner. 4) Vertical Backpropagation: Gradients are sent back to feature owners. 5) Local Model Update by each party.
- Governance: Specifies roles, data formats, encryption methods, and aggregation rules.
- Framework Implementation: Codified in VFL frameworks like FATE or TensorFlow Federated.

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.
Partnered with leading AI, data, and software stack.
How We Work
Custom AI workflows for your Business
One-fit-all AI don't work for modern businesses. At Inferensys, we aim to understand your business & custom requirements; which we use to define most efficient agentic workflows, the data, and the tools for your business.
01
Review the use case
We understand the task, the users, and where AI can actually help.
Read more02
Pick the right approach
We define what needs search, automation, or product integration.
Read more03
Build the first useful version
We implement the part that proves the value first.
Read more04
Improve from there
We add the checks and visibility needed to keep it useful.
Read moreThe first call is a practical review of your use case and the right next step.
Talk to Us