S. Lee, T. Zhang, and A. S. Avestimehr, “Layer-Wise Adaptive Model Aggregation for Scalable Federated Learning”, AAAI, vol. 37, no. 7, pp. 8491-8499, Jun. 2023.