{"year":"2024","volume":235,"date_published":"2024-09-01T00:00:00Z","oa_version":"Preprint","citation":{"ama":"Scott JA, Cahill Á. Improved modelling of federated datasets using mixtures-of-Dirichlet-multinomials. In: <i>Proceedings of the 41st International Conference on Machine Learning</i>. Vol 235. ML Research Press; 2024:44012-44037.","ieee":"J. A. Scott and Á. Cahill, “Improved modelling of federated datasets using mixtures-of-Dirichlet-multinomials,” in <i>Proceedings of the 41st International Conference on Machine Learning</i>, Vienna, Austria, 2024, vol. 235, pp. 44012–44037.","ista":"Scott JA, Cahill Á. 2024. Improved modelling of federated datasets using mixtures-of-Dirichlet-multinomials. Proceedings of the 41st International Conference on Machine Learning. ICML: International Conference on Machine Learning, PMLR, vol. 235, 44012–44037.","short":"J.A. Scott, Á. Cahill, in:, Proceedings of the 41st International Conference on Machine Learning, ML Research Press, 2024, pp. 44012–44037.","mla":"Scott, Jonathan A., and Áine Cahill. “Improved Modelling of Federated Datasets Using Mixtures-of-Dirichlet-Multinomials.” <i>Proceedings of the 41st International Conference on Machine Learning</i>, vol. 235, ML Research Press, 2024, pp. 44012–37.","apa":"Scott, J. A., &#38; Cahill, Á. (2024). Improved modelling of federated datasets using mixtures-of-Dirichlet-multinomials. In <i>Proceedings of the 41st International Conference on Machine Learning</i> (Vol. 235, pp. 44012–44037). Vienna, Austria: ML Research Press.","chicago":"Scott, Jonathan A, and Áine Cahill. “Improved Modelling of Federated Datasets Using Mixtures-of-Dirichlet-Multinomials.” In <i>Proceedings of the 41st International Conference on Machine Learning</i>, 235:44012–37. ML Research Press, 2024."},"abstract":[{"lang":"eng","text":"In practice, training using federated learning can be orders of magnitude slower than standard centralized training. This severely limits the amount of experimentation and tuning that can be done, making it challenging to obtain good performance on a given task. Server-side proxy data can be used to run training simulations, for instance for hyperparameter tuning. This can greatly speed up the training pipeline by reducing the number of tuning runs to be performed overall on the true clients. However, it is challenging to ensure that these simulations accurately reflect the dynamics of the real federated training. In particular, the proxy data used for simulations often comes as a single centralized dataset without a partition into distinct clients, and partitioning this data in a naive way can lead to simulations that poorly reflect real federated training. In this paper we address the challenge of how to partition centralized data in a way that reflects the statistical heterogeneity of the true federated clients. We propose a fully federated, theoretically justified, algorithm that efficiently learns the distribution of the true clients and observe improved server-side simulations when using the inferred distribution to create simulated clients from the centralized data."}],"language":[{"iso":"eng"}],"type":"conference","conference":{"location":"Vienna, Austria","start_date":"2024-07-21","end_date":"2024-07-27","name":"ICML: International Conference on Machine Learning"},"scopus_import":"1","date_updated":"2024-10-01T09:21:32Z","publication":"Proceedings of the 41st International Conference on Machine Learning","corr_author":"1","publisher":"ML Research Press","status":"public","day":"01","quality_controlled":"1","intvolume":"       235","acknowledgement":"We would like to thank: Mona Chitnis and everyone in the Private Federated Learning team at Apple for their help and support throughout the entire project; Audra McMillan, Martin Pelikan, Anosh Raj and Barry Theobold for feedback on the initial versions of the paper; and Christoph Lampert for valuable feedback on the paper structure and suggestions for additional experiments.","author":[{"id":"e499926b-f6e0-11ea-865d-9c63db0031e8","full_name":"Scott, Jonathan A","last_name":"Scott","first_name":"Jonathan A"},{"first_name":"Áine","last_name":"Cahill","full_name":"Cahill, Áine"}],"external_id":{"arxiv":["2406.02416"]},"_id":"18120","publication_identifier":{"eissn":["2640-3498"]},"date_created":"2024-09-22T22:01:45Z","oa":1,"user_id":"2DF688A6-F248-11E8-B48F-1D18A9856A87","month":"09","title":"Improved modelling of federated datasets using mixtures-of-Dirichlet-multinomials","article_processing_charge":"No","page":"44012-44037","publication_status":"published","alternative_title":["PMLR"],"main_file_link":[{"open_access":"1","url":"https://doi.org/10.48550/arXiv.2406.02416"}],"department":[{"_id":"ChLa"}]}