Model Multiplicity for Adversarial Detection in Small Language Model Training on Edge Devices
Not on the map. Why?
Abstract (arXiv)
The rise of edge-based machine learning has enabled distributed adaptation of language models across mobile and IoT devices, offering privacy preservation and real-time responsiveness. However, distributed fine-tuning of language models on untrusted or heterogeneous edge nodes introduces new vulnerabilities. Compromised or unreliable devices can inject poisoned updates, leading to stealthy model manipulation or convergence degradation. Classical defenses such as robust aggregation or temporal anomaly detection operate on a single global model and are therefore limited in detecting coordinated or persistent poisoning. This work proposes a new system-level defense based on model multiplicity. Instead of maintaining one global model, the system rotates or concurrently trains multiple small language models (e.g., DistilGPT-2), each updated by independently sampled subsets of edge nodes. These models evolve under distinct training trajectories, creating multiple independent views of the same distributed population. Divergence between models quantified through gradient similarity, loss evolution, or parameter variance serves as a signal of anomalous or adversarial behavior. When one model deviates significantly from the ensemble mean, the system flags its contributing nodes for isolation or re-weighting. We implement this framework and evaluate it on edge-scale simulations of Small Language Model (SLM) training under varying heterogeneity and attack conditions. Results show that model multiplicity enables earlier and more reliable detection of poisoning compared to classical single-model defenses such as Flanders and Robust methods. Our findings demonstrate that diversity in model evolution can serve as a practical and effective defense mechanism for secure distributed learning on resource-constrained edge devices.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 19 references that are in the map. First posted on arXiv 5 Jun 2026.
Builds on
14 of 19 references.
References · 14
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2018How To Backdoor Federated Learningcited by 765
- 2018Byzantine-Robust Distributed Learning: Towards Optimal Statistical Ratescited by 711
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 632
- 2024TinyLlama: An Open-Source Small Language Modelcited by 618
- 2019Local Model Poisoning Attacks to Byzantine-Robust Federated Learningcited by 493
- 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 482
- 2020FLTrust: Byzantine-robust Federated Learning via Trust Bootstrappingcited by 346
- 2018The Hidden Vulnerability of Distributed Learning in Byzantiumcited by 329
- 2019Robust Aggregation for Federated Learningcited by 317
- 2019A Little Is Enough: Circumventing Defenses For Distributed Learningcited by 288
2 more
Built on by
Not cited yet.