arXiv preprint ·
Client-Aware Security Adaptive Early Exit Vision Transformers for Heterogeneous Federated Learning
- Published
- Source
- arXiv preprint, posted before peer review
- Original
- Read it at arXiv
- Topics
Abstract
Federated learning (FL) trains a shared model across data-owning clients without centralizing raw data, but client data difficulty and compute budgets can differ. We study local early-exit Vision Transformer (ViT) inference, where an intermediate classifier may return a prediction before the final Transformer stage. The proposed method replaces one global acceptance threshold with a client-specific base threshold from label entropy and shallow-exit confidence, then adjusts each input using predictive entropy, confidence, top-two class margin, and exit depth. Aggregation uses only clients that updated a parameter. Security-aware routing reaches 88.70% CIFAR-10 accuracy with 32.0% depth saving and 73.7% ImageNet-1K top-1 accuracy with 30\% saving. On matched six-client non-IID CIFAR-100, full-depth accuracy is 72.28% versus 68.20--71.80% for four heterogeneous-FL baselines; robustness-aware routing reaches 70.01% with 32.55% saving.
The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.
Read it on ReadSearch
Each opens after you create an account, and brings you back to this paper.
More in Machine Learning
Replay Pruning and Few-Bit Integer Inference for Hardware-Efficient Speed-Sign Recognition
SENTRY: A Streaming Edge Network Trust Engine for Real-Time Audio Deepfake Detection
qGAMMA: A Framework for Optimal DNN Mapping on Multi-Precision Accelerators
Cryogenic Recharacterization of a Commercial 22 nm FD-SOI Foundry Standard-Cell Library
ConQUESTA: Topology-Aware Post-Training Structural Pruning of Quantum Neural Networks
Get the few papers worth your time in Machine Learning every morning.
ReadSearch reads the new papers in your field each day and brings you the few worth reading, each with its key finding in a sentence.