Skip to Content
Applied SciencesApplied Sciences
  • Article
  • Open Access

16 April 2026

21 Pages

A Zero Trust Driven Federative Learning Algorithm for Privacy Enhancement

,
and
Department of Electronic and Computer Engineering, Faculty of Engineering and Built Environment, Steve Biko Campus, Durban University of Technology, Durban 4001, South Africa
*
Authors to whom correspondence should be addressed.

Abstract

The proliferation of Enterprise Networks, characterized by heterogeneous devices, distributed data sources, and increasingly sophisticated cyber threats, has exposed the limitations of traditional perimeter-based security models. Guided by the principles of Zero Trust Architecture (ZTA), this paper presents a Zero-Trust (ZT)-Driven Federated Learning Algorithm for Privacy Enhancement (ZT-FL-PE), designed to safeguard model and data confidentiality in decentralized learning environments. By integrating ZTA’s “never trust, always verify” posture with Federated Learning’s (FL) decentralized training paradigm, the proposed framework eliminates the need for centralized data aggregation and significantly reduces the attack surface. The algorithm specifically targets two prominent threats to model privacy: property inference attacks (PIAs) and membership inference attacks (MIAs). We introduce adaptive verification mechanisms and privacy-preserving update transformations that enforce continuous authentication, constrain adversarial behavior, and strengthen resilience against inference-based exploitation. Experimental results demonstrate that ZT-FL-PE substantially enhances privacy protection while maintaining high model accuracy and imposing only low-to-moderate computational overhead, making it a practical and robust solution for modern ZT Enterprise environments.

1. Introduction

The rapid evolution of enterprise networking, driven by cloud adoption, hybrid architecture, remote and hybrid work models, and the widespread use of diverse endpoint devices, has fundamentally transformed organizational IT infrastructures. No longer confined to clearly defined perimeters, modern enterprises must seamlessly integrate cloud-based services, on-premises resources, and remote access from heterogeneous user devices. Under these conditions, traditional perimeter-based security models, which implicitly trust devices located inside the network boundary, have become increasingly inadequate. Consequently, organizations are adopting Zero Trust Architecture (ZTA), a security paradigm that rejects the notion of implicit trust. Instead, ZTA is guided by the principle of “never trust, always verify.” Under this model, no user, device, or application is trusted based on network location; every request is dynamically evaluated using identity, contextual information, and policy-based decisions [1,2].
As illustrated in Figure 1, ZTA deployments comprise several logical components that may reside on-premises or in cloud environments. A central element is the Policy Decision Point (PDP), typically subdivided into the Policy Engine (PE) and the Policy Administrator (PA), which interact on the control plane, while user traffic flows through a separate data plane. The primary objective of ZTA is to strengthen access control and mitigate risks to organizational assets through continuous monitoring and adaptive defense strategies. According to the National Institute of Standards and Technology (NIST) [1,2] ZTA requires the application of dynamic policies as close to critical resources as possible, minimizing exposure of sensitive information such as credentials.
Figure 1. ZTA Core Logical Components.
Policy decisions rely on numerous inputs, e.g., user attributes, role assignments, historical behaviours, and external threat intelligence—all of which are maintained in policy repositories to support real-time, contextual decision-making. However, strict data-sharing restrictions often result in fragmented data silos, hindering the use of centralized artificial intelligence (AI) tools. FL addresses this challenge by enabling model training across multiple entities without requiring raw data sharing; instead, only model updates or parameters are exchanged. In the context of cloud-native deployments, hybrid workforces, and highly dynamic endpoint environments, ZTA offers a compelling approach for securing resources independently of network topology [3]. At the core of ZTA lies the concept of fine-grained, context-aware access control: a logically centralized PDP evaluates each access request based on identity, device posture, behavioural patterns, and external risk indicators. Meanwhile, a separate data plane ensures isolation of user traffic. This separation of control and data flow, combined with continuous verification, strengthens resilience against compromise and lateral movement [4]. However, enforcing dynamic, context-sensitive policies across large, heterogeneous enterprises is operationally demanding. Continuous policy updates, real-time device and user monitoring, and adaptive decision-making impose a significant administrative burden. Manual policy management does not scale, is error-prone, and cannot readily accommodate rapid contextual changes (e.g., device health, behavioural anomalies, or real-time threat alerts). These challenges highlight the need for automated, intelligent policy evaluation—an area where AI emerges as a natural enabler. Yet traditional centralized AI methods, which aggregate raw data to a central server, conflict with ZTA’s principles. ZT deployments intentionally restrict data movement; raw data often resides in isolated silos (on-premises, in the cloud, and on remote devices), and legal or contractual constraints may prohibit cross-boundary data transfers. This complicates the use of centralized machine learning for policy generation, anomaly detection, or risk scoring. FL offers a viable solution by enabling distributed training without raw data exchange. Under FL, participants compute local updates (e.g., gradients or parameter deltas) which are aggregated centrally, preserving data confidentiality while enabling global model training [5]. Furthermore, FL has gained traction in privacy-sensitive domains such as healthcare, finance, and enterprise security environments where data locality, regulatory compliance, and confidentiality are paramount. Yet FL is not inherently secure. A growing body of research demonstrates that FL systems are vulnerable to several privacy-compromising inference attacks, including Membership Inference Attacks (MIA), Property Inference Attacks (PIA), and model inversion attacks, which exploit shared model updates to infer private information about local datasets [5]. For example, even without access to raw data, membership inference attacks can determine whether a specific record was used during training, compromising privacy at the sample level [6]. Property inference attacks can reveal sensitive attributes or aggregate statistics of participants’ data (e.g., demographic distributions), which may never be intended for disclosure [3]. These vulnerabilities pose significant challenges for deploying FL in high-security zero-trust environments, where confidentiality and strict privacy guarantees are essential.
Accordingly, despite the complementary strengths of ZTA and FL, a key challenge would be in designing an FL-based AI mechanism that enhances ZTA while remaining robust against inference-based privacy attacks. Thus, this paper addresses this challenge by introducing a Zero-Trust–Driven Federated Learning Privacy Enhancement Algorithm (ZT-FL-PE). Our approach integrates federated learning into ZTA environments while incorporating privacy-preserving mechanisms such as secure aggregation and update transformations. The proposed system enables enterprises to leverage AI-powered policy enforcement and access control decisions without compromising data privacy or violating zero-trust principles.
In this regard, the contributions of this paper are as follows:
  • Feasibility demonstration: We show that FL-based AI techniques can be realistically deployed within ZTA environments, enabling intelligent and dynamic policy enforcement while preserving data locality and regulatory constraints.
  • An architectural framework: It presents a comprehensive architecture that aligns ZTA components such as PDP, control, and data planes, identity, and credential management with FL workflows, enhancing secure cross-domain collaboration.
  • Privacy Enhancement Algorithm. We propose a ZT-PL-PE algorithm that incorporates privacy-preserving update transformations and secure aggregation mechanisms to mitigate membership and property inference attacks in ZT deployments.
  • Empirical evaluation: We demonstrate, through experiments on representative datasets and adversarial models, that FL-PE significantly reduces susceptibility to inference attacks with minimal degradation to model performance and moderate computational overhead.
The rest of the paper is organised as follows: Section 2 reviews related work, followed by a discussion of membership-related attacks in Section 3. Section 4 details the proposed ZT-FL-PE algorithm and architecture, including its mathematical formalisation and theoretical analysis, while Section 5 provides the performance evaluation and discussion. Section 6 concludes the paper.

3. Membership-Related Attacks in Federated Learning

In an FL setup, each client k ∈ { 1 , … , K } possesses a local dataset D k that is typically insufficient to train a statistically robust model in isolation. Due to privacy and data-governance constraints, clients do not share their raw data with other participants or the central server. Instead, after agreeing on a common model architecture and objective function, each client locally minimizes an empirical risk function expressed as follows:
F k ( ω ) = 1 D k ∑ ( x , y ) ∈ D k l ( ω ; x , y )
This it accomplishes utilizing its private data.
During each communication round, clients perform multiple steps of local stochastic gradient descent (SGD) to compute either local gradients or updated model parameters. These updates are sent to the server, which aggregates them—typically via weighted averaging (as in Federated Averaging (FedAvg)) or gradient-based aggregation (as in Federated SGD (FedSGD)) to obtain a global model update. The server then broadcasts the aggregated model back to the clients for the next round of local optimization.
By reducing communication frequency and operating on decentralized, potentially non-IID data distributions, federated optimization introduces unique convergence challenges. Nevertheless, under standard assumptions such as smoothness and bounded variance, algorithms like FedAvg have been shown to converge to stationary points of the following global objective:
F ( ω ) = ∑ k = 1 K D k D F k ( ω )
The global objective in (2) generally focuses on minimizing the weighted empirical risk across all clients. The aggregated global model is then broadcast back to the clients for subsequent rounds of local optimization, and under standard smoothness and bounded variance assumptions, such federated optimization algorithms are known to converge to stationary points of the global objective while significantly reducing communication overhead compared to centralized training. Despite the absence of raw data sharing, FL systems remain vulnerable to various inference attacks that exploit information exposed through model updates or outputs. These attacks can be broadly categorized into Membership Inference Attacks (MIAs), Property Inference Attacks (PIAs), Model Inversion Attacks (MIs), and Model Extraction (model inference) attacks.
MIAs aim to determine whether a specific data sample was included in a target model’s training set. MIAs can be classified into binary classification–based versus metric-based approaches. Binary classification–based MIAs rely on training an auxiliary attack model to distinguish between members and non-members by capturing differences in the target model’s behavior. Typically, the adversary constructs multiple shadow models that replicate the target model’s architecture and training process. The outputs of these shadow models are labeled according to whether the input belongs to the training dataset, and a binary classifier is trained on these labels to predict membership status. Training such classifiers is often challenging due to limited access to representative data and discrepancies between the shadow and target models.
Metric-based MIAs, in contrast, apply a predefined metric to the target model’s outputs and compare the resulting value against a threshold to infer membership [26]. Common metrics include prediction correctness, prediction loss, prediction confidence, adversarial perturbation, and prediction entropy. Prediction entropy quantifies the uncertainty of the model’s output distribution and is defined as;
H ( x ) = − ∑ i p i ( x ) log p i ( x )
where p i ( x ) denotes the confidence assigned by the target model to class, i for input sample, x .
PIAs, on the other hand, aim to infer sensitive attributes or aggregate properties of training data without accessing the raw data. Early PIAs targeted individual attributes, such as gender, political orientation, or sexual preference, using publicly available information. Later work extended PIAs to infer global properties of the training dataset, including population-level statistics such as age distribution or gender ratios. In a typical PIA scenario, the adversary possesses an auxiliary dataset, partitions it based on the presence or absence of a property, and uses model updates or snapshots to extract features for each subset. An attack model is then trained on this labeled dataset to infer whether the property exists in the target training data.
PIAs can be either passive or active. Passive attacks do not interfere with the training process and rely solely on observing model outputs or updates. Active attacks manipulate the training process, for example, by injecting crafted samples or perturbations—to amplify the distinguishability of the target property. A notable subclass of active PIAs leverages generative models: the adversary samples latent variables z ~ p ( z ) and queries the target generative model to obtain synthetic samples x ~ = G ( z ) , constructs a property classifier on these samples, and defines a function that aggregates classifier outputs to quantify the presence of the property in the underlying training distribution.
Finally, MIs aim to reconstruct sensitive input data from model outputs or parameters. By iteratively optimizing inputs to match target outputs, an adversary can infer features of the training data, such as reconstructing images, text, or other private information. MIs exploit the fact that the global model implicitly encodes information from multiple clients’ datasets.
Model extraction or model inference attacks target the model’s structure or parameters. An adversary may query the model or analyze gradients to replicate its functionality, effectively stealing intellectual property or creating a surrogate model that mimics the original.
In analyzing these attacks, several threat-model assumptions are commonly made:
  • Black-box vs. white-box access: In black-box attacks, the adversary can only query the model and observe outputs, whereas in white-box attacks, full access to model parameters and gradients is assumed.
  • Honest-but-curious server: The server follows the federated learning protocol correctly but may attempt to infer private information from client updates.
  • Malicious clients: Clients may deviate from the protocol to extract information about other clients’ data or the global model.
  • Active vs. passive adversaries observe model updates without influencing training, while active adversaries manipulate training inputs, gradients, or aggregation to improve attack success.
Together, these attacks and threat assumptions illustrate that, despite the privacy-preserving design of federated learning, careful defenses including differential privacy, secure aggregation, and gradient obfuscation—are essential to protect client data and model integrity.

4. ZT-Driven FL Privacy Enhancement Algorithm

The decentralized, collaborative nature of FL enables participating clients to train models locally using their private data, while sharing only model parameters with a centralized server for aggregation and consolidation into a global model. This approach inherently enhances privacy. However, FL-based training systems remain vulnerable to Membership Inference Attacks (MIAs), which attempt to extract sensitive information about the training data from the exchanged models.

4.1. ZT-FL-PE Architecture

In this section, we propose and evaluate a ZT-FL-PE framework algorithm with separate client-side and server-side algorithms, explicitly incorporating key elements such as ZTA controls, Adaptive model training, Model personalization, and model obfuscation (privacy protection). The overall framework structure comprises entities such as clients ( C i ) which are edge devices holding private datasets D i , an FL- server S a FP-PE that aggregates updates and enforces policy, a FP-PE that evaluates trust and access, a PEP that enforces decisions, and finally a Model Registry and Attestation Service that ensures model integrity. The proposed framework is based on key ZT assumptions: that there is no implicit trust in any client or update; that identity, behavior, and model updates are continuously verified; that model access is least privileged; and that all updates are potentially malicious.
Potential threats can take various forms, e.g., an adversary may observe or tamper with model updates or launch inference attacks such as membership, property, or reconstruction. An adversary may submit poisoned or malformed updates, and can also pose as an authenticated but untrusted client.
Summarily, its core features include:
  • Privacy-preserving local updates: Each client applies differential privacy or gradient obfuscation to its local model updates before transmission, reducing the risk of MIAs and property inference attacks.
  • Secure aggregation with zero-trust verification: The server aggregates client updates using cryptographic protocols that prevent information leakage while enforcing verification of contributions without revealing individual client data.
  • Adaptive utility–privacy trade-off: The algorithm dynamically adjusts noise levels and aggregation weights to maintain model performance (utility) while ensuring strong privacy guarantees.
  • Computational efficiency: To minimize overhead, ZT-FL-PE employs selective update strategies, pruning redundant gradients, and exploiting partial aggregation techniques without compromising convergence.
The implementation takes place on the client side to protect both the overall FL model and individual client models. It introduces a privacy-sensitive layer that masks and safeguards model updates before they are sent to the FL server. Consequently, the server only receives an aggregated global model containing the masked version of this layer. When clients receive the global model, they restore their original, unmasked privacy-sensitive layer—preserved from the previous FL round—and integrate it with the global model to create a personalized model for predictions. The goal is to assess how different model layers influence Membership Inference Attacks (MIAs), which aim to identify whether a data sample was included in the training data. Using a trained FL model, we examine predictions on both member and non-member samples, calculate gradients for each layer, and analyze the generalization gap the difference between member and non-member gradients. A larger gap suggests a stronger MIA signal, making membership identification easier.
Figure 2 illustrates the overall algorithm structure, where each client executes the process independently. The ZT-FL-PE privacy-preserving strategy targets the penultimate layer, the privacy-sensitive layer, as it is most susceptible to information leakage [18]. While this layer is used locally for predictions, it is always encrypted before updates are sent to the FL server. As a result, the global model stored by the server never contains the raw privacy-sensitive layer. When clients receive the updated global model, they reinsert their previously stored non-obfuscated version, creating a personalised model that balances privacy and predictive performance. To further elaborate on the key steps to be executed by both the client and server-side algorithms, in Table 1, we define the following notations to be used further:
Figure 2. ZT-FL-PE algorithm structure.
Table 1. Notations [20].
The client-side algorithm (Algorithm 1) is outlined as follows:
The initialisation phase commences with the generation of an Obfuscated update ( Δ i t ~ ) from inputs such as the Global model ( θ t ), local data set ( θ i t ), trust policy ( Π ), and Privacy budget ( ε i t ). This is followed by Zero Trust verification, i.e.
VeryfyIdentity ( C i ) ∧ VerifyDeviceState ( C i )
If the verification fails, the client aborts participation. Next, Model Personalization, in which the splitting of model parameters takes place;
θ t = [ θ g t ′ θ p t ]
followed by the generation of a personalized model:
θ i t = θ g t ⊕ Ρ i ( θ p t )
Adaptive Local Training, which comprises an adaptive learning rate based on trust and data drift:
η i t = η 0 ∙ f ( Τ i t , ∇ L i ) and local optimization θ i t + 1 = θ i t − η i t ∇ L ( D i , θ i t ) follows next.
Computing the update follows next;
Δ i t ~ = θ i t + 1 − θ t
Model Obfuscation is next.
Δ i t ~ = O Δ i t , ε i t
The final step involves secure transmission
S e n d Δ i t ~ → m T L S + a t t e s t a t i o n S
Algorithm 1: ZT-FL-PE algorithm on client’s side (Client C i )
   //Initialization:
    i n p u t _ θ t , D i ` , η 0 , ε i t , Π , P i ( ∙ ) , O ( ∙ ) ; %, Global model, Local dataset, Initial learning rate,
   Privacy budget, Trust policy, Personalization function, Obfuscation mechanism
    o u t p u t _ _ Δ i t ~ ; % Obfuscated model update
   1: //-------- Zero Trust Pre-Access Verification --------
   2:  if NOT VerifyIdentity ( C i ) ∧  VerifyDeviceState ( C i ) then
   3:      Abort participation
   4:  end if
   5: //-------- Model Personalization --------
   6:  Decompose θ t into shared and personalized parameters:
   7:      : θ t ← [ θ g t ′ θ p t ]
   8:  Construct personalized model:
   9:       θ i t ← θ g t ⊕ Ρ i ( θ p t )
   10://-------- Adaptive Local Training --------
   11: Compute local loss gradient:
   12:      g i t ← ∀ L ( D i , θ i t )
   13: Compute adaptive learning rate:
   14:      η i t ← η 0 ∙ A d a p t ( g i t , T r u s t S c o r e ( C i ) )
   15: Update personalized model:
   16:    η i t + 1 ← θ i t − η i t ∙ g i t
   17://-------- Model Update Computation --------
   18: Compute raw model update:
   19:   Δ i t ~ ← θ i t + 1 − θ t
   20://-------- Model Obfuscation (Privacy Enforcement) --------
   21: Clip update norm:
   22:      Δ i t ← C l i p ( Δ i t , C m a x )
   23: Apply obfuscation mechanism:
   24:      Δ i t ~ ← O ( Δ i t , ε i t )
   25://-------- Secure Transmission --------
   26: Transmit Δ i t ~ to server using mTLS and remote attestation
   27: return Δ i t ~
The Seve-side algorithm (Algorithm 2) focuses on ZT-driven secure aggregation and control, and its key sequential steps are as follows:
  • Initialization: It receives obfuscated updates { Δ i t ~ } and in turn updates the global model θ t + 1 .
  • Step I: Continuous Trust Evaluation
For each client:
T i t = g ( h i s t o r y i , ‖ Δ i t ~ ‖ , b e h a v i o u r )
Reject updates if:
T i t < τ
  • Step II: Zero Trust Policy Enforcement
This is subject to controls such as update clipping, Partial aggregation, and increased noise for low-trust clients.
P E P ( Δ i t ~ ) ← P E ( T i t , Π )
  • Step III: Robust Secure Aggregation
In this case, the _Trust-weighted aggregation is as follows:
θ t + 1 = θ t + ∑ i ∈ C t ω i t ∙ Δ i t ~ , ,   where   ω i t = T i t ∑ j T j t
  • Step IV: Global Model Integrity Check
v e r i f y ∗ ( θ t + 1 ) ∧ S i g n ( θ t + 1 )
  • Step V: Controlled Model Release
This will be based on the following: Least-privilege model access, Versioned distribution, and enforced re-authentication for the next round.
  • Step VI: Privacy–Utility–Trust Trade-off
max θ   E U ( θ )   s . t .   ∑ i ε i t ≤ ε max , T i t ≥ τ
Algorithm 2: ZT-FL-PE algorithm on the server’s side.
// I n i t i a l i s a t i o n :
i n p u t _ 00 , C , τ , g ( ⋅ ) , Agg ( ⋅ ) , Π , T ` ;%, initial Global model, client set, Trust threshold,
Trust evaluation function, aggregation function, ZT policy, total round
o u t p u t _ _ θ T , % final global model
1: initialize, 00
2:  for t = 0 to T − 1  do
3:     //-------- Zero Trust Client Admission --------
4:       C t ← 0
5:      for each C i ∈ C  do
6:          if V e r i f y I d e n t i t y ( C i ) & V e r f i y P o l i c y C o m p l i a n c e ( C i , Π )  then
7:              C t ← C t ∪ { C i }
8:          end if
9:      end for
10:      Broadcast θ t t to all C i ∈ C t
11:    //-------- Secure Update Collection --------
12:     Receive obfuscated updates } ~ from C t
13:    //-------- Continuous Trust Evaluation --------
14:     for each received update Δ i t ~  do
15:         T i t ← g ( C i   h i s t o r y , Δ i t ~   s t a t i s t i c s ,   b e h a v i o u r   m e t r i c s )
16:        if  Τ i t < τ  then
17:            Reject Δ i t ~
18:         end if
19:     end for
20:    //-------- Zero Trust Policy Enforcement -----------
21:     for each accepted update Δ i t ~  do
22:          Δ i t ~ ← E n f o r c e P o l i c y ( Δ i t ~ , T i t , Π )
23:     end for
24:    //-------- Robust Secure Aggregation --------
25:      θ t + 1 ← θ t + A g g r ( Δ i t ~ , { T i t } )
26:    //-------- Model Integrity Verification --------
27:     if NOT  V e r i f y M o d e l I n t e g r i t y ( θ t + 1 ) then
28:         Rollback to θ t
29:     end if
30:    //-------- Controlled Model Release --------
31:     Sign θ t + 1
32:     Log all trust and policy decisions
33: end for
34: return θ Τ
The following algorithm (Algorithm 3) is the trust-weighted robust algorithm utlised in Algorithm 2.
Algorithm 3: Trust Weighted Robust Aggregation (used in Algorithm 2)
I n p u t _ { Δ i t } ~ , { Τ i t } % Obfuscated updates, Trust scores
o u t p u t _ Δ t % Aggregated update
1:  Normalize trust weights:
2:       ω i t ← Τ i t / ∑ j Τ j t
3:  Remove outlier updates using a robust filter:
4:       Δ ~ f i l t e r e d ← R o b u s t F i l t e r ( { Δ i t } ~ )
5:  Aggregate updates:
6:       Δ t ← ∑ i ω i t ⋅ Δ ~   f i l t e r e d [ i ]
7:  return Δ t

4.2. Mathematical Formalization and Theoretical Analysis

The core innovation of ZT-FL-PE is the operationalization of the ZTA principle across the FL workflow. This is achieved through a dynamic, multi-factor trust score Τ i t ∈ [ 0 , 1 ] for each client C i at round t , which governs their participation, influence, and the level of privacy obfuscation applied. The score is not static; it evolves based on continuous behavioural telemetry and context. We define the trust evaluation function g ( ⋅ ) (from Algorithm 2, Step 15) as a weighted composite of historical, behavioural, and contextual signals:
Τ i t = λ H ⋅ H i t + λ B ⋅ Β i t + λ C ⋅ C i t
subject to
λ H + λ B + λ C = 1
where
( H i t ) - is a moving average of past contributions, penalizing volatility, and is often referred to as a historical trust. Note that if Update Δ ~ i t − 1 was accepted, then
d H i t = β ⋅ Τ i t − 1 + ( 1 − β ) ⋅ Ι
where β ∈ ( 0 , 1 ) is a decay factor that emphasizes recent behaviour.
Behavioural Trust ( Β i t ) : Measures the statistical “normality” of the client’s update relative to the population, detecting outliers or potential poisoning.
Β i t = exp ( − ‖ Δ ~   i t − μ Δ t ‖ 2 2 2 ⋅ ( σ Δ t ) 2 + ε )
μ Δ t and σ Δ t are the mean and standard deviation of all updates in the round t . This term uses a Radial Basis Function (RBF) to assign high trust to updates close to the consensus and low trust to statistical outliers.
Contextual Trust ( C i t ) : Encodes real-time ZTA signals (device posture, network location, time-of-access).
C i t = s i g m a o i d ( ω 1 ⋅ P o s t u r e S c o r e + ω 2 ⋅ G e o C o m p l i a n c e − ξ )
This transforms binary or categorical policy checks (e.g., “device encrypted”, “on corporate VPN”) into a continuous trust metric. The threshold ξ allows for strict policy enforcement. Dynamic Adjustment Across Learning Phases is accomplished in three phases as follows:
  • Initial Phase ( t = 0 ) : Trust is bootstrapped via strong cryptographic authentication ( C i 0 ) and may rely on initial attestation. Weights λ C are high.
  • Mid-Training Phase: As historical data accumulates, λ H increases, allowing the system to identify consistently reliable or unreliable clients.  Β i t becomes critical for detecting model drift or concept drift in non-IID data, as updates from a client whose local data distribution diverges significantly will appear as outliers.
  • Late Phase/Convergence: The system stabilizes. H i t dominates, rewarding clients with long-term, consistent participation. The adaptive mechanism automatically reduces the verification overhead for high-trust clients (e.g., by sampling their updates for full audit less frequently), while maintaining scrutiny on low-trust clients.
Algorithm Complexity Analysis
The computational and communication overhead introduced by ZT-FL-PE is additive and scales linearly with standard FL operations. Let N be the number of clients, P the number of model parameters, and K the size of the historical window for trust evaluation.
Client-Side Complexity per Round:
  • Local SGD Training: 0 ( | D | ⋅ P ) , where D i is the local dataset size. This is unchanged from standard FL
  • ZT Operations: The additional steps—model personalization, norm clipping, lightweight obfuscation 0 ( ⋅ ) , and secure transmission preparation are all 0 ( P ) . The dominant cost remains the local training. The adaptive verification (Step I in Algorithm 2) involves a constant-time cryptographic handshake, 0 ( 1 ) .
Server-Side Complexity per Round:
  • Trust Score Computation: Calculating Β i t for all clients involves computing the mean and variance of N updates of size P , resulting in 0 ( N ⋅ P ) complexity. Updating H i t is 0 ( N ⋅ K ) .
  • Trust-Weighted Aggregation: This is a weighted sum of N vectors of size P , which is 0 ( N ⋅ P ) , identical to the aggregation cost in FedAvg.
  • Policy Enforcement & Integrity Checks: These operations are also linear in model size or number of clients, 0 ( N ⋅ P ) .
Total Overhead: Both client and server overheads scale as 0 ( P ) and 0 ( N ⋅ P ) respectively, aligning with the fundamental complexity of federated learning. The constant factors are increased due to cryptographic and trust computations, but the asymptotic scalability remains unchanged. This analysis confirms the feasibility of deployment in large-scale, heterogeneous networks.
Convergence Guarantees
We provide a theoretical sketch demonstrating that ZT-FL-PE converges to a stationary point of the global objective under standard assumptions, despite the introduced trust-weighting and obfuscation. This is done under the following assumptions:
  • L-Lipschitz Smooth Gradient: The global loss function F ( ω ) has L-Lipschitz continuous gradients.
  • Bounded Stochastic Gradient Variance: The variance of stochastic gradients on each client is bounded by σ 2 .
  • Bounded Trust-Induced Perturbation: The trust-weighted aggregation and privacy obfuscation introduce a bounded bias. Formally, the aggregated update Δ t satisfies:
Ε [ Δ t ] = ∀ F ( ω t ) + ξ t ,   with ‖ ξ t ‖ ≤ Γ ,   and   its   variance   is   bounded   by   σ ~ 2
Rationale for Bounded Perturbation: The trust weights ω t i are normalized ( ∑ i ω i t = 1 ) and the obfuscation function 0 ( ⋅ ) (e.g., bounded noise injection, clipping) is designed to have a finite, tunable impact. This ensures Γ is a controllable parameter.
Convergence Result: Using a decaying learning rate η t = γ t , the expected gradient norm after T rounds is bounded by:
1 T ∑ t = 1 Τ Ε ‖ ∀ F ( ω t ) ‖ 2 ≤ C 1 Τ + C 2 ⋅ Γ 2
where C 1 , C 2 are constants depending on L , σ 2 , σ ~ 2 , and the initial conditions. Note that in the term C i T represents the standard sublinear convergence rate of SGD-based methods. The term C 2 ⋅ Γ 2 is a non-vanishing bias introduced by our ZTA-driven modifications. Crucially, Γ is a design parameter controlled by the intensity of the privacy obfuscation and the selectivity of the trust filter. By setting these mechanisms to operate within moderate bounds (as evidenced by our empirical results showing minimal accuracy loss), Γ remains small. Thus, ZT-FL-PE converges to a neighbourhood of a stationary point, with the neighbourhood radius proportional to the strength of the privacy and trust enforcement. This theoretical insight aligns perfectly with our experimental findings: ZT-FL-PE achieves a near-identical final accuracy as non-private FL (a small Γ ) while providing substantial privacy and security enhancements. The framework offers a principled trade-off where increased security/privacy ( Γ ) can be explicitly balanced against model utility.
In summary, the ZT-FL-PE algorithm provides:
  • Adaptive Security: A mathematically defined trust score that dynamically adjusts verification rigor based on client behaviour and the learning phase.
  • Controlled Overhead: Computational and communication complexity that preserves the fundamental scalability of federated learning.
  • Provable Convergence: Formal guarantees of convergence to a bounded region around an optimal point, with the region size explicitly linked to the privacy-security parameters.
As such, this formal foundation demonstrates that the integration of Zero Trust principles is not merely a heuristic addition but a rigorously designed component that enhances security and privacy while maintaining the core performance guarantees of federated learning.

5. Performance Results and Discussion

To ensure comprehensive evaluation across diverse domains and to rigorously validate the generalization of ZT-FL-PE for real-world ZTA scenarios, we employed datasets spanning image recognition, speech processing, structured tabular tasks, and critically, network security and endpoint management. The latter domains are of paramount importance as they directly mirror the data sources (network logs, device telemetry) used for AI-driven policy engines in ZTA. Table 2 summarizes all datasets. The inclusion of UNSW-NB15 and CIC-IDS2017 allows us to test FL-PE’s efficacy in privacy-preserving collaborative intrusion detection. The Drebin dataset tests its application to malware classification across heterogeneous mobile endpoints, a common challenge in zero-trust environments.
Table 2. Datasets used for Evaluation.
We benchmarked the proposed framework against three established privacy-preserving mechanisms from the literature: Local Differential Privacy (LDP), Central Differential Privacy (CDP), and Weak Differential Privacy (WDP). All experiments were conducted using PyTorch ver 2.10 [34] with NVIDIA A100 GPUs. The federated learning setup involved 100 clients with non-IID data distribution, following standard practices in federated learning research [9,17]. Each experiment was repeated 10 times with different random seeds, and results are reported as mean ± standard deviation.
We focused on evaluating metrics such as Attack Success Rate (ASR), Attack AUC (50% = random guessing), Privacy Budget Utilization, Model Utility, alongside computational metrics (time complexity, memory footprint, and communication overhead). We evaluated against established attack methodologies from the literature, including Membership Inference Attacks (MIA) [13,14], property inference attacks [15], gradient leakage attacks [16,17], and adversarial performance metrics.
We begin by examining how the proposed algorithm defends against membership inference attacks (MIAs) on both local and global models. Following established methodologies [13,14], each dataset is split into training and test sets, with half of the data allocated as the adversary’s prior knowledge and the remaining portion used for evaluation. We benchmark it against common defenses, namely WDP, LDP, and CDP [9,17].
Figure 3, Figure 4 and Figure 5 present the attack AUC for local and global models separately, with each bar representing a different defense. Across all evaluated datasets, ZT-FL-PE consistently lowers the attack AUC toward the random baseline of 50%, demonstrating strong privacy protection.
Figure 3. Comparative Analysis of Privacy Leakages: ZT-FL-PE vs. State-of-the-Art Protection Mechanisms.
Figure 4. Comparative Analysis of Privacy Leakages: FL-PE vs. State-of-the-Art Protection Mechanisms.
Figure 5. Comparative Analysis of Privacy Leakages: ZT-FL-PE vs. State-of-the-Art Protection Mechanisms.
Differential privacy methods, by contrast, demonstrate less reliable performance. For instance, on local models, ZT-FL-PE achieves up to a 29% reduction in privacy leakage (Figure 3), whereas WDP yields only 3%, and CDP reaches 25%.
This effectiveness arises from ZT-FL-PE’s design: sensitive layers are concealed, and parameters are substituted with randomized values. The resulting perturbation ensures that adversaries observe a distorted version of the target model, making its outputs difficult to align with those of the shadow model. This undermines the attack process and explains the observed drop in AUC. To assess the influence of ZT-FL-PE on model behavior, we study the loss profiles of member and non-member samples under each defense. Ideally, these distributions should overlap—preventing membership signals—while remaining concentrated around low loss values to reflect good predictive performance. On the Purchase100 dataset, we compute the loss separately for members and non-members across all defense settings.
The results show that ZT-FL-PE effectively reduces divergence between the two distributions while retaining mostly low loss values. Differential privacy mechanisms, however, either fail to close the gap or cause substantial accuracy loss.
Finally, for computational efficiency, we analyze the resource costs of ZT-FL-PE relative to differential privacy. We consider three metrics: client training time per round, server aggregation time, and GPU memory usage.
As illustrated in Figure 6, LDP, WDP, and CDP introduce significant slowdowns due to noise addition and gradient perturbation. ZT-FL-PE, however, incurs negligible computational cost, since its protection mechanism relies on lightweight parameter obfuscation rather than expensive gradient manipulation. This enables ZT-FL-PE to maintain efficiency while ensuring robust privacy, effectively addressing one of the primary drawbacks of differential privacy in FL. We further evaluate the temporal cost of server-side aggregation across different privacy-preserving configurations. Aggregation latency is the time elapsed between receiving all client model updates and transmitting the aggregated model. The simulation results in Figure 7 demonstrate that CDP incurs substantial latency overhead, primarily because it injects calibrated noise into the global parameter set prior to dissemination. This operation significantly prolongs the aggregation process, with delays of seconds observed in our experiments.
Figure 6. Client-side model training time frames.
Figure 7. Server-side FL aggregation time frames.
Conversely, ZT-FL-PE, LDP, and WDP exhibit aggregation times of comparable order to the non-privacy-preserving baseline, indicating that these approaches introduce negligible additional computational burden during aggregation.
We further evaluate GPU memory utilization, with an emphasis on the overhead induced by differential privacy (DP) mechanisms. Local DP, Central DP, and Weak DP each impose substantial memory requirements, resulting in a 15% increase over the no-defence baseline. This overhead stems primarily from (i) storing calibrated noise vectors, (ii) maintaining privacy budget bookkeeping, and (iii) buffering model updates prior to perturbation. As shown in Figure 8, ZT-FL-PE, in contrast, avoids noise injection and budget management by design, thereby incurring negligible GPU memory overhead, making it memory-optimal relative to DP-based defences.
Figure 8. Client-side memory utilisation.
Overall, while aggregate metrics (Figure 6, Figure 7 and Figure 8) demonstrate ZT-FL-PE’s efficiency relative to DP-based methods, a granular breakdown is essential to assess its practical deployment feasibility. We further breakdown the per-round computational overhead on a standard client device (8-core CPU, 16 GB RAM) for the CIFAR-100 experiment, comparing the baseline (No Defense), ZT-FL-PE, and CDP as a representative high-overhead method.
The breakdown reveals that ZT-FL-PE’s primary overhead sources are (1) the essential ZTA verification steps and (3) its efficient update obfuscation. Crucially, it avoids the expensive noise injection and privacy accounting that dominate CDP’s cost (Update Processing). The ~5% increase in total runtime is a reasonable trade-off for integrated ZTA compliance and strong privacy. The server-side overhead (additional logic for trust-weighted aggregation) is moderate and scales linearly with the number of clients, making it manageable for enterprise-grade infrastructure.

6. Conclusions

We proposed and evaluated a ZT-FL-PE algorithm designed primarily to address one of the most pressing privacy risks in FL MIAs. Unlike existing solutions, ZT-FL-PE was tested across a wide range of datasets, neural network architectures, and advanced privacy-preserving mechanisms, showing consistent performance. Our results demonstrate that the algorithm not only enhances privacy but also preserves model accuracy while maintaining low and well-understood computational costs—making it both effective and practical for real-world deployment. The overhead analysis, as provided in Table 3, confirms that ZT-FL-PE’s primary cost stems from mandatory ZTA verification, not from cumbersome cryptographic or noise-addition processes, resulting in a modest ~5% runtime increase over an undefended baseline.
Table 3. Per-Round Client-Side Computational Overhead Breakdown (Mean ± Std Dev in seconds).
Importantly, its benefits extend beyond membership inference. The algorithm shows strong potential to defend against broader privacy threats across diverse data domains, including network intrusion detection and endpoint security, as validated on multiple cross-domain benchmarks. This makes it a versatile tool for safeguarding FL systems within the heterogeneous data ecosystems of modern enterprises. Furthermore, integrating quantum-secure communication protocols could enhance the security of model update transmissions against emerging quantum threats, paving the way for even stronger defenses in FL systems.

Author Contributions

Conceptualization, K.S.; Investigation, B.P.; Resources, K.S.; Writing—original draft, B.P. and B.N.; Writing—review & editing, B.P., B.N. and K.S. All authors have read and agreed to the published version of the manuscript.

Funding

This research received no external funding.

Institutional Review Board Statement

Not applicable.

Data Availability Statement

The raw data supporting the conclusions of this article will be made available by the authors on request.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. National Institute of Standards and Technology. Zero Trust Architecture, Special Publication (NIST SP); NIST: Gaithersburg, MD, USA, 2020. [Google Scholar] [CrossRef] [Scilit]
  2. Gambo, M.L.; Almulhem, A. Zero trust architecture: A systematic literature review. J. Netw. Syst. Manag. 2026, 34, 25. [Google Scholar] [CrossRef] [Scilit]
  3. Dhiman, P.; Saini, N.; Gulzar, Y.; Turaev, S.; Kaur, A.; Nisa, K.U.; Hamid, Y. A Review and Comparative Analysis of Relevant Approaches of Zero Trust Network Model. Sensors 2024, 24, 1328. [Google Scholar] [CrossRef] [Scilit] [PubMed]
  4. Huber, B.; Kandah, F. Zero Trust+: A Trusted-based Zero Trust Architecture for IoT at Scale. In Proceedings of the 2024 IEEE International Conference on Consumer Electronics (ICCE), Las Vegas, NV, USA, 6–8 January 2024; pp. 1–6. [Google Scholar] [CrossRef] [Scilit]
  5. Hu, K.; Gong, S.; Zhang, Q.; Seng, C.; Xia, M.; Jiang, S. An overview of implementing security and privacy in federated learning. Artif. Intell. Rev. 2024, 57, 204. [Google Scholar] [CrossRef] [Scilit]
  6. Huber, B.; Kandah, F.I.; Skjellum, A. BEAST: Behavior as a Service for Trust management in IoT devices. Future Gener. Comput. Syst. 2023, 144, 165–178. [Google Scholar] [CrossRef] [Scilit]
  7. Chokkanathan, K.; Karpagavalli, S.M.; Priyanka, G.; Vanitha, K.; Anitha, K.; Shenbagavalli, P. AI-Driven Zero Trust Architecture: Enhancing Cyber-Security Resilience. In Proceedings of the 2024 8th International Conference on Computational System and Information Technology for Sustainable Solutions (CSITSS), Bengaluru, India, 7–9 November 2024; pp. 1–6. [Google Scholar] [CrossRef] [Scilit]
  8. Prajwalasimha, S.N.; Pimpalkar, A.; Shelke, N.M.; Bahadur Saini, D.K.J. Zero Trust Architectures Empowered by AI: A Paradigm Shift in Cloud and Edge Cybersecurity. In Proceedings of the 2025 3rd International Conference on Sustainable Computing and Data Communication Systems (ICSCDS), Erode, India, 6–8 August 2025; pp. 328–335. [Google Scholar] [CrossRef] [Scilit]
  9. Gosselin, R.; Vieu, L.; Loukil, F.; Benoit, A. Privacy and Security in Federated Learning: A Survey. Appl. Sci. 2022, 12, 9901. [Google Scholar] [CrossRef] [Scilit]
  10. Lazaros, K.; Koumadorakis, D.E.; Vrahatis, A.G.; Kotsiantis, S. Federated Learning: Navigating the Landscape of Collaborative Intelligence. Electronics 2024, 13, 4744. [Google Scholar] [CrossRef] [Scilit]
  11. Chai, L.; Xie, J.; Zhou, N. Prototype-based fine-tuning for mitigating data heterogeneity in federated learning. Future Gener. Comput. Syst. 2025, 170, 107831. [Google Scholar] [CrossRef] [Scilit]
  12. Pang, M.; Zhang, W.; Lu, Y.; Cheung, Y.-M.; Zhou, N. A unified multi-domain face normalization framework for cross-domain prototype learning and heterogeneous face recognition. IEEE Trans. Inf. Forensics Secur. 2025, 20, 5282–5295. [Google Scholar] [CrossRef] [Scilit]
  13. Shokri, R.; Stronati, M.; Song, C.; Shmatikov, V. Membership inference attacks against machine learning models. In Proceedings of the 2017 IEEE Symposium on Security and Privacy (SP), San Jose, CA, USA, 22–24 May 2017; IEEE: Piscataway, NJ, USA, 2017; pp. 3–18. [Google Scholar] [CrossRef] [Scilit]
  14. Yeom, S.; Giacomelli, I.; Fredrikson, M.; Jha, S. Privacy risk in machine learning: Analyzing the connection to overfitting. In Proceedings of the 2018 IEEE 31st Computer Security Foundations Symposium (CSF), Oxford, UK, 9–12 July 2018; IEEE: Piscataway, NJ, USA, 2018; pp. 268–282. [Google Scholar] [CrossRef] [Scilit]
  15. Ganju, K.; Wang, Q.; Yang, W.; Gunter, C.A.; Borisov, N. Property inference attacks on fully connected neural networks using permutation invariant representations. In Proceedings of the 2018 ACM SIGSAC Conference on Computer and Communications Security (CCS), Toronto, ON, Canada, 15–19 October 2018; ACM: New York, NY, USA, 2018; pp. 619–633. [Google Scholar] [CrossRef] [Scilit]
  16. Mo, F.; Borovykh, A.; Malekzadeh, M.; Haddadi, H.; Demetriou, S. Quantifying Information Leakage from Gradients. arXiv 2021, arXiv:2105.13929. [Google Scholar] [CrossRef] [Scilit]
  17. Zhu, L.; Liu, Z.; Han, S. Deep Leakage from Gradients. In Advances in Neural Information Processing Systems (NeurIPS); NSF: Alexandria, VA, USA, 2019; Available online: https://dl.acm.org/doi/10.5555/3454287.3455610 (accessed on 9 March 2026).
  18. Mo, F.; Xu, J.; Yang, W.; Borovykh, A.; Demetriou, S. Layer-wise Characterization of Latent Information Leakage in Federated Learning. In Proceedings of the Workshop on Distributed and Private Machine Learning at the International Conference on Learning Representations (ICLR), Virtual, 3–7 May 2021; Available online: https://arxiv.org/pdf/2010.08762v3 (accessed on 9 March 2026).
  19. Chhetri, B.; Gopali, S.; Olapojoye, R.; Dehbashi, S.; Namin, A.S. A Survey on Blockchain-Based Federated Learning and Data Privacy. In Proceedings of the 2023 IEEE 47th Annual Computers, Software, and Applications Conference (COMPSAC), Torino, Italy, 26–30 June 2023; IEEE: Piscataway, NJ, USA, 2023; pp. 1311–1318. [Google Scholar] [CrossRef] [Scilit]
  20. Kairouz, P.; McMahan, H.B.; Avent, B.; Bellet, A.; Bennis, M.; Bhagoji, A.N.; Bonawitz, K.; Charles, Z.; Cormode, G.; Cummings, R.; et al. Advances and open problems in federated learning. Found. Trends Mach. Learn. 2021, 14, 210p. [Google Scholar] [CrossRef] [Scilit]
  21. Fu, Y.; Xie, Y.; Yi, W.; Poudel, B.; Cao, J.; Li, H. IPO-ZTA: An Intelligent Policy Orchestration Zero Trust Architecture for B5G and 6G. Comput. Netw. 2025, 269, 111450. [Google Scholar] [CrossRef] [Scilit]
  22. Li, K.; Li, C.; Wang, Y.; Zhang, Z.; Liu, Y.; Chen, X.; Ni, W.; Niyato, D.; Jamalipour, A.; Dressler, F.; et al. Zero-trust foundation models: A new paradigm for secure and collaborative artificial intelligence for Internet of Things. arXiv 2025, arXiv:2505.23792. [Google Scholar] [CrossRef] [Scilit]
  23. Haque, E.; Hasan, K.; Ahmed, I.; Ullah, S.; Islam, T.; Pritom, M.M.A. Adaptive Intrusion Detection with PPO and Zero Trust Continuous Verification. In Proceedings of SoutheastCon 2025, Concord, NC, USA, 27–30 March 2025; pp. 868–873. [Google Scholar] [CrossRef] [Scilit]
  24. Krizhevsky, A. Learning Multiple Layers of Features from Tiny Images; University of Toronto Technical Report. 2009. Available online: http://www.cs.utoronto.ca/~kriz/learning-features-2009-TR.pdf (accessed on 9 March 2026).
  25. Xia, Q.; Tao, Z.; Li, Q. Defending against Byzantine attacks in quantum federated learning. In Proceedings of the 2021 17th International Conference on Mobility, Sensing and Networking (MSN), Exeter, UK, 13–15 December 2021; IEEE: Piscataway, NJ, USA, 2021; pp. 145–152. [Google Scholar] [CrossRef] [Scilit]
  26. He, X.; Xu, Y.; Zhang, S.; Xu, W.; Yan, J. Enhance membership inference attacks in federated learning. Comput. Secur. 2024, 136, 103535. [Google Scholar] [CrossRef] [Scilit]
  27. Stallkamp, J.; Schlipsing, M.; Salmen, J.; Igel, C. Man vs. computer: Benchmarking machine learning algorithms for traffic sign recognition. Neural Netw. 2012, 32, 323–332. [Google Scholar] [CrossRef] [Scilit] [PubMed]
  28. Warden, P. Speech commands: A dataset for limited-vocabulary speech recognition. arXiv 2018, arXiv:1804.03209. [Google Scholar] [CrossRef] [Scilit]
  29. Chen, C. Kaggle Acquire Valued Shoppers Challenge [Source Code and Dataset]. GitHub. 2014. Available online: https://github.com/ChenglongChen/Kaggle_Acquire_Valued_Shoppers_Challenge (accessed on 9 March 2026).
  30. Texas Department of State Health Services. Texas Hospital Inpatient Discharge Data [Data set]. Available online: https://rpcconsulting.com/health-data-store/texas-health-facility-data/thcic-inpatient-discharge-data/ (accessed on 9 March 2026).
  31. Liu, Z.; Luo, P.; Wang, X.; Tang, X. Deep learning face attributes in the wild. In Proceedings of the IEEE International Conference on Computer Vision (ICCV), Santiago, Chile, 13–16 December, 2015; pp. 3730–3738. [Google Scholar] [CrossRef] [Scilit]
  32. Moustafa, N.; Slay, J. UNSW-NB15: A comprehensive data set for network intrusion detection systems. In Proceedings of the Military Communications and Information Systems Conference (MilCIS), Canberra, Australia, 10–12 November 2015; pp. 1–6. [Google Scholar] [CrossRef] [Scilit]
  33. Sharafaldin, I.; Lashkari, A.H.; Ghorbani, A.A. Toward generating a new intrusion detection dataset and intrusion traffic characterization. In Proceedings of the International Conference on Information Systems Security and Privacy (ICISSP), Funchal, Portugal, 22–24 January 2018; pp. 108–116. [Google Scholar] [CrossRef] [Scilit]
  34. Paszke, A.; Gross, S.; Massa, F.; Lerer, A.; Bradbury, J.; Chanan, G.; Killeen, T.; Lin, Z.; Gimelshein, N.; Antiga, L.; et al. PyTorch: An imperative style, high-performance deep learning library. In Advances in Neural Information Processing Systems 32; Curran Associates, Inc.: Dutchess County, NY, USA, 2019; pp. 8024–8035. [Google Scholar]
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Article Metrics

Citations

Article Access Statistics

Multiple requests from the same IP address are counted as one view.