Skip to content

Optional Extension: Secure Entity Alignment for Vertical FL

The main path in this guide assumes horizontal FL, where participants have similar feature spaces but different records. Some use cases require vertical FL, where participants hold different features about overlapping entities. In those cases, the parties must first determine which records overlap without revealing their entire datasets to one another.

Secure entity alignment flow for VFL

Figure 6. Example secure entity alignment flow for VFL. Parties align entities through PSI and train only over the aligned set.

QuestionGuidance
When is entity alignment needed?When participants need to train on overlapping entities but hold different feature sets.
What identifiers may be used?Use the minimum necessary identifiers and apply identifier hygiene before matching. Avoid casual use of direct identifiers where safer alternatives exist.
Should this be in the main tutorial?Only if the repository has a working and tested PSI module. Otherwise, present it as an advanced extension.
  • Use only identifiers approved for the matching purpose.

  • Normalise identifiers consistently before matching, for example trimming, casing, format checks, or salt handling where applicable.

  • Avoid retaining raw identifiers in logs or exported artefacts.

  • Record the approved purpose, participants, matching date, and match-count outputs.

Private Set Intersection (PSI) is a common approach for privacy-preserving entity alignment. The exact protocol should be selected based on the repository implementation, participant capabilities, identifier sensitivity, and performance requirements.

StepOperator action
Prepare identifiersConfirm approval, normalise locally, and validate format.
Run PSI protocolExecute the approved PSI runner, such as ECDH-PSI or KKRT where supported.
Review match countsCheck whether overlap is sufficient for the VFL task.
Approve aligned setRecord governance decision for using the aligned entity set.
Proceed to VFL trainingTrain only over approved aligned records and log the protocol output.
EvidenceWhy it matters
Protocol and versionSupports reproducibility and security review.
Participants and purposeShows that alignment was performed only for an approved collaboration.
Identifier preparation methodDocuments normalisation and hygiene steps.
Match counts and error conditionsShows whether the aligned set is viable.
Approval decisionShows whether aligned outputs may be used for training.
Failure modeResponse
Low overlapReview whether the VFL use case remains viable or should be redesigned.
Unexpected high overlapCheck identifier preparation and whether matching inputs were correct.
Protocol failureInspect local logs, participant configuration, and version compatibility.
Identifier-quality issueFix local identifier preparation before re-running alignment.
Governance objectionDo not proceed until the participant agreement and approval basis are resolved.