1. A method, comprising: determining, by a device, that feature messages being sent as input data to a machine learning model from a plurality of source nodes to an aggregation node are causing network congestion, wherein the feature messages are messages sent by the plurality of source nodes to the device, the messages including features relevant to the machine learning model; and in response to determining that the feature messages are causing network congestion: selecting, by the device, a set of one or more other nodes to perform aggregation of the machine learning model input data, selecting, by the device, a type of aggregation to be performed by the set of one or more other nodes, and instructing, by the device, the set of one or more other nodes to perform the selected type of aggregation on the data sent from the source nodes.