1. A method, comprising: determining, by a device, that input data to a machine learning model sent from a plurality of source nodes to an aggregation node is causing network congestion; selecting a set of one or more other nodes to perform aggregation of the machine learning model input data; selecting a type of aggregation to be performed by the set of one or more other nodes; and instructing the set of one or more other nodes to perform the selected type of aggregation on the data sent from the source nodes.