google-deepmind / google-deepmind/asyncdiloco

Enhancing DiLoCo robustness: Addressing model poisoning via Semantic Gating

Open
#2 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
51
Forks
3
PR merge metrics
No merged PRs in 30d

Description

Hi DeepMind Team,

I’ve been studying your work on DiLoCo (Distributed Low-Communication) and its efficiency in high-latency environments. It's a game-changer for decentralized training.

However, during my benchmarks, I identified a significant vulnerability: the architecture is highly susceptible to model poisoning attacks during the local SGD phases. Since global synchronization is sparse, a single malicious actor can inject significant drift that bypasses standard numerical aggregators.

I’ve developed a defense layer called Semantic Guard that uses 32-dimensional latent representations (Semantic Atoms) to validate the "intent" of local updates before global aggregation. This ensures that only semantically consistent updates are merged.

Key Result:
In a simulation with 40% malicious nodes, the standard DiLoCo baseline collapsed (~50% accuracy), while the Semantic Guard maintained 98% accuracy.

I’ve open-sourced a reference implementation and the results here:
🔗 https://github.com/LelloOmwei/semantic-guard-diloco

I’d love to hear your thoughts on integrating such semantic validation layers into future iterations of asyncdiloco to enhance security in untrusted decentralized setups.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.