mcpField
A
ai-federated-learning

ai-federated-learning

github.com/mukul975Security32342 installsUpdated 32w ago
Repository

About ai-federated-learning

A guide for machine learning and privacy compliance teams covering cross-device, cross-silo, and vertical federated learning, along with secure aggregation, differential privacy, and GDPR considerations. It explains why keeping data local does not eliminate privacy risks.

Key features

  1. 01Describe three federated learning patterns
  2. 02Compare secure aggregation protocols
  3. 03Explain local and central differential privacy
  4. 04Assess GDPR compliance considerations
  5. 05Analyze gradient leakage and unlearning challenges

Use cases

  1. 01Plan cross-device or cross-organization training
  2. 02Choose secure aggregation for model updates
  3. 03Assess privacy budgets for federated training
  4. 04Identify GDPR risks in federated learning

Install

Copy the command and run it in a terminal.

$npx skills add https://github.com/mukul975/privacy-data-protection-skills --skill ai-federated-learning

After installing, type/ai-federated-learningin Claude Code to confirm it loaded.

Bundled files

References and scripts installed along with SKILL.md.

assets/
template.md
references/
standards.md
workflows.md
scripts/
process.py

Details

From the project README

Federated Learning for GDPR Compliance

Overview

Federated learning (FL) is a distributed machine learning approach that trains models across multiple data holders without centralising personal data. Instead of collecting training data into a central repository, federated learning sends the model to the data, computes local updates on each participant's device or server, and aggregates only model updates (gradients or weights) at a central coordinator. This architecture directly addresses GDPR data minimisation (Art. 5(1)(c)) and data protection by design (Art. 25) principles by eliminating the need to transfer and centralise personal data for AI training. However, federated learning is not a privacy silver bullet — it introduces its own privacy risks that must be managed through complementary techniques.

Federated Learning Architecture Patterns

Pattern 1: Cross-Device Federated Learning

Use case: Training on data from millions of user devices (smartphones, tablets, IoT).

ComponentDescription
ParticipantsEnd-user devices (smartphones, tablets, wearables)
ScaleThousands to millions of participants
DataSmall per-device, large aggregate (e.g., keyboard predictions, health metrics)
CoordinationCentral server selects participants per round, distributes model, aggregates updates
CommunicationCompressed gradient updates over mobile networks
Privacy riskIndividual gradient updates may leak information about device data

GDPR Analysis:

  • Data minimisation: personal data never leaves the device — strong compliance
  • Controller role: platform operator is controller; device owners are not processors
  • Lawful basis: consent or legitimate interest for on-device processing
  • International transfers: no personal data transfer if aggregation is privacy-preserving
  • Right to erasure: device can be excluded from future rounds; model unlearning may be needed
Pattern 2: Cross-Silo Federated Learning

Use case: Training across organisational boundaries (hospitals, banks, subsidiaries).

ComponentDescription
ParticipantsOrganisational data silos (hospitals, branches, partner companies)
Scale2 to 100 participants
DataLarge per-silo, structured (e.g., medical records, financial transactions)
CoordinationTrusted aggregator or peer-to-peer protocol
CommunicationModel updates over secure channels between organisations
Privacy riskGradient updates may reveal institutional data patterns

GDPR Analysis:

  • Joint controller determination: participants and aggregator may be joint controllers (Art. 26) — requires joint controller agreement
  • Data processing agreements: if aggregator is a processor, Art. 28 DPA required
  • Data minimisation: personal data stays within each silo — strong compliance
  • International transfers: if silos are in different jurisdictions, gradient transfer legality depends on privacy guarantees
  • Lawful basis: each silo needs independent lawful basis for local training; aggregation needs separate basis
Pattern 3: Vertical Federated Learning

Use case: Different organisations hold different features for the same individuals.

ComponentDescription
ParticipantsOrganisations with complementary data (bank + retailer sharing customer features)
Scale2 to 10 participants
DataDifferent features for overlapping individuals
CoordinationSecure multi-party computation for feature combination
CommunicationEncrypted intermediate representations
Privacy riskFeature linkage may reveal individual attributes across parties

GDPR Analysis:

  • Joint controller: participants are likely joint controllers — each processes personal data for a common purpose
  • Purpose limitation: combined training purpose must be compatible with each party's original collection purpose
  • Data subject notification: data subjects must be informed about cross-organisation federated training
  • Consent: may be required for combining data across organisations

Secure Aggregation Protocols

Protocol 1: Masking-Based Secure Aggregation

Participants mask their local updates with pairwise random masks that cancel out upon aggregation. The aggregator receives the sum without seeing individual updates.

PropertyValue
Privacy guaranteeIndividual updates not visible to aggregator or other participants
Computational costO(n^2) pairwise key agreement, O(n) masking per round
Communication cost2x baseline (masks + masked updates)
Dropout toleranceHandles participant dropout if sufficient participants remain
Collusion resistanceSecure against aggregator + up to t-1 participant collusion
Protocol 2: Homomorphic Encryption Aggregation

Participants encrypt their updates with a homomorphic encryption scheme. The aggregator computes the sum on encrypted data without decryption.

PropertyValue
Privacy guaranteeComputationally secure — updates encrypted throughout
Computational cost100-1000x overhead for encryption/decryption operations
Communication cost2-10x baseline (ciphertext expansion)
Dropout toleranceExcellent — encrypted updates can be summed independently
Collusion resistanceSecure against aggregator (does not hold decryption key)
Protocol 3: Trusted Execution Environment (TEE)

Aggregation occurs within a hardware-protected enclave (Intel SGX, ARM TrustZone). Participants send updates to the TEE, which performs aggregation in isolated memory.

PropertyValue
Privacy guaranteeHardware-based isolation — aggregator cannot inspect updates
Computational costNear-native (small overhead for enclave transitions)
Communication costBaseline (no encryption expansion for enclave-to-enclave)
Dropout toleranceExcellent
Collusion resistanceDepends on hardware trust model; vulnerable to side-channel attacks

Differential Privacy Integration

Local Differential Privacy (LDP)

Each participant adds noise to their gradient update before sending to the aggregator:

  • Mechanism: Gaussian noise calibrated to sensitivity and privacy budget (epsilon, delta)
  • Guarantee: Individual update is differentially private — aggregator cannot infer specific data points
  • Trade-off: Higher noise reduces model accuracy; need more participants to compensate
  • Recommended epsilon: 1-10 per round, with privacy accounting across rounds
Central Differential Privacy (CDP)

The aggregator adds noise to the aggregated update before applying to the global model:

  • Mechanism: Gaussian noise added to the sum of clipped gradients
  • Guarantee: Global model is differentially private with respect to any single participant's data
  • Trade-off: Less noise needed than LDP for same accuracy (noise averages out); requires trusted aggregator
  • Recommended epsilon: 1-8 total privacy budget
Privacy Budget Management
ParameterDescriptionGuidance
Epsilon (ε)Privacy loss parameter — lower is more privateε ≤ 1: strong privacy; ε ≤ 8: moderate; ε > 10: weak
Delta (δ)Probability of privacy failureδ < 1/N where N is dataset size
Rounds (T)Number of federated training roundsPrivacy degrades with rounds — use composition theorems
Clip norm (C)Maximum gradient norm per participantBalance between privacy (lower C) and convergence (higher C)
Noise multiplier (σ)Ratio of noise to sensitivityDetermined by ε, δ, C, and composition method

GDPR Compliance Assessment for Federated Learning

Data Protection by Design (Art. 25) Alignment
GDPR PrincipleFL ImplementationCompliance Status
Data minimisation (Art. 5(1)(c))Personal data stays local — only model updates transmittedStrong compliance
Purpose limitation (Art. 5(1)(b))Local processing for specified training purposeRequires per-participant purpose documentation
Storage limitation (Art. 5(1)(e))No central training data repository — data retained locally per participant's policyCompliance depends on participant retention
Integrity and confidentiality (Art. 5(1)(f))Secure aggregation protects update confidentialityStrong with SA + DP
Accuracy (Art. 5(1)(d))Model accuracy may differ from centralised trainingMonitor and document accuracy trade-offs
Remaining GDPR Challenges
  1. Gradient leakage: Model updates can leak training data even without transmitting raw data — secure aggregation and DP mitigate but do not eliminate
  2. Model memorization: The aggregated model may still memorize individual records — privacy auditing required post-training
  3. Right to erasure: Removing a participant's contribution from the trained model is challenging — machine unlearning or retraining required
  4. Controller/processor determination: Complex multi-party FL architectures require clear role delineation
  5. Lawful basis: Each participant needs independent lawful basis; the aggregator needs basis for processing updates
  6. Transparency: Data subjects must be informed about federated learning participation

Implementation Considerations

Communication Efficiency
TechniqueDescriptionPrivacy Impact
Gradient compressionQuantise or sparsify gradients before transmissionMay interact with DP noise — careful calibration needed
Federated averaging (FedAvg)Multiple local SGD steps before communicationReduces communication rounds; may increase per-round privacy cost
Gradient selectionSend only top-k gradient componentsLeaks which components are most significant — privacy concern
Participant Selection
StrategyDescriptionPrivacy Consideration
Random selectionUniformly random participant sampling per roundFair representation; privacy amplification through subsampling
Availability-basedSelect participants with sufficient resourcesMay bias toward certain participant profiles
Contribution-basedSelect participants whose data improves model mostReveals information about data distribution — privacy risk

Enforcement Relevance

While no enforcement action has specifically addressed federated learning, the technology is directly relevant to:

  • CNIL AI recommendations (2024): Identified federated learning as a recommended privacy-enhancing technology for AI training
  • EDPB Guidelines 04/2025: Mentioned distributed training as a data minimisation measure for AI systems
  • Garante v. OpenAI (2023): Centralised data collection was a key compliance issue — federated alternatives could have mitigated
  • AI Act Art. 10: Data governance for training data — FL enables data governance while maintaining data locality

Integration Points

  • ai-dpia: FL architecture reduces data centralisation risk in DPIA assessment
  • ai-training-lawfulness: FL may simplify lawful basis by keeping data local
  • ai-model-privacy-audit: Federated models require distributed privacy auditing
  • ai-privacy-inference: FL complements confidential computing for inference privacy
  • ai-data-retention: Data locality in FL simplifies retention compliance

Comments

Markdown supported