kth.sePublications KTH
Change search
Link to record
Permanent link

Direct link
Alternative names
Publications (10 of 114) Show all publications
Azimi Abarghouyi, S. M. & Fodor, V. (2026). A Hierarchical Federated Learning Approach for Internet of Things. IEEE Internet of Things Journal
Open this publication in new window or tab >>A Hierarchical Federated Learning Approach for Internet of Things
2026 (English)In: IEEE Internet of Things Journal, ISSN 2327-4662Article in journal (Refereed) Epub ahead of print
Abstract [en]

This paper presents a novel federated learning solution, QHetFed, suitable for Internet of Things deployments, addressing the challenges of clustered geographic distribution, communication resource limitation, and data heterogeneity. QHetFed is based on hierarchical federated learning over multiple device clusters, where the learning process and learning parameters take the necessary data quantization and the data heterogeneity into consideration to achieve high accuracy and fast convergence. Unlike conventional hierarchical federated learning algorithms, the proposed approach combines gradient aggregation in intra-cluster iterations with model aggregation in inter-cluster iterations. We offer a comprehensive analytical framework to evaluate its optimality gap and convergence rate, and give a closed form expression for the optimal learning parameters under a deadline, that accounts for communication and computation times. Our findings reveal that QHetFed consistently achieves high learning accuracy and significantly outperforms other hierarchical algorithms, particularly under heterogeneous data distributions.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2026
Keywords
data heterogeneity, distributed systems, Hierarchical federated learning, quantization
National Category
Computer Sciences
Identifiers
urn:nbn:se:kth:diva-375980 (URN)10.1109/JIOT.2026.3651118 (DOI)2-s2.0-105028004779 (Scopus ID)
Note

QC 20260205

Available from: 2026-02-05 Created: 2026-02-05 Last updated: 2026-02-05Bibliographically approved
Azimi Abarghouyi, S. M., Bastianello, N., Johansson, K. H. & Fodor, V. (2025). Hierarchical Federated ADMM. IEEE NETWORKING LETTERS, 7(1), 11-15
Open this publication in new window or tab >>Hierarchical Federated ADMM
2025 (English)In: IEEE NETWORKING LETTERS, ISSN 2576-3156, Vol. 7, no 1, p. 11-15Article in journal (Refereed) Published
Abstract [en]

In this letter, we depart from the widely-used gradient descent-based hierarchical federated learning (FL) algorithms to develop a novel hierarchical FL framework based on the alternating direction method of multipliers (ADMM), leveraging a network architecture consisting of a single cloud server and multiple edge servers, where each edge server is dedicated to a specific client set. Within this framework, we propose two novel FL algorithms, which both use ADMM in the top layer: one that employs ADMM in the lower layer and another that uses the conventional gradient descent-based approach. The proposed framework enhances privacy, and experiments demonstrate the superiority of the proposed algorithms compared to the conventional algorithms in terms of learning convergence and accuracy. Additionally, gradient descent on the lower layer performs well even if the number of local steps is very limited, while ADMM on both layers lead to better performance otherwise.

Place, publisher, year, edition, pages
IEEE, 2025
Keywords
Servers, Convex functions, Optimization, Linear programming, Privacy, Vectors, Training, Federated learning, Computational modeling, Accuracy, Machine learning, distributed optimization, ADMM, hierarchical networks
National Category
Control Engineering
Identifiers
urn:nbn:se:kth:diva-373461 (URN)10.1109/lnet.2025.3527161 (DOI)001554443500007 ()41116384 (PubMedID)2-s2.0-105001067715 (Scopus ID)
Note

QC 20251204

Available from: 2025-12-04 Created: 2025-12-04 Last updated: 2025-12-04Bibliographically approved
Naseer, M. Z., Fodor, V. & Ekstedt, M. (2025). Informed Defense: How Attacker Profiles Transform Vulnerability Assessments. In: Proceedings of the 2025 IEEE International Conference on Cyber Security and Resilience, CSR 2025: . Paper presented at 5th IEEE International Conference on Cyber Security and Resilience, CSR 2025, Chania, Greece, August 4-6, 2025 (pp. 453-460). Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>Informed Defense: How Attacker Profiles Transform Vulnerability Assessments
2025 (English)In: Proceedings of the 2025 IEEE International Conference on Cyber Security and Resilience, CSR 2025, Institute of Electrical and Electronics Engineers (IEEE) , 2025, p. 453-460Conference paper, Published paper (Refereed)
Abstract [en]

In the face of an evolving and increasingly complex threat landscape, organizations must adopt proactive approaches to assess and improve the resilience of their IT infrastructures against potential adversaries. Attack graphs are an effective tool to illustrate adversarial actions, but they often fail to capture the decision-making process of adversaries. To address this limitation, we map MITRE techniques to the attack steps in the attack graph and weight attempt probabilities at decision points according to the threat profile of the attacker. Considering a realistic, large IT infrastructure, we analyze how variations in attacker decision-making impact success rates, path diversity, the most frequent paths, and applied techniques. Our findings show that integrating attacker profiles into threat modeling can support accurate identification of the threat landscape and the optimization of defense strategies.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2025
Keywords
adversary profiles, attack graphs, attack simulation, threat modeling
National Category
Control Engineering Computer Sciences
Identifiers
urn:nbn:se:kth:diva-370758 (URN)10.1109/CSR64739.2025.11130094 (DOI)2-s2.0-105016165460 (Scopus ID)
Conference
5th IEEE International Conference on Cyber Security and Resilience, CSR 2025, Chania, Greece, August 4-6, 2025
Note

Part of ISBN 9798331535919

QC 20251001

Available from: 2025-10-01 Created: 2025-10-01 Last updated: 2025-10-01Bibliographically approved
Jeong, J., Hellström, H., Özgür, A., Fodor, V. & Fischione, C. (2025). Majority Vote Compressed Sensing for Over-the-Air Histogram Estimation. In: ICC 2025 - IEEE International Conference on Communications: . Paper presented at 2025 IEEE International Conference on Communications, ICC 2025, Montreal, Canada, June 8-12, 2025 (pp. 5742-5748). Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>Majority Vote Compressed Sensing for Over-the-Air Histogram Estimation
Show others...
2025 (English)In: ICC 2025 - IEEE International Conference on Communications, Institute of Electrical and Electronics Engineers (IEEE) , 2025, p. 5742-5748Conference paper, Published paper (Refereed)
Abstract [en]

We consider the problem of non-coherent over-the-air computation (AirComp), where n devices carry highdimensional data vectors xi ∈ Rd of sparsity ||xi||0 ≤ k and the sum of these data vectors has to be computed at a receiver. Previous results on non-coherent AirComp require more than d channel uses to compute functions of xi, where the extra redundancy is used to combat non-coherent signal aggregation. However, if the data vectors are sparse, sparsity can be exploited to offer significantly cheaper communication. In this paper, we propose to use random transforms to transmit lower-dimensional projections si ∈ RT of the data vectors. These projected vectors are communicated to the receiver using a majority vote (MV)AirComp scheme, which estimates the bit-vector corresponding to the signs of the aggregated projections, i.e., y=sign (Σi si). By leveraging 1-bit compressed sensing (1bCS) at the receiver, the real-valued and high-dimensional aggregate Σi xi can be recovered from y. We prove analytically that the proposed MVCS scheme estimates the aggregate data vector Σixi with ℓ2-norm error ϵ in T=O (k n log (d) / ϵ2) channel uses. We consider distributed histogram estimation, a canonical building block for federated analytics, as an aplication for MVCS where the data vectors xi are inherently 1 -sparse. Our numerical evaluations demonstrate that our scheme achieves the same order of communication cost as state-of-the-art methods while avoiding the complexity and overhead of additional cryptographic tools.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2025
Keywords
Compressed Sensing, Histogram Estimation, Majority Vote, Non-Coherent, Over-the-Air Computation
National Category
Telecommunications Signal Processing Communication Systems
Identifiers
urn:nbn:se:kth:diva-372516 (URN)10.1109/ICC52391.2025.11160930 (DOI)2-s2.0-105018466506 (Scopus ID)
Conference
2025 IEEE International Conference on Communications, ICC 2025, Montreal, Canada, June 8-12, 2025
Note

Part of ISBN 9798331505219

QC 20251107

Available from: 2025-11-07 Created: 2025-11-07 Last updated: 2025-11-07Bibliographically approved
Javeed, A., Fodor, V. & Dán, G. (2025). PERX: Energy-aware O-RAN Service Orchestration with Pairwise Performance Profiling. In: IEEE Conference on Computer Communications Workshops, INFOCOM WKSHPS 2025: . Paper presented at 2025 IEEE Conference on Computer Communications Workshops, INFOCOM WKSHPS 2025, London, United Kingdom of Great Britain and Northern Ireland, May 19, 2025. Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>PERX: Energy-aware O-RAN Service Orchestration with Pairwise Performance Profiling
2025 (English)In: IEEE Conference on Computer Communications Workshops, INFOCOM WKSHPS 2025, Institute of Electrical and Electronics Engineers (IEEE) , 2025Conference paper, Published paper (Refereed)
Abstract [en]

Motivated by the potential of machine-learning- based (ML) algorithms for radio access network (RAN) control and management, we consider the problem of energy-aware O-RAN service orchestration subject to ML inference time constraints. While ML applications enable complex operations in RAN control, guaranteeing service level agreements to close RAN operations in real time is a key requirement to facilitating their wider adoption. In this paper, we focus on orchestrating ML/AI workloads as near-real-time applications in O-RAN Cloud (O-Cloud). We propose PERX, an energy-efficient and performance-aware O-RAN orchestrator that predicts the performance of diverse sets of colocated ML/AL applications by learning a pairwise characterization of application inference times via hierarchical Bayesian learning. We formulate a latency-constrained integer optimization problem for application orchestration and propose an iterative procedure to solve the problem. In line with industry standards, we adopt Kubernetes as the orchestration framework to develop a latency-aware O-Cloud orchestrator. Experimental results reveal up to 50 % increase in profit with guaranteed service level agreements, compared to state of the art benchmarks.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2025
Keywords
O-RAN, Performance profiling, Service orchestration
National Category
Computer Systems Computer Sciences
Identifiers
urn:nbn:se:kth:diva-372337 (URN)10.1109/INFOCOMWKSHPS65812.2025.11152926 (DOI)2-s2.0-105017960408 (Scopus ID)
Conference
2025 IEEE Conference on Computer Communications Workshops, INFOCOM WKSHPS 2025, London, United Kingdom of Great Britain and Northern Ireland, May 19, 2025
Note

Part of ISBN 9798331543709

QC 20251106

Available from: 2025-11-06 Created: 2025-11-06 Last updated: 2025-11-06Bibliographically approved
Patil, R. S., Källman, I. & Fodor, V. (2024). DefenceRank - Ranking Based Attack Graph Analysis and Defence Prioritization. In: 2024 IEEE INTERNATIONAL CONFERENCE ON CYBER SECURITY AND RESILIENCE, CSR: . Paper presented at 4th IEEE Annual International Conference on Cyber Security and Resilience (IEEE CSR), SEP 02-04, 2024, London, ENGLAND (pp. 466-473). Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>DefenceRank - Ranking Based Attack Graph Analysis and Defence Prioritization
2024 (English)In: 2024 IEEE INTERNATIONAL CONFERENCE ON CYBER SECURITY AND RESILIENCE, CSR, Institute of Electrical and Electronics Engineers (IEEE), 2024, p. 466-473Conference paper, Published paper (Refereed)
Abstract [en]

Cyberinfrastructures are becoming larger and more complex, and it is increasingly challenging to assess for potential attacks and activate the appropriate defences. Attack graphs have been proven as a promising tool for risk assessment, but they also face the challenge of scalability. This paper proposes DefenceRank, an adaptation of Google's PageRank algorithm, to analyze large attack graphs and prioritize defences with low complexity. It incorporates the difficulty of the attack steps through the time-to-compromise parameter, the capabilities of the attack steps and the vulnerability of the assets. The proposed DefenceRank is evaluated on various realistic attack graphs. The results show that it achieves a reasonably high level of accuracy compared to optimal defence selection, while its time complexity increases polynomially with the size of the attack graph and remains in the order of seconds even for very large graphs and a large set of defences. In conclusion, DefenceRank demonstrates a viable alternative for the security assessment of cyberinfrastructures represented by attack graphs.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2024
National Category
Electrical Engineering, Electronic Engineering, Information Engineering
Identifiers
urn:nbn:se:kth:diva-356460 (URN)10.1109/CSR61664.2024.10679390 (DOI)001327167900070 ()2-s2.0-85206202681 (Scopus ID)
Conference
4th IEEE Annual International Conference on Cyber Security and Resilience (IEEE CSR), SEP 02-04, 2024, London, ENGLAND
Note

QC 20241119

Part of ISBN 979-8-3503-7536-7

Available from: 2024-11-19 Created: 2024-11-19 Last updated: 2024-11-19Bibliographically approved
Azimi Abarghouyi, S. M. & Fodor, V. (2024). Hierarchical Over-the-Air Federated Learning with Awareness of Interference and Data Heterogeneity. In: 2024 IEEE WIRELESS COMMUNICATIONS AND NETWORKING CONFERENCE, WCNC 2024: . Paper presented at IEEE Wireless Communications and Networking Conference (IEEE WCNC), APR 21-24, 2024, Dubai, U ARAB EMIRATES. Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>Hierarchical Over-the-Air Federated Learning with Awareness of Interference and Data Heterogeneity
2024 (English)In: 2024 IEEE WIRELESS COMMUNICATIONS AND NETWORKING CONFERENCE, WCNC 2024, Institute of Electrical and Electronics Engineers (IEEE) , 2024Conference paper, Published paper (Refereed)
Abstract [en]

When implementing hierarchical federated learning over wireless networks, scalability assurance and the ability to handle both interference and device data heterogeneity are crucial. This work introduces a learning method designed to address these challenges, along with a scalable transmission scheme that efficiently uses a single wireless resource through over-the-air computation. To provide resistance against data heterogeneity, we employ gradient aggregations. Meanwhile, the impact of interference is minimized through optimized receiver normalizing factors. For this, we model a multi-cluster wireless network using stochastic geometry, and characterize the mean squared error of the aggregation estimations as a function of the network parameters. We show that despite the interference and the data heterogeneity, the proposed scheme achieves high learning accuracy and can significantly outperform the conventional hierarchical algorithm.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2024
Series
IEEE Wireless Communications and Networking Conference, ISSN 1525-3511
Keywords
Federated learning, hierarchical networks, overthe-air computation, interference, stochastic geometry
National Category
Telecommunications
Identifiers
urn:nbn:se:kth:diva-354374 (URN)10.1109/WCNC57260.2024.10570672 (DOI)001268569301004 ()2-s2.0-85185388325 (Scopus ID)
Conference
IEEE Wireless Communications and Networking Conference (IEEE WCNC), APR 21-24, 2024, Dubai, U ARAB EMIRATES
Note

Part of ISBN 9798350303582

QC 20251002

Available from: 2024-10-04 Created: 2024-10-04 Last updated: 2025-10-02Bibliographically approved
Hellström, H., Jeong, J., Chen, W. N., Ozgur, A., Fodor, V. & Fischione, C. (2024). Over-the-Air Histogram Estimation. In: ICC 2024 - IEEE International Conference on Communications: . Paper presented at 59th Annual IEEE International Conference on Communications, ICC 2024, June 9-13, 2024, Denver, United States of America (pp. 4717-4722). Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>Over-the-Air Histogram Estimation
Show others...
2024 (English)In: ICC 2024 - IEEE International Conference on Communications, Institute of Electrical and Electronics Engineers (IEEE) , 2024, p. 4717-4722Conference paper, Published paper (Refereed)
Abstract [en]

We consider the problem of secure histogram es-timation, where n users hold private items xi from a size-d domain and a server aims to estimate the histogram of the user items. Previous results utilizing orthogonal communication schemes have shown that this problem can be solved securely with a total communication cost of O(n2log(d)) bits by hiding each item xi with a mask. In this paper, we offer a different approach to achieving secure aggregation. Instead of masking the data, our scheme protects individuals by aggregating their messages via a multiple-access channel. A naive communication scheme over the multiple-access channel requires d channel uses, which is generally worse than the O(n21og(d)) bits communication cost of the prior art in the most relevant regime d >> n. Instead, we propose a new scheme that we call Over-the-Air Group Testing (AirG T) which uses group testing codes to solve the histogram estimation problem in O(n log(d)) channel uses. AirGT reconstructs the histogram exactly with a vanishing probability of error Perror= O(d-T) that drops exponentially in the number of channel uses T.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2024
Keywords
Goal-Oriented Communications, Group Testing, Histogram Estimation, Non-Coherent, Over-the-Air Computation
National Category
Communication Systems Telecommunications Signal Processing
Identifiers
urn:nbn:se:kth:diva-353509 (URN)10.1109/ICC51166.2024.10622573 (DOI)001300022504139 ()2-s2.0-85202875872 (Scopus ID)
Conference
59th Annual IEEE International Conference on Communications, ICC 2024, June 9-13, 2024, Denver, United States of America
Note

Part of ISBN: 9781728190549

QC 20240924

Available from: 2024-09-19 Created: 2024-09-19 Last updated: 2025-12-05Bibliographically approved
Azimi Abarghouyi, S. M. & Fodor, V. (2024). Scalable Hierarchical Over-the-Air Federated Learning. IEEE Transactions on Wireless Communications, 23(8), 8480-8496
Open this publication in new window or tab >>Scalable Hierarchical Over-the-Air Federated Learning
2024 (English)In: IEEE Transactions on Wireless Communications, ISSN 1536-1276, E-ISSN 1558-2248, Vol. 23, no 8, p. 8480-8496Article in journal (Refereed) Published
Abstract [en]

When implementing hierarchical federated learning over wireless networks, scalability assurance and the ability to handle both interference and device data heterogeneity are crucial. This work introduces a new two-level learning method designed to address these challenges, along with a scalable over-the-air aggregation scheme for the uplink and a bandwidth-limited broadcast scheme for the downlink that efficiently use a single wireless resource. To provide resistance against data heterogeneity, we employ gradient aggregations. Meanwhile, the impact of uplink and downlink interference is minimized through optimized receiver normalizing factors. We present a comprehensive mathematical approach to derive the convergence bound for the proposed algorithm, applicable to a multi-cluster wireless network encompassing any count of collaborating clusters, and provide special cases and design remarks. As a key step to enable a tractable analysis, we develop a spatial model for the setup by modeling devices as a Poisson cluster process over the edge servers and rigorously quantify uplink and downlink error terms due to the interference. Finally, we show that despite the interference and data heterogeneity, the proposed algorithm not only achieves high learning accuracy for a variety of parameters but also significantly outperforms the conventional hierarchical learning algorithm.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2024
Keywords
Federated learning, machine learning, hierarchical systems, over-the-air computation
National Category
Telecommunications
Identifiers
urn:nbn:se:kth:diva-355308 (URN)10.1109/TWC.2024.3350923 (DOI)001329887800025 ()2-s2.0-85182923496 (Scopus ID)
Note

QC 20250922

Available from: 2024-10-29 Created: 2024-10-29 Last updated: 2025-09-22Bibliographically approved
Hellström, H., Fodor, V. & Fischione, C. (2023). Federated Learning Over-the-Air by Retransmissions. IEEE Transactions on Wireless Communications, 22(12), 9143-9156
Open this publication in new window or tab >>Federated Learning Over-the-Air by Retransmissions
2023 (English)In: IEEE Transactions on Wireless Communications, ISSN 1536-1276, E-ISSN 1558-2248, Vol. 22, no 12, p. 9143-9156Article in journal (Refereed) Published
Abstract [en]

Motivated by the increasing computational capabilities of wireless devices, as well as unprecedented levels of user- and device-generated data, new distributed machine learning (ML) methods have emerged. In the wireless community, Federated Learning (FL) is of particular interest due to its communication efficiency and its ability to deal with the problem of non-IID data. FL training can be accelerated by a wireless communication method called Over-the-Air Computation (AirComp) which harnesses the interference of simultaneous uplink transmissions to efficiently aggregate model updates. However, since AirComp utilizes analog communication, it introduces inevitable estimation errors. In this paper, we study the impact of such estimation errors on the convergence of FL and propose retransmissions as a method to improve FL accuracy over resource-constrained wireless networks. First, we derive the optimal AirComp power control scheme with retransmissions over static channels. Then, we investigate the performance of Over-the-Air FL with retransmissions and find two upper bounds on the FL loss function. Numerical results demonstrate that the power control scheme offers significant reductions in mean squared error. Additionally, we provide simulation results on MNIST classification with a deep neural network that reveals significant improvements in classification accuracy for low-SNR scenarios.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2023
Keywords
Federated Learning, Over-the-Air Computation, Retransmissions
National Category
Communication Systems Signal Processing Telecommunications
Research subject
Electrical Engineering
Identifiers
urn:nbn:se:kth:diva-327825 (URN)10.1109/twc.2023.3268742 (DOI)001128031700032 ()2-s2.0-85159703045 (Scopus ID)
Note

QC 20250923

Available from: 2023-05-31 Created: 2023-05-31 Last updated: 2025-09-23Bibliographically approved
Organisations
Identifiers
ORCID iD: ORCID iD iconorcid.org/0000-0002-2764-8099

Search in DiVA

Show all publications