Single-Round Scalable Analytic Federated Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bacellar, Alan T. L., Munir, Mustafa, França, Felipe M. G., Lima, Priscila M. V., Marculescu, Radu, John, Lizy K.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911549389537280
author Bacellar, Alan T. L.
Munir, Mustafa
França, Felipe M. G.
Lima, Priscila M. V.
Marculescu, Radu
John, Lizy K.
author_facet Bacellar, Alan T. L.
Munir, Mustafa
França, Felipe M. G.
Lima, Priscila M. V.
Marculescu, Radu
John, Lizy K.
contents Federated Learning (FL) is plagued by two key challenges: high communication overhead and performance collapse on heterogeneous (non-IID) data. Analytic FL (AFL) provides a single-round, data distribution invariant solution, but is limited to linear models. Subsequent non-linear approaches, like DeepAFL, regain accuracy but sacrifice the single-round benefit. In this work, we break this trade-off. We propose SAFLe, a framework that achieves scalable non-linear expressivity by introducing a structured head of bucketed features and sparse, grouped embeddings. We prove this non-linear architecture is mathematically equivalent to a high-dimensional linear regression. This key equivalence allows SAFLe to be solved with AFL's single-shot, invariant aggregation law. Empirically, SAFLe establishes a new state-of-the-art for analytic FL, significantly outperforming both linear AFL and multi-round DeepAFL in accuracy across all benchmarks, demonstrating a highly efficient and scalable solution for federated vision.
format Preprint
id arxiv_https___arxiv_org_abs_2512_03336
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Single-Round Scalable Analytic Federated Learning
Bacellar, Alan T. L.
Munir, Mustafa
França, Felipe M. G.
Lima, Priscila M. V.
Marculescu, Radu
John, Lizy K.
Machine Learning
Artificial Intelligence
Federated Learning (FL) is plagued by two key challenges: high communication overhead and performance collapse on heterogeneous (non-IID) data. Analytic FL (AFL) provides a single-round, data distribution invariant solution, but is limited to linear models. Subsequent non-linear approaches, like DeepAFL, regain accuracy but sacrifice the single-round benefit. In this work, we break this trade-off. We propose SAFLe, a framework that achieves scalable non-linear expressivity by introducing a structured head of bucketed features and sparse, grouped embeddings. We prove this non-linear architecture is mathematically equivalent to a high-dimensional linear regression. This key equivalence allows SAFLe to be solved with AFL's single-shot, invariant aggregation law. Empirically, SAFLe establishes a new state-of-the-art for analytic FL, significantly outperforming both linear AFL and multi-round DeepAFL in accuracy across all benchmarks, demonstrating a highly efficient and scalable solution for federated vision.
title Single-Round Scalable Analytic Federated Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2512.03336