Aspects presented herein may enable a network entity to configure a group of UEs to simultaneously transmit reference signals and to simultaneously transmit gradient vectors to the network entity, such that the network entity may receive the gradient vectors from the group of UEs as an aggregated gradient vector over the air. In one aspect, a base transmits, to a group of UEs, a configuration that configures the group of UEs to simultaneously transmit one or more group-common reference signals and to simultaneously transmit one or more gradient vectors associated with a federated learning procedure. The network entity receives, from the group of UEs, the one or more group-common reference signals and the one or more gradient vectors based on the configuration via multiple channels. The network entity calculates an average gradient vector based on the one or more group-common reference signals and the one or more gradient vectors.